[test] fixed test tooling

DataDog · Jun 9, 2017 · 1d4cfcd · 1d4cfcd
1 parent 6f39962
commit 1d4cfcd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/test/concurrent_test.rb b/test/concurrent_test.rb
@@ -12,7 +12,7 @@ def traced_task
       # and the instant the root span is done.
       sleep delay
     end
-    @tracer.writer.spans()
+    @tracer.writer.trace0_spans()
   end
 
   def test_parallel_tasks

diff --git a/test/helper.rb b/test/helper.rb
@@ -81,6 +81,13 @@ def spans
     spans.flatten
   end
 
+  def trace0_spans
+    return [] if @spans.empty?
+    spans = @spans[0]
+    @spans = @spans[1..@spans.size]
+    spans
+  end
+
   def services
     services = @services
     @services = []