oboe 2.2.6 → 2.3.2

data/test/instrumentation/dalli_test.rb

@@ -0,0 +1,157 @@
+require 'minitest_helper'
+
+describe Oboe::Inst::Dalli do
+  before do
+    clear_all_traces 
+    @dc = Dalli::Client.new
+    @collect_backtraces = Oboe::Config[:dalli][:collect_backtraces]
+  end
+
+  after do
+    Oboe::Config[:dalli][:collect_backtraces] = @collect_backtraces
+  end
+
+  it 'Stock Dalli should be loaded, defined and ready' do
+    defined?(::Dalli).wont_match nil 
+    defined?(::Dalli::Client).wont_match nil
+  end
+
+  it 'should have oboe methods defined' do
+    [ :perform_with_oboe, :get_multi_with_oboe ].each do |m|
+      ::Dalli::Client.method_defined?(m).must_equal true
+    end
+  end
+
+  it 'should trace set' do
+    Oboe::API.start_trace('dalli_test', '', {}) do
+      @dc.get('some_key')
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 5
+
+    validate_outer_layers(traces, 'dalli_test')
+
+    traces[1].has_key?("KVOp").must_equal true
+    traces[1].has_key?("KVKey").must_equal true
+    traces[1]['Layer'].must_equal "memcache"
+    traces[1]['KVKey'].must_equal "some_key"
+
+    traces[2]['Layer'].must_equal "memcache"
+    traces[2]['Label'].must_equal "info"
+    traces[2].has_key?('KVHit')
+  end
+
+  it 'should trace get' do
+    Oboe::API.start_trace('dalli_test', '', {}) do
+      @dc.set('some_key', 1)
+    end
+    
+    traces = get_all_traces
+    traces.count.must_equal 4
+    
+    validate_outer_layers(traces, 'dalli_test')
+
+    traces[1]['KVOp'].must_equal "set"
+    traces[1]['KVKey'].must_equal "some_key"
+    traces[2]['Label'].must_equal "exit"
+  end
+
+  it 'should trace get_multi' do
+    Oboe::API.start_trace('dalli_test', '', {}) do
+      @dc.get_multi([:one, :two, :three, :four, :five, :six])
+    end
+    
+    traces = get_all_traces
+    traces.count.must_equal 5
+    
+    validate_outer_layers(traces, 'dalli_test')
+    
+    traces[1]['KVOp'].must_equal "get_multi"
+    traces[2]['Label'].must_equal "info"
+    traces[2].has_key?('KVKeyCount').must_equal true
+    traces[2].has_key?('KVHitCount').must_equal true
+    traces[3]['Label'].must_equal "exit"
+  end
+
+  it "should trace increment" do
+    Oboe::API.start_trace('dalli_test', '', {}) do
+      @dc.incr("some_key_counter", 1, nil, 0)
+    end
+    
+    traces = get_all_traces
+    traces.count.must_equal 4
+    
+    validate_outer_layers(traces, 'dalli_test')
+
+    traces[1]['KVOp'].must_equal "incr"
+    traces[1]['KVKey'].must_equal "some_key_counter"
+    traces[2]['Label'].must_equal "exit"
+  end
+
+  it "should trace decrement" do
+    Oboe::API.start_trace('dalli_test', '', {}) do
+      @dc.decr("some_key_counter", 1, nil, 0)
+    end
+    
+    traces = get_all_traces
+    traces.count.must_equal 4
+    
+    validate_outer_layers(traces, 'dalli_test')
+
+    traces[1]['KVOp'].must_equal "decr"
+    traces[1]['KVKey'].must_equal "some_key_counter"
+    traces[2]['Label'].must_equal "exit"
+  end
+
+  it "should trace replace" do
+    Oboe::API.start_trace('dalli_test', '', {}) do
+      @dc.replace("some_key", "woop")
+    end
+    
+    traces = get_all_traces
+    traces.count.must_equal 4
+
+    validate_outer_layers(traces, 'dalli_test')
+    
+    traces[1]['KVOp'].must_equal "replace"
+    traces[1]['KVKey'].must_equal "some_key"
+    traces[2]['Label'].must_equal "exit"
+  end
+
+  it "should trace delete" do
+    Oboe::API.start_trace('dalli_test', '', {}) do
+      @dc.delete("some_key")
+    end
+    
+    traces = get_all_traces
+    traces.count.must_equal 4
+
+    validate_outer_layers(traces, 'dalli_test')
+
+    traces[1]['KVOp'].must_equal "delete"
+    traces[1]['KVKey'].must_equal "some_key"
+  end
+  
+  it "should obey :collect_backtraces setting when true" do
+    Oboe::Config[:dalli][:collect_backtraces] = true
+
+    Oboe::API.start_trace('dalli_test', '', {}) do
+      @dc.get('some_key')
+    end
+
+    traces = get_all_traces
+    layer_has_key(traces, 'memcache', 'Backtrace')
+  end
+
+  it "should obey :collect_backtraces setting when false" do
+    Oboe::Config[:dalli][:collect_backtraces] = false
+
+    Oboe::API.start_trace('dalli_test', '', {}) do
+      @dc.get('some_key')
+    end
+
+    traces = get_all_traces
+    layer_doesnt_have_key(traces, 'memcache', 'Backtrace')
+  end
+end

data/test/instrumentation/http_test.rb

@@ -0,0 +1,74 @@
+require 'minitest_helper'
+require 'net/http'
+
+describe Oboe::Inst do
+  before do
+    clear_all_traces 
+    @collect_backtraces = Oboe::Config[:nethttp][:collect_backtraces]
+  end
+
+  after do
+    Oboe::Config[:nethttp][:collect_backtraces] = @collect_backtraces
+  end
+
+  it 'Net::HTTP should be defined and ready' do
+    defined?(::Net::HTTP).wont_match nil 
+  end
+
+  it 'Net::HTTP should have oboe methods defined' do
+    [ :request_with_oboe ].each do |m|
+      ::Net::HTTP.method_defined?(m).must_equal true
+    end
+  end
+
+  it "should trace a Net::HTTP request" do
+    Oboe::API.start_trace('net-http_test', '', {}) do
+      uri = URI('https://www.appneta.com')
+      http = Net::HTTP.new(uri.host, uri.port)
+      http.use_ssl = true
+      http.get('/?q=test').read_body
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 5
+    
+    validate_outer_layers(traces, 'net-http_test')
+
+    traces[1]['Layer'].must_equal 'net-http'
+    traces[2]['IsService'].must_equal "1"
+    traces[2]['RemoteProtocol'].must_equal "HTTPS"
+    traces[2]['RemoteHost'].must_equal "www.appneta.com"
+    traces[2]['ServiceArg'].must_equal "/?q=test"
+    traces[2]['HTTPMethod'].must_equal "GET"
+    traces[2]['HTTPStatus'].must_equal "200"
+    traces[2].has_key?('Backtrace').must_equal Oboe::Config[:nethttp][:collect_backtraces]
+  end
+  
+  it "should obey :collect_backtraces setting when true" do
+    Oboe::Config[:nethttp][:collect_backtraces] = true
+
+    Oboe::API.start_trace('nethttp_test', '', {}) do
+      uri = URI('https://www.appneta.com')
+      http = Net::HTTP.new(uri.host, uri.port)
+      http.use_ssl = true
+      http.get('/?q=test').read_body
+    end
+
+    traces = get_all_traces
+    layer_has_key(traces, 'net-http', 'Backtrace')
+  end
+
+  it "should obey :collect_backtraces setting when false" do
+    Oboe::Config[:nethttp][:collect_backtraces] = false
+
+    Oboe::API.start_trace('nethttp_test', '', {}) do
+      uri = URI('https://www.appneta.com')
+      http = Net::HTTP.new(uri.host, uri.port)
+      http.use_ssl = true
+      http.get('/?q=test').read_body
+    end
+
+    traces = get_all_traces
+    layer_doesnt_have_key(traces, 'net-http', 'Backtrace')
+  end
+end

data/test/instrumentation/memcache_test.rb

@@ -0,0 +1,251 @@
+require 'minitest_helper'
+require 'memcache'
+
+describe Oboe::API::Memcache do
+  before do
+    clear_all_traces 
+    @mc = ::MemCache.new('localhost')
+
+    # These are standard entry/exit KVs that are passed up with all mongo operations
+    @entry_kvs = {
+      'Layer' => 'memcache',
+      'Label' => 'entry' }
+
+    @info_kvs = {
+      'Layer' => 'memcache',
+      'Label' => 'info' }
+
+    @exit_kvs = { 'Layer' => 'memcache', 'Label' => 'exit' }
+    @collect_backtraces = Oboe::Config[:memcache][:collect_backtraces]
+  end
+
+  after do
+    Oboe::Config[:memcache][:collect_backtraces] = @collect_backtraces
+  end
+
+  it 'Stock MemCache should be loaded, defined and ready' do
+    defined?(::MemCache).wont_match nil 
+  end
+
+  it 'MemCache should have oboe methods defined' do
+    Oboe::API::Memcache::MEMCACHE_OPS.each do |m|
+      if ::MemCache.method_defined?(m)
+        ::MemCache.method_defined?("#{m}_with_oboe").must_equal true 
+      end
+      ::MemCache.method_defined?(:request_setup_with_oboe).must_equal true 
+      ::MemCache.method_defined?(:cache_get_with_oboe).must_equal true 
+      ::MemCache.method_defined?(:get_multi_with_oboe).must_equal true 
+    end
+  end
+  
+  it "should trace set" do
+    Oboe::API.start_trace('memcache_test', '', {}) do
+      @mc.set('msg', 'blah')
+    end
+    
+    traces = get_all_traces
+
+    traces.count.must_equal 5
+    validate_outer_layers(traces, 'memcache_test')
+
+    validate_event_keys(traces[1], @entry_kvs)
+    
+    traces[1]['KVOp'].must_equal "set"
+    traces[1].has_key?('Backtrace').must_equal Oboe::Config[:memcache][:collect_backtraces]
+    
+    validate_event_keys(traces[2], @info_kvs)
+    traces[2]['KVKey'].must_equal "msg"
+    traces[2].has_key?('Backtrace').must_equal Oboe::Config[:memcache][:collect_backtraces]
+
+    validate_event_keys(traces[3], @exit_kvs)
+  end
+  
+  it "should trace get" do
+    Oboe::API.start_trace('memcache_test', '', {}) do
+      @mc.get('msg')
+    end
+    
+    traces = get_all_traces
+    
+    traces.count.must_equal 6
+    validate_outer_layers(traces, 'memcache_test')
+
+    validate_event_keys(traces[1], @entry_kvs)
+    
+    traces[1]['KVOp'].must_equal "get"
+    traces[1].has_key?('Backtrace').must_equal Oboe::Config[:memcache][:collect_backtraces]
+    
+    validate_event_keys(traces[2], @info_kvs)
+    traces[2]['KVKey'].must_equal "msg"
+    traces[2]['RemoteHost'].must_equal "localhost"
+    traces[2].has_key?('Backtrace').must_equal Oboe::Config[:memcache][:collect_backtraces]
+
+    traces[3].has_key?('KVHit').must_equal true
+    traces[3].has_key?('Backtrace').must_equal Oboe::Config[:memcache][:collect_backtraces]
+
+    validate_event_keys(traces[4], @exit_kvs)
+  end
+  
+  it "should trace get_multi" do
+    Oboe::API.start_trace('memcache_test', '', {}) do
+      @mc.get_multi(['one', 'two', 'three', 'four', 'five', 'six'])
+    end
+    
+    traces = get_all_traces
+    
+    traces.count.must_equal 5
+    validate_outer_layers(traces, 'memcache_test')
+
+    validate_event_keys(traces[1], @entry_kvs)
+    
+    traces[1]['KVOp'].must_equal "get_multi"
+    
+    validate_event_keys(traces[2], @info_kvs)
+    traces[2]['KVKeyCount'].must_equal "6"
+    traces[2].has_key?('KVHitCount').must_equal true
+    traces[2].has_key?('Backtrace').must_equal Oboe::Config[:memcache][:collect_backtraces]
+
+    validate_event_keys(traces[3], @exit_kvs)
+  end
+    
+  it "should trace add for existing key" do
+    @mc.set('testKey', 'x', 1200)
+    Oboe::API.start_trace('memcache_test', '', {}) do
+      @mc.add('testKey', 'x', 1200)
+    end
+    
+    traces = get_all_traces
+    
+    traces.count.must_equal 5
+    validate_outer_layers(traces, 'memcache_test')
+
+    validate_event_keys(traces[1], @entry_kvs)
+    
+    traces[1]['KVOp'].must_equal "add"
+    traces[1].has_key?('Backtrace').must_equal Oboe::Config[:memcache][:collect_backtraces]
+    
+    validate_event_keys(traces[2], @info_kvs)
+    traces[2]['KVKey'].must_equal "testKey"
+    
+    validate_event_keys(traces[3], @exit_kvs)
+  end
+  
+  it "should trace append" do
+    @mc.set('rawKey', "Peanut Butter ", 600, :raw => true)
+    Oboe::API.start_trace('memcache_test', '', {}) do
+      @mc.append('rawKey', "Jelly")
+    end
+    
+    traces = get_all_traces
+    
+    traces.count.must_equal 5
+    validate_outer_layers(traces, 'memcache_test')
+
+    validate_event_keys(traces[1], @entry_kvs)
+    
+    traces[1]['KVOp'].must_equal "append"
+    traces[1].has_key?('Backtrace').must_equal Oboe::Config[:memcache][:collect_backtraces]
+    
+    validate_event_keys(traces[2], @info_kvs)
+    
+    traces[2]['KVKey'].must_equal "rawKey"
+    traces[2].has_key?('Backtrace').must_equal Oboe::Config[:memcache][:collect_backtraces]
+    
+    validate_event_keys(traces[3], @exit_kvs)
+  end
+  
+  it "should trace decr" do
+    Oboe::API.start_trace('memcache_test', '', {}) do
+      @mc.decr('some_key_counter', 1)
+    end
+    
+    traces = get_all_traces
+    
+    traces.count.must_equal 5
+    validate_outer_layers(traces, 'memcache_test')
+
+    validate_event_keys(traces[1], @entry_kvs)
+    
+    traces[1]['KVOp'].must_equal "decr"
+    traces[1].has_key?('Backtrace').must_equal Oboe::Config[:memcache][:collect_backtraces]
+    
+    traces[2]['KVKey'].must_equal "some_key_counter"
+    traces[2].has_key?('Backtrace').must_equal Oboe::Config[:memcache][:collect_backtraces]
+    
+    validate_event_keys(traces[3], @exit_kvs)
+  end
+
+  it "should trace increment" do
+    Oboe::API.start_trace('memcache_test', '', {}) do
+      @mc.incr("some_key_counter", 1)
+    end
+    
+    traces = get_all_traces
+    
+    traces.count.must_equal 5
+    validate_outer_layers(traces, 'memcache_test')
+    
+    validate_event_keys(traces[1], @entry_kvs)
+    traces[1]['KVOp'].must_equal "incr"
+    validate_event_keys(traces[2], @info_kvs)
+    traces[2]['KVKey'].must_equal "some_key_counter"
+    validate_event_keys(traces[3], @exit_kvs)
+  end
+
+  it "should trace replace" do
+    @mc.set("some_key", "blah")
+    Oboe::API.start_trace('memcache_test', '', {}) do
+      @mc.replace("some_key", "woop")
+    end
+    
+    traces = get_all_traces
+    
+    traces.count.must_equal 5
+    validate_outer_layers(traces, 'memcache_test')
+
+    validate_event_keys(traces[1], @entry_kvs)
+    traces[1]['KVOp'].must_equal "replace"
+    traces[2]['KVKey'].must_equal "some_key"
+    validate_event_keys(traces[2], @info_kvs)
+    validate_event_keys(traces[3], @exit_kvs)
+  end
+
+  it "should trace delete" do
+    Oboe::API.start_trace('memcache_test', '', {}) do
+      @mc.delete("some_key")
+    end
+    
+    traces = get_all_traces
+    
+    traces.count.must_equal 5
+    validate_outer_layers(traces, 'memcache_test')
+
+    validate_event_keys(traces[1], @entry_kvs)
+    traces[1]['KVOp'].must_equal "delete"
+    traces[2]['KVKey'].must_equal "some_key"
+    validate_event_keys(traces[2], @info_kvs)
+    validate_event_keys(traces[3], @exit_kvs)
+  end
+    
+  it "should obey :collect_backtraces setting when true" do
+    Oboe::Config[:memcache][:collect_backtraces] = true
+
+    Oboe::API.start_trace('memcache_test', '', {}) do
+      @mc.set('some_key', 1)
+    end
+
+    traces = get_all_traces
+    layer_has_key(traces, 'memcache', 'Backtrace')
+  end
+
+  it "should obey :collect_backtraces setting when false" do
+    Oboe::Config[:memcache][:collect_backtraces] = false
+
+    Oboe::API.start_trace('memcache_test', '', {}) do
+      @mc.set('some_key', 1)
+    end
+
+    traces = get_all_traces
+    layer_doesnt_have_key(traces, 'memcache', 'Backtrace')
+  end
+end