Keep everything to sucker_punch test file.

Author: sarahchen6

spec/datadog/tracing/contrib/sucker_punch/patcher_spec.rb

@@ -50,6 +50,10 @@ def perform(action = :none, **_)
     end
   end
 
+  def get_spans_count(tracer = self.tracer)
+    fetch_traces(tracer).sum { |trace| trace.spans.length }
+  end
+
   context 'successful job' do
     subject(:dummy_worker_success) { worker_class.perform_async }
 

spec/datadog/tracing/contrib/support/tracer_helpers.rb

@@ -53,15 +53,6 @@ def fetch_spans(tracer = self.tracer)
       end
     end
 
-    def get_spans_count(tracer = self.tracer)
-      count = 0
-      traces = fetch_traces(tracer)
-      traces.each do |trace|
-        count += trace.instance_variable_get(:@spans).length
-      end
-      count
-    end
-
     # Remove all traces from the current tracer instance and
     # busts cache of +#spans+ and +#span+.
     def clear_traces!