Merge pull request #183 from DataDog/pedro/update-provider

Enable access to context provider

lib/ddtrace/provider.rb

@@ -8,6 +8,11 @@ def initialize
       @context = Datadog::ThreadLocalContext.new
     end
 
+    # Sets the current context.
+    def context=(ctx)
+      @context.local = ctx
+    end
+
     # Return the current context.
     def context
       @context.local

lib/ddtrace/tracer.rb

@@ -18,7 +18,7 @@ module Datadog
   # of these function calls and sub-requests would be encapsulated within a single trace.
   # rubocop:disable Metrics/ClassLength
   class Tracer
-    attr_reader :writer, :sampler, :services, :tags
+    attr_reader :writer, :sampler, :services, :tags, :provider
     attr_accessor :enabled
     attr_writer :default_service
 

test/helper.rb

@@ -14,7 +14,6 @@ class Writer
   class Tracer
     remove_method :writer
     attr_accessor :writer
-    attr_reader :provider
   end
   module Workers
     class AsyncTransport

test/provider_test.rb

@@ -19,4 +19,12 @@ def test_default_provider
 
     assert_equal(ctx, ctx2)
   end
+
+  def test_setting_a_context
+    provider = Datadog::DefaultContextProvider.new
+    custom_context = Datadog::Context.new
+    provider.context = custom_context
+
+    assert_same(provider.context, custom_context)
+  end
 end

test/tracer_test.rb

@@ -356,4 +356,10 @@ def test_child_span_has_no_pid_metadata
     child = tracer.trace('something_else')
     assert_nil(child.get_tag('system.pid'))
   end
+
+  def test_provider
+    provider = Datadog::DefaultContextProvider.new
+    tracer = Datadog::Tracer.new(context_provider: provider)
+    assert_same(provider, tracer.provider)
+  end
 end