appoptics_apm 4.0.2

data/test/mocked/excon_mocked_test.rb

@@ -0,0 +1,166 @@
+# Copyright (c) 2016 SolarWinds, LLC.
+# All rights reserved.
+
+if !defined?(JRUBY_VERSION)
+
+  require 'minitest_helper'
+  require 'webmock/minitest'
+  require 'mocha/mini_test'
+  WebMock.allow_net_connect!
+
+  class ExconTest < Minitest::Test
+
+    def setup
+      WebMock.enable!
+      WebMock.disable_net_connect!
+      AppOpticsAPM.config_lock.synchronize {
+        @sample_rate = AppOpticsAPM::Config[:sample_rate]
+        @tracing_mode= AppOpticsAPM::Config[:tracing_mode]
+      }
+    end
+
+    def teardown
+      AppOpticsAPM.config_lock.synchronize {
+        AppOpticsAPM::Config[:sample_rate] = @sample_rate
+        AppOpticsAPM::Config[:tracing_mode] = @tracing_mode
+        AppOpticsAPM::Config[:blacklist] = []
+      }
+      WebMock.reset!
+      WebMock.allow_net_connect!
+      WebMock.disable!
+    end
+
+    # ========== excon get =================================
+
+    def test_xtrace_no_trace
+      stub_request(:get, "http://127.0.0.6:8101/")
+
+      AppOpticsAPM.config_lock.synchronize do
+        ::Excon.get("http://127.0.0.6:8101/")
+      end
+
+      assert_requested :get, "http://127.0.0.6:8101/", times: 1
+      assert_not_requested :get, "http://127.0.0.6:8101/", headers: {'X-Trace'=>/^.*$/}
+    end
+
+    def test_xtrace_tracing
+      stub_request(:get, "http://127.0.0.7:8101/").to_return(status: 200, body: "", headers: {})
+
+      AppOpticsAPM::API.start_trace('excon_tests') do
+        ::Excon.get("http://127.0.0.7:8101/")
+      end
+
+      assert_requested :get, "http://127.0.0.7:8101/", times: 1
+      assert_requested :get, "http://127.0.0.7:8101/", headers: {'X-Trace'=>/^2B[0-9,A-F]{56}01/}, times: 1
+    end
+
+    def test_xtrace_tracing_not_sampling
+      stub_request(:get, "http://127.0.0.4:8101/").to_return(status: 200, body: "", headers: {})
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config[:sample_rate] = 0
+        AppOpticsAPM::API.start_trace('excon_test') do
+          ::Excon.get("http://127.0.0.4:8101/")
+        end
+      end
+
+      assert_requested :get, "http://127.0.0.4:8101/", times: 1
+      assert_requested :get, "http://127.0.0.4:8101/", headers: {'X-Trace'=>/^2B[0-9,A-F]*00$/}, times: 1
+      assert_not_requested :get, "http://127.0.0.4:8101/", headers: {'X-Trace'=>/^2B0*$/}
+    end
+
+    def test_xtrace_tracing_blacklisted
+      stub_request(:get, "http://127.0.0.3:8101/").to_return(status: 200, body: "", headers: {})
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config.blacklist << '127.0.0.3'
+        AppOpticsAPM::API.start_trace('excon_tests') do
+          ::Excon.get("http://127.0.0.3:8101/")
+        end
+      end
+
+      assert_requested :get, "http://127.0.0.3:8101/", times: 1
+      assert_not_requested :get, "http://127.0.0.3:8101/", headers: {'X-Trace'=>/.*/}
+    end
+
+    # ========== excon pipelined =================================
+
+    def test_xtrace_pipelined_tracing
+      stub_request(:get, "http://127.0.0.5:8101/").to_return(status: 200, body: "", headers: {})
+      stub_request(:put, "http://127.0.0.5:8101/").to_return(status: 200, body: "", headers: {})
+
+      AppOpticsAPM::API.start_trace('excon_tests') do
+        connection = ::Excon.new('http://127.0.0.5:8101/')
+        connection.requests([{:method => :get}, {:method => :put}])
+      end
+
+      assert_requested :get, "http://127.0.0.5:8101/", times: 1
+      assert_requested :put, "http://127.0.0.5:8101/", times: 1
+      assert_requested :get, "http://127.0.0.5:8101/", headers: {'X-Trace'=>/^2B[0-9,A-F]*01$/}, times: 1
+      assert_requested :put, "http://127.0.0.5:8101/", headers: {'X-Trace'=>/^2B[0-9,A-F]*01$/}, times: 1
+    end
+
+    def test_xtrace_pipelined_tracing_not_sampling
+      stub_request(:get, "http://127.0.0.2:8101/").to_return(status: 200, body: "", headers: {})
+      stub_request(:put, "http://127.0.0.2:8101/").to_return(status: 200, body: "", headers: {})
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config[:sample_rate] = 0
+        AppOpticsAPM::API.start_trace('excon_tests') do
+          connection = ::Excon.new('http://127.0.0.2:8101/')
+          connection.requests([{:method => :get}, {:method => :put}])
+        end
+      end
+
+      assert_requested :get, "http://127.0.0.2:8101/", times: 1
+      assert_requested :put, "http://127.0.0.2:8101/", times: 1
+      assert_requested :get, "http://127.0.0.2:8101/", headers: {'X-Trace'=>/^2B[0-9,A-F]*00$/}, times: 1
+      assert_requested :put, "http://127.0.0.2:8101/", headers: {'X-Trace'=>/^2B[0-9,A-F]*00$/}, times: 1
+      assert_not_requested :get, "http://127.0.0.2:8101/", headers: {'X-Trace'=>/^2B0*$/}
+      assert_not_requested :put, "http://127.0.0.2:8101/", headers: {'X-Trace'=>/^2B0*$/}
+    end
+
+    def test_xtrace_pipelined_no_trace
+      stub_request(:get, "http://127.0.0.8:8101/").to_return(status: 200, body: "", headers: {})
+      stub_request(:put, "http://127.0.0.8:8101/").to_return(status: 200, body: "", headers: {})
+
+      connection = ::Excon.new('http://127.0.0.8:8101/')
+      connection.requests([{:method => :get}, {:method => :put}])
+
+      assert_requested :get, "http://127.0.0.8:8101/", times: 1
+      assert_requested :put, "http://127.0.0.8:8101/", times: 1
+      assert_not_requested :get, "http://127.0.0.8:8101/", headers: {'X-Trace'=>/^.*$/}
+      assert_not_requested :put, "http://127.0.0.8:8101/", headers: {'X-Trace'=>/^.*$/}
+    end
+
+    def test_xtrace_pipelined_tracing_blacklisted
+      stub_request(:get, "http://127.0.0.9:8101/").to_return(status: 200, body: "", headers: {})
+      stub_request(:put, "http://127.0.0.9:8101/").to_return(status: 200, body: "", headers: {})
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config.blacklist << '127.0.0.9'
+        AppOpticsAPM::API.start_trace('excon_tests') do
+          connection = ::Excon.new('http://127.0.0.9:8101/')
+          connection.requests([{:method => :get}, {:method => :put}])
+        end
+      end
+
+      assert_requested :get, "http://127.0.0.9:8101/", times: 1
+      assert_requested :put, "http://127.0.0.9:8101/", times: 1
+      assert_not_requested :get, "http://127.9.0.8:8101/", headers: {'X-Trace'=>/^.*$/}
+      assert_not_requested :put, "http://127.9.0.8:8101/", headers: {'X-Trace'=>/^.*$/}
+    end
+
+    # ========== excon make sure headers are preserved =============================
+    def test_preserves_custom_headers
+      stub_request(:get, "http://127.0.0.6:8101/").to_return(status: 200, body: "", headers: {})
+
+      AppOpticsAPM::API.start_trace('excon_tests') do
+        Excon.get('http://127.0.0.6:8101', headers: { 'Custom' => 'specialvalue' })
+      end
+
+      assert_requested :get, "http://127.0.0.6:8101/", headers: {'Custom'=>'specialvalue'}, times: 1
+    end
+  end
+end
+

data/test/mocked/faraday_mocked_test.rb

@@ -0,0 +1,93 @@
+# Copyright (c) 2016 SolarWinds, LLC.
+# All rights reserved.
+
+if !defined?(JRUBY_VERSION)
+
+  require 'minitest_helper'
+  require 'webmock/minitest'
+  require 'mocha/mini_test'
+  WebMock.allow_net_connect!
+
+  class FaradayMockedTest < Minitest::Test
+
+    def setup
+      WebMock.enable!
+      WebMock.disable_net_connect!
+      AppOpticsAPM.config_lock.synchronize {
+        @sample_rate = AppOpticsAPM::Config[:sample_rate]
+      }
+    end
+
+    def teardown
+      AppOpticsAPM.config_lock.synchronize {
+        AppOpticsAPM::Config[:blacklist] = []
+        AppOpticsAPM::Config[:sample_rate] = @sample_rate
+      }
+      WebMock.reset!
+      WebMock.allow_net_connect!
+      WebMock.disable!
+    end
+
+    def test_tracing_sampling
+      stub_request(:get, "http://127.0.0.1:8101/")
+
+      AppOpticsAPM::API.start_trace('faraday_test') do
+        conn = Faraday.new(:url => 'http://127.0.0.1:8101') do |faraday|
+          faraday.adapter  Faraday.default_adapter  # make requests with Net::HTTP
+        end
+        conn.get
+      end
+
+      assert_requested :get, "http://127.0.0.1:8101/", times: 1
+      assert_requested :get, "http://127.0.0.1:8101/", headers: {'X-Trace'=>/^2B[0-9,A-F]*01$/}, times: 1
+    end
+
+    def test_tracing_not_sampling
+      stub_request(:get, "http://127.0.0.12:8101/")
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config[:sample_rate] = 0
+        AppOpticsAPM::API.start_trace('faraday_test') do
+          conn = Faraday.new(:url => 'http://127.0.0.12:8101') do |faraday|
+            faraday.adapter  Faraday.default_adapter  # make requests with Net::HTTP
+          end
+          conn.get
+        end
+      end
+
+      assert_requested :get, "http://127.0.0.12:8101/", times: 1
+      assert_requested :get, "http://127.0.0.12:8101/", headers: {'X-Trace'=>/^2B[0-9,A-F]*00$/}, times: 1
+      assert_not_requested :get, "http://127.0.0.12:8101/", headers: {'X-Trace'=>/^2B0*$/}
+    end
+
+    def test_no_xtrace
+      stub_request(:get, "http://127.0.0.3:8101/")
+
+      conn = Faraday.new(:url => 'http://127.0.0.3:8101') do |faraday|
+        faraday.adapter  Faraday.default_adapter  # make requests with Net::HTTP
+      end
+      conn.get
+
+      assert_requested :get, "http://127.0.0.3:8101/", times: 1
+      assert_not_requested :get, "http://127.0.0.3:8101/", headers: {'X-Trace'=>/^.*$/}
+    end
+
+    def test_blacklisted
+      stub_request(:get, "http://127.0.0.4:8101/")
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config.blacklist << '127.0.0.4'
+        AppOpticsAPM::API.start_trace('faraday_test') do
+          conn = Faraday.new(:url => 'http://127.0.0.4:8101') do |faraday|
+            faraday.adapter  Faraday.default_adapter  # make requests with Net::HTTP
+          end
+          conn.get
+        end
+      end
+
+      assert_requested :get, "http://127.0.0.4:8101/", times: 1
+      assert_not_requested :get, "http://127.0.0.4:8101/", headers: {'X-Trace'=>/^.*$/}
+    end
+
+  end
+end

data/test/mocked/http_mocked_test.rb

@@ -0,0 +1,129 @@
+# Copyright (c) 2016 SolarWinds, LLC.
+# All rights reserved.
+
+unless defined?(JRUBY_VERSION)
+  require 'minitest_helper'
+  require 'mocha/mini_test'
+  require 'net/http'
+
+  class HTTPMockedTest < Minitest::Test
+
+    class MockResponse
+      def get_fields(x); nil; end
+      def code; 200; end
+    end
+
+    def setup
+      AppOpticsAPM.config_lock.synchronize do
+        @sample_rate = AppOpticsAPM::Config[:sample_rate]
+      end
+    end
+
+    def teardown
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config[:sample_rate] = @sample_rate
+        AppOpticsAPM::Config[:blacklist] = []
+      end
+    end
+
+    # webmock not working, interferes with instrumentation
+
+    def test_tracing_sampling
+      Net::HTTP.any_instance.expects(:request_without_appoptics).with do |req, _|
+        !req.to_hash['x-trace'].nil? &&
+            req.to_hash['x-trace'].first =~ /^2B[0-9,A-F]*01$/
+      end.returns(MockResponse.new)
+
+      AppOpticsAPM::API.start_trace('net_http_test') do
+        uri = URI('http://127.0.0.1:8101/?q=1')
+        Net::HTTP.start(uri.host, uri.port) do |http|
+          request = Net::HTTP::Get.new(uri)
+          http.request(request)
+        end
+      end
+    end
+
+    def test_tracing_not_sampling
+      Net::HTTP.any_instance.expects(:request_without_appoptics).with do |req, _, _|
+        !req.to_hash['x-trace'].nil? &&
+            req.to_hash['x-trace'].first =~ /^2B[0-9,A-F]*00$/ &&
+            req.to_hash['x-trace'].first !~ /^2B0*$/
+      end.returns(MockResponse.new)
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config[:sample_rate] = 0
+        AppOpticsAPM::API.start_trace('Net::HTTP_test') do
+          uri = URI('http://127.0.0.1:8101/?q=1')
+          Net::HTTP.start(uri.host, uri.port) do |http|
+            request = Net::HTTP::Get.new(uri)
+            http.request(request) # Net::HTTPResponse object
+          end
+        end
+      end
+    end
+
+    def test_no_xtrace
+      Net::HTTP.any_instance.expects(:request_without_appoptics).with do |req, _, _|
+        req.to_hash['x-trace'].nil?
+      end.returns(MockResponse.new)
+
+      uri = URI('http://127.0.0.1:8101/?q=1')
+      Net::HTTP.start(uri.host, uri.port) do |http|
+        request = Net::HTTP::Get.new(uri)
+        http.request(request) # Net::HTTPResponse object
+      end
+    end
+
+    def test_blacklisted
+      Net::HTTP.any_instance.expects(:request_without_appoptics).with do |req, _, _|
+        req.to_hash['x-trace'].nil?
+      end.returns(MockResponse.new)
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config.blacklist << '127.0.0.1'
+        AppOpticsAPM::API.start_trace('Net::HTTP_tests') do
+          uri = URI('http://127.0.0.1:8101/?q=1')
+          Net::HTTP.start(uri.host, uri.port) do |http|
+            request = Net::HTTP::Get.new(uri)
+            http.request(request) # Net::HTTPResponse object
+          end
+        end
+      end
+    end
+
+    def test_not_sampling_blacklisted
+      Net::HTTP.any_instance.expects(:request_without_appoptics).with do |req, _, _|
+        req.to_hash['x-trace'].nil?
+      end.returns(MockResponse.new)
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config[:sample_rate] = 0
+        AppOpticsAPM::Config.blacklist << '127.0.0.1'
+        AppOpticsAPM::API.start_trace('Net::HTTP_tests') do
+          uri = URI('http://127.0.0.1:8101/?q=1')
+          Net::HTTP.start(uri.host, uri.port) do |http|
+            request = Net::HTTP::Get.new(uri)
+            http.request(request) # Net::HTTPResponse object
+          end
+        end
+      end
+    end
+
+    # ========== make sure headers are preserved =============================
+    def test_preserves_custom_headers
+      Net::HTTP.any_instance.expects(:request_without_appoptics).with do |req, _|
+        assert req.to_hash['custom'], "Custom header missing"
+        assert_match /specialvalue/, req.to_hash['custom'].first
+      end.returns(MockResponse.new)
+
+      AppOpticsAPM::API.start_trace('Net::HTTP_tests') do
+        uri = URI('http://127.0.0.1:8101/?q=1')
+        Net::HTTP.start(uri.host, uri.port) do |http|
+          request = Net::HTTP::Get.new(uri)
+          request['Custom'] = 'specialvalue'
+          http.request(request) # Net::HTTPResponse object
+        end
+      end
+    end
+  end
+end

data/test/mocked/httpclient_mocked_test.rb

@@ -0,0 +1,245 @@
+# Copyright (c) 2016 SolarWinds, LLC.
+# All rights reserved.
+
+unless defined?(JRUBY_VERSION)
+  require 'minitest_helper'
+  require 'webmock/minitest'
+  require 'mocha/mini_test'
+  WebMock.allow_net_connect!
+
+  class HTTPClientMockedTest < Minitest::Test
+
+    def setup
+      WebMock.enable!
+      WebMock.disable_net_connect!
+      AppOpticsAPM.config_lock.synchronize do
+        @sample_rate = AppOpticsAPM::Config[:sample_rate]
+      end
+    end
+
+    def teardown
+      WebMock.reset!
+      WebMock.allow_net_connect!
+      WebMock.disable!
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config[:sample_rate] = @sample_rate
+        AppOpticsAPM::Config[:blacklist] = []
+      end
+    end
+
+    #====== DO REQUEST ===================================================
+
+    def test_do_request_tracing_sampling_array_headers
+      stub_request(:get, "http://127.0.0.1:8101/")
+      AppOpticsAPM::API.start_trace('httpclient_test') do
+        clnt = HTTPClient.new
+        clnt.get('http://127.0.0.1:8101/', nil, [['some_header', 'some_value'], ['some_header2', 'some_value2']])
+      end
+
+      assert_requested :get, "http://127.0.0.1:8101/", times: 1
+      assert_requested :get, "http://127.0.0.1:8101/", headers: {'X-Trace'=>/^2B[0-9,A-F]*01$/}, times: 1
+    end
+
+    def test_do_request_tracing_sampling_hash_headers
+      stub_request(:get, "http://127.0.0.6:8101/")
+      AppOpticsAPM::API.start_trace('httpclient_test') do
+        clnt = HTTPClient.new
+        clnt.get('http://127.0.0.6:8101/', nil, { 'some_header' => 'some_value', 'some_header2' => 'some_value2' })
+      end
+
+      assert_requested :get, "http://127.0.0.6:8101/", times: 1
+      assert_requested :get, "http://127.0.0.6:8101/", headers: {'X-Trace'=>/^2B[0-9,A-F]*01$/}, times: 1
+    end
+
+    def test_do_request_tracing_not_sampling
+      stub_request(:get, "http://127.0.0.2:8101/")
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config[:sample_rate] = 0
+        AppOpticsAPM::API.start_trace('httpclient_test') do
+          clnt = HTTPClient.new
+          clnt.get('http://127.0.0.2:8101/')
+        end
+      end
+
+      assert_requested :get, "http://127.0.0.2:8101/", times: 1
+      assert_requested :get, "http://127.0.0.2:8101/", headers: {'X-Trace'=>/^2B[0-9,A-F]*00$/}, times: 1
+      assert_not_requested :get, "http://127.0.0.2:8101/", headers: {'X-Trace'=>/^2B0*$/}
+    end
+
+    def test_do_request_no_xtrace
+      stub_request(:get, "http://127.0.0.3:8101/")
+      clnt = HTTPClient.new
+      clnt.get('http://127.0.0.3:8101/')
+
+      assert_requested :get, "http://127.0.0.3:8101/", times: 1
+      assert_not_requested :get, "http://127.0.0.3:8101/", headers: {'X-Trace'=>/^.*$/}
+    end
+
+    def test_do_request_blacklisted
+      stub_request(:get, "http://127.0.0.4:8101/")
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config.blacklist << '127.0.0.4'
+        AppOpticsAPM::API.start_trace('httpclient_tests') do
+          clnt = HTTPClient.new
+          clnt.get('http://127.0.0.4:8101/')
+        end
+      end
+
+      assert_requested :get, "http://127.0.0.4:8101/"
+      assert_not_requested :get, "http://127.0.0.4:8101/", headers: {'X-Trace'=>/^.*$/}
+    end
+
+    def test_do_request_not_sampling_blacklisted
+      stub_request(:get, "http://127.0.0.5:8101/")
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config[:sample_rate] = 0
+        AppOpticsAPM::Config.blacklist << '127.0.0.5'
+        AppOpticsAPM::API.start_trace('httpclient_tests') do
+          clnt = HTTPClient.new
+          clnt.get('http://127.0.0.5:8101/')
+        end
+      end
+
+      assert_requested :get, "http://127.0.0.5:8101/"
+      assert_not_requested :get, "http://127.0.0.5:8101/", headers: {'X-Trace'=>/^.*$/}
+    end
+
+    #====== ASYNC REQUEST ================================================
+    # using expectations in these tests because stubbing doesn't work with threads
+
+    def test_async_tracing_sampling_array_headers
+      WebMock.disable!
+
+      Thread.expects(:new).yields   # continue without forking off a thread
+
+      HTTPClient.any_instance.expects(:do_get_stream_without_appoptics).with do |req, _, _|
+        'http://127.0.0.11:8101/' == req.header.request_uri.to_s &&
+            req.header['X-Trace'].first =~ /^2B[0-9,A-F]*01$/
+      end
+
+      AppOpticsAPM::API.start_trace('httpclient_test') do
+        clnt = HTTPClient.new
+        clnt.get_async('http://127.0.0.11:8101/', nil, [['some_header', 'some_value'], ['some_header2', 'some_value2']])
+      end
+    end
+
+    def test_async_tracing_sampling_hash_headers
+      WebMock.disable!
+
+      Thread.expects(:new).yields   # continue without forking off a thread
+
+      HTTPClient.any_instance.expects(:do_get_stream_without_appoptics).with do |req, _, _|
+        'http://127.0.0.16:8101/' == req.header.request_uri.to_s &&
+        req.header['X-Trace'].first =~ /^2B[0-9,A-F]*01$/
+      end
+
+      AppOpticsAPM::API.start_trace('httpclient_test') do
+        clnt = HTTPClient.new
+        clnt.get_async('http://127.0.0.16:8101/', nil, { 'some_header' => 'some_value', 'some_header2' => 'some_value2' })
+      end
+    end
+
+    def test_async_tracing_not_sampling
+      WebMock.disable!
+
+      Thread.expects(:new).yields   # continue without forking off a thread
+
+      HTTPClient.any_instance.expects(:do_get_stream_without_appoptics).with do |req, _, _|
+        'http://127.0.0.12:8101/' == req.header.request_uri.to_s &&
+            req.header['X-Trace'].first =~  /^2B[0-9,A-F]*00$/ &&
+            req.header['X-Trace'].first !~ /^2B0*$/
+      end
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config[:sample_rate] = 0
+        AppOpticsAPM::API.start_trace('httpclient_test') do
+          clnt = HTTPClient.new
+          clnt.get_async('http://127.0.0.12:8101/')
+        end
+      end
+    end
+
+    def test_async_no_xtrace
+      WebMock.disable!
+
+      Thread.expects(:new).yields   # continue without forking off a thread
+
+      HTTPClient.any_instance.expects(:do_get_stream_without_appoptics).with do |req, _, _|
+        'http://127.0.0.13:8101/' == req.header.request_uri.to_s &&
+            req.header['X-Trace'].empty?
+      end
+
+      clnt = HTTPClient.new
+      clnt.get_async('http://127.0.0.13:8101/')
+    end
+
+    def test_async_blacklisted
+      WebMock.disable!
+
+      Thread.expects(:new).yields   # continue without forking off a thread
+
+      HTTPClient.any_instance.expects(:do_get_stream_without_appoptics).with do |req, _, _|
+        'http://127.0.0.14:8101/' == req.header.request_uri.to_s &&
+            req.header['X-Trace'].empty?
+      end
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config.blacklist << '127.0.0.14'
+        AppOpticsAPM::API.start_trace('httpclient_tests') do
+          clnt = HTTPClient.new
+          clnt.get_async('http://127.0.0.14:8101/')
+        end
+      end
+    end
+
+    def test_async_not_sampling_blacklisted
+      WebMock.disable!
+
+      Thread.expects(:new).yields   # continue without forking off a thread
+
+      HTTPClient.any_instance.expects(:do_get_stream_without_appoptics).with do |req, _, _|
+        'http://127.0.0.15:8101/' == req.header.request_uri.to_s &&
+            req.header['X-Trace'].empty?
+      end
+
+      AppOpticsAPM.config_lock.synchronize do
+        AppOpticsAPM::Config[:sample_rate] = 0
+        AppOpticsAPM::Config.blacklist << '127.0.0.15'
+        AppOpticsAPM::API.start_trace('httpclient_tests') do
+          clnt = HTTPClient.new
+          clnt.get_async('http://127.0.0.15:8101/')
+        end
+      end
+    end
+
+    # ========== make sure headers are preserved =============================
+    def test_preserves_custom_headers
+      stub_request(:get, "http://127.0.0.6:8101/").to_return(status: 200, body: "", headers: {})
+
+      AppOpticsAPM::API.start_trace('httpclient_tests') do
+        clnt = HTTPClient.new
+        clnt.get('http://127.0.0.6:8101/', nil, [['Custom', 'specialvalue'], ['some_header2', 'some_value2']])
+      end
+
+      assert_requested :get, "http://127.0.0.6:8101/", headers: {'Custom'=>'specialvalue'}, times: 1
+    end
+
+    def test_async_preserves_custom_headers
+      WebMock.disable!
+
+      Thread.expects(:new).yields   # continue without forking off a thread
+
+      HTTPClient.any_instance.expects(:do_get_stream_without_appoptics).with do |req, _, _|
+        assert req.headers['Custom'], "Custom header missing"
+        assert_match(/^specialvalue$/, req.headers['Custom'] )
+      end
+
+      AppOpticsAPM::API.start_trace('httpclient_tests') do
+        clnt = HTTPClient.new
+        clnt.get_async('http://127.0.0.6:8101/', nil, [['Custom', 'specialvalue'], ['some_header2', 'some_value2']])
+      end
+    end
+  end
+end