Reuse TracePoint#return_value

Shopify · Sep 28, 2022 · bce5419 · bce5419
1 parent b534af3
commit bce5419
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/tapioca/runtime/trackers/constant_definition.rb b/lib/tapioca/runtime/trackers/constant_definition.rb
@@ -42,9 +42,10 @@ class ConstantLocation < T::Struct
 
         Tapioca.register_trace(:c_return) do |tp|
           next unless tp.method_id == :new
-          next unless Module === tp.return_value
 
           key = tp.return_value
+          next unless Module === key
+
           loc = build_constant_location(tp, caller_locations)
           (@class_files[key] ||= Set.new) << loc
         end