Keep everything to sucker_punch test file.

DataDog · Feb 7, 2025 · e02ad49 · e02ad49
1 parent cd45c06
commit e02ad49
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 9 deletions.
diff --git a/spec/datadog/tracing/contrib/sucker_punch/patcher_spec.rb b/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 }
 

diff --git a/spec/datadog/tracing/contrib/support/tracer_helpers.rb b/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!