CCOL-2039: Add mock span to mock tracer

lib/deimos/test_helpers.rb

@@ -28,6 +28,7 @@ def unit_test!
           deimos_config.kafka.seed_brokers ||= ['test_broker']
           deimos_config.schema.backend = Deimos.schema_backend_class.mock_backend
           deimos_config.producers.backend = :test
+          deimos_config.tracer = Deimos::Tracing::Mock.new
         end
       end
 

lib/deimos/tracing/mock.rb

@@ -10,6 +10,7 @@ class Mock < Tracing::Provider
       def initialize(logger=nil)
         @logger = logger || Logger.new(STDOUT)
         @logger.info('MockTracingProvider initialized')
+        @active_span = MockSpan.new
       end
 
       # @param span_name [String]
@@ -32,18 +33,22 @@ def finish(span)
 
       # :nodoc:
       def active_span
-        nil
+        @active_span ||= MockSpan.new
       end
 
       # :nodoc:
       def set_tag(tag, value, span=nil)
-        nil
+        if span
+          span.set_tag(tag, value)
+        else
+          @span.set_tag(tag, value)
+        end
       end
 
       # Get a tag from a span with the specified tag.
       # @param tag [String]
       def get_tag(tag)
-        nil
+        @span.get_tag(tag)
       end
 
       # :nodoc:
@@ -53,5 +58,23 @@ def set_error(span, exception)
         @logger.info("Mock span '#{name}' set an error: #{exception}")
       end
     end
+
+    # Mock Span class
+    class MockSpan
+      # :nodoc:
+      def initialize
+        @span = {}
+      end
+
+      # :nodoc:
+      def set_tag(tag, value)
+        @span[tag] = value
+      end
+
+      # :nodoc:
+      def get_tag(tag)
+        @span[tag]
+      end
+    end
   end
 end

spec/active_record_batch_consumer_spec.rb

@@ -37,11 +37,6 @@ class Widget < ActiveRecord::Base
       consumer_class.config[:bulk_import_id_column] = :bulk_import_id # default
     end
 
-    before(:each) do
-      allow(Deimos.config.tracer.active_span).to receive(:set_tag)
-      allow(Deimos.config.tracer.active_span).to receive(:get_tag).with('topic').and_return(%w(topic:mytopic))
-    end
-
     around(:each) do |ex|
       # Set and freeze example time
       travel_to start do

spec/active_record_consume/mass_updater_spec.rb

@@ -45,7 +45,6 @@
       stub_const('Widget', widget_class)
       stub_const('Detail', detail_class)
       Widget.reset_column_information
-      allow(Deimos.config.tracer.active_span).to receive(:get_tag).with('topic').and_return(%w(topic:mytopic))
     end
 
     describe '#mass_update' do