oboe 2.7.0.3-java

data/test/instrumentation/redis_strings_test.rb

@@ -0,0 +1,333 @@
+require 'minitest_helper'
+require "redis"
+
+describe Oboe::Inst::Redis, :strings do
+  attr_reader :entry_kvs, :exit_kvs, :redis, :redis_version
+
+  def min_server_version(version)
+    unless Gem::Version.new(@redis_version) >= Gem::Version.new(version.to_s)
+      skip "supported only on redis-server #{version} or greater"
+    end
+  end
+
+  before do
+    clear_all_traces
+
+    @redis ||= Redis.new
+
+    @redis_version ||= @redis.info["redis_version"]
+
+    # These are standard entry/exit KVs that are passed up with all moped operations
+    @entry_kvs ||= { 'Layer' => 'redis_test', 'Label' => 'entry' }
+    @exit_kvs  ||= { 'Layer' => 'redis_test', 'Label' => 'exit' }
+  end
+
+  it "should trace append" do
+    @redis.set("yourkey", "test")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.append("yourkey", "blah")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "append"
+    traces[2]['KVKey'].must_equal "yourkey"
+  end
+
+  it "should trace bitcount (>=2.6)" do
+
+    min_server_version("2.6")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.bitcount("yourkey")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "bitcount"
+    traces[2]['start'].must_equal "0"
+    traces[2]['stop'].must_equal "-1"
+  end
+
+  it "should trace bitop (>=2.6)" do
+
+    min_server_version("2.6")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.bitop("not", "bitopkey", "yourkey")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "bitop"
+    traces[2]['operation'].must_equal "not"
+    traces[2]['destkey'].must_equal "bitopkey"
+  end
+
+  it "should trace decr" do
+    @redis.setex("decr", 60, 0)
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.decr("decr")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "decr"
+    traces[2]['KVKey'].must_equal "decr"
+  end
+
+  it "should trace decrby" do
+    @redis.setex("decr", 60, 0)
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.decrby("decr", 1)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "decrby"
+    traces[2]['KVKey'].must_equal "decr"
+    traces[2]['decrement'].must_equal "1"
+  end
+
+  it "should trace get" do
+    @redis.setex("diwore", 60, "okokok")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @rv = @redis.get("diwore")
+    end
+
+    @rv.must_equal "okokok"
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "get"
+    traces[2]['KVKey'].must_equal "diwore"
+  end
+
+  it "should trace getbit" do
+    min_server_version(2.2)
+
+    @redis.setex("diwore", 60, "okokok")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.getbit("diwore", 3)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "getbit"
+    traces[2]['KVKey'].must_equal "diwore"
+    traces[2]['offset'].must_equal "3"
+  end
+
+  it "should trace getrange" do
+    min_server_version(2.2)
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.getrange("yourkey", 0, 3)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "getrange"
+    traces[2]['KVKey'].must_equal "yourkey"
+    traces[2]['start'].must_equal "0"
+    traces[2]['end'].must_equal "3"
+  end
+
+  it "should trace getset" do
+    min_server_version(2.2)
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.getset("dollar", 0)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "getset"
+    traces[2]['KVKey'].must_equal "dollar"
+    traces[2]['value'].must_equal "0"
+  end
+
+  it "should trace incr" do
+    @redis.setex("dotcom", 60, 0)
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.incr("dotcom")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "incr"
+    traces[2]['KVKey'].must_equal "dotcom"
+  end
+
+  it "should trace incrby" do
+    @redis.setex("incr", 60, 0)
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.incrby("incr", 1)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "incrby"
+    traces[2]['KVKey'].must_equal "incr"
+    traces[2]['increment'].must_equal "1"
+  end
+
+  it "should trace incrbyfloat" do
+    min_server_version(2.6)
+
+    @redis.setex("incrfloat", 60, 0.0)
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.incrbyfloat("incrfloat", 1.01)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "incrbyfloat"
+    traces[2]['KVKey'].must_equal "incrfloat"
+    traces[2]['increment'].must_equal "1.01"
+  end
+
+  it "should trace mget" do
+    @redis.setex("france", 60, "ok")
+    @redis.setex("denmark", 60, "ok")
+    @redis.setex("germany", 60, "ok")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.mget(["france", "nothing", "denmark"])
+      @redis.mget("germany")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 6
+    traces[2]['KVOp'].must_equal "mget"
+    traces[2]['KVKeyCount'].must_equal "3"
+    traces[2]['KVHitCount'].must_equal "2"
+    traces[4]['KVOp'].must_equal "mget"
+    traces[4]['KVKeyCount'].must_equal "1"
+    traces[4]['KVHitCount'].must_equal "1"
+  end
+
+  it "should trace mset" do
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.mset(["one", 1, "two", 2, "three", 3])
+      @redis.mset("one", 1)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 6
+    traces[2]['KVOp'].must_equal "mset"
+    traces[2]['KVKeyCount'].must_equal "3"
+    traces[4]['KVOp'].must_equal "mset"
+    traces[4]['KVKeyCount'].must_equal "1"
+  end
+
+  it "should trace msetnx" do
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.msetnx(["one", 1, "two", 2, "three", 3])
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "msetnx"
+  end
+
+  it "should trace psetex (>= v2.6)" do
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.psetex("one", 60, "hello")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "psetex"
+    traces[2]['KVKey'].must_equal "one"
+    traces[2]['ttl'].must_equal "60"
+  end
+
+  it "should trace basic set" do
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.set("one",   "hello")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "set"
+    traces[2]['KVKey'].must_equal "one"
+  end
+
+  it "should trace setbit" do
+    min_server_version(2.2)
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.setbit("yourkey", 3, 0)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "setbit"
+    traces[2]['KVKey'].must_equal "yourkey"
+    traces[2]['offset'].must_equal "3"
+  end
+
+  it "should trace setex" do
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.setex("one", 60, "hello")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "setex"
+    traces[2]['KVKey'].must_equal "one"
+    traces[2]['ttl'].must_equal "60"
+  end
+
+  it "should trace setnx" do
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.setnx("one", "hello")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "setnx"
+    traces[2]['KVKey'].must_equal "one"
+  end
+
+  it "should trace setrange" do
+    min_server_version(2.2)
+
+    @redis.setex("spandau_ballet", 60, "XXXXXXXXXXXXXXX")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.setrange("yourkey", 2, "ok")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "setrange"
+    traces[2]['KVKey'].must_equal "yourkey"
+    traces[2]['offset'].must_equal "2"
+  end
+
+  it "should trace strlen" do
+    min_server_version(2.2)
+
+    @redis.setex("talking_heads", 60, "burning down the house")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.strlen("talking_heads")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "strlen"
+    traces[2]['KVKey'].must_equal "talking_heads"
+  end
+end

data/test/instrumentation/resque_test.rb

@@ -0,0 +1,62 @@
+require 'minitest_helper'
+
+describe Oboe::Inst::Resque do
+  before do
+    clear_all_traces
+
+    # These are standard entry/exit KVs that are passed up with all moped operations
+    @entry_kvs = {
+      'Layer' => 'resque',
+      'Label' => 'entry' }
+
+    @exit_kvs = { 'Layer' => 'resque', 'Label' => 'exit' }
+  end
+
+  it 'Stock Resque should be loaded, defined and ready' do
+    defined?(::Resque).wont_match nil
+    defined?(::Resque::Worker).wont_match nil
+    defined?(::Resque::Job).wont_match nil
+  end
+
+  it 'Resque should have oboe methods defined' do
+    [ :enqueue, :enqueue_to, :dequeue ].each do |m|
+      ::Resque.method_defined?("#{m}_with_oboe").must_equal true
+    end
+
+    ::Resque::Worker.method_defined?("perform_with_oboe").must_equal true
+    ::Resque::Job.method_defined?("fail_with_oboe").must_equal true
+  end
+
+  it "should trace enqueue" do
+    skip
+    Oboe::API.start_trace('resque-client_test', '', {}) do
+      Resque.enqueue(OboeResqueJob, { :generate => :activerecord, :delay => rand(5..30).to_f })
+      Resque.enqueue(OboeResqueJobThatFails)
+      Resque.dequeue(OboeResqueJob, { :generate => :moped })
+    end
+
+    traces = get_all_traces
+
+    traces.count.must_equal 4
+    validate_outer_layers(traces, 'resque-client_test')
+
+    validate_event_keys(traces[1], @entry_kvs)
+    validate_event_keys(traces[2], @exit_kvs)
+  end
+
+  it "should trace dequeue" do
+    skip
+    Oboe::API.start_trace('resque-client_test', '', {}) do
+      Resque.dequeue(OboeResqueJob, { :generate => :moped })
+    end
+
+    traces = get_all_traces
+
+    traces.count.must_equal 4
+    validate_outer_layers(traces, 'resque-client_test')
+
+    validate_event_keys(traces[1], @entry_kvs)
+    validate_event_keys(traces[2], @exit_kvs)
+  end
+end
+

data/test/minitest_helper.rb

@@ -0,0 +1,148 @@
+require "minitest/spec"
+require "minitest/autorun"
+require "minitest/reporters"
+
+ENV["RACK_ENV"] = "test"
+ENV["OBOE_GEM_TEST"] = "true"
+
+# FIXME: Temp hack to fix padrino-core calling RUBY_ENGINE when it's
+# not defined under Ruby 1.8.7 and 1.9.3
+RUBY_ENGINE = "ruby" unless defined?(RUBY_ENGINE)
+
+Minitest::Spec.new 'pry'
+
+unless RUBY_VERSION =~ /^1.8/
+  MiniTest::Reporters.use! MiniTest::Reporters::SpecReporter.new
+end
+
+if defined?(JRUBY_VERSION)
+  ENV['JAVA_OPTS'] = "-J-javaagent:/usr/local/tracelytics/tracelyticsagent.jar"
+end
+
+require 'rubygems'
+require 'bundler'
+
+# Preload memcache-client
+require 'memcache'
+
+Bundler.require(:default, :test)
+
+@trace_dir = "/tmp/"
+$trace_file = @trace_dir + "trace_output.bson"
+
+# Configure Oboe
+Oboe::Config[:verbose] = true
+Oboe::Config[:tracing_mode] = "always"
+Oboe::Config[:sample_rate] = 1000000
+Oboe::Ruby.initialize
+Oboe.logger.level = Logger::DEBUG
+
+##
+# clear_all_traces
+#
+# Truncates the trace output file to zero
+#
+def clear_all_traces
+  Oboe::Reporter.clear_all_traces
+end
+
+##
+# get_all_traces
+#
+# Retrieves all traces written to the trace file
+#
+def get_all_traces
+  Oboe::Reporter.get_all_traces
+end
+
+##
+# validate_outer_layers
+#
+# Validates that the KVs in kvs are present
+# in event
+#
+def validate_outer_layers(traces, layer)
+  traces.first['Layer'].must_equal layer
+  traces.first['Label'].must_equal 'entry'
+  traces.last['Layer'].must_equal layer
+  traces.last['Label'].must_equal 'exit'
+end
+
+##
+# validate_event_keys
+#
+# Validates that the KVs in kvs are present
+# in event
+#
+def validate_event_keys(event, kvs)
+  kvs.each do |k, v|
+    event.has_key?(k).must_equal true
+    event[k].must_equal v
+  end
+end
+
+##
+# layer_has_key
+#
+# Checks an array of trace events if a specific layer (regardless of event type)
+# has he specified key
+#
+def layer_has_key(traces, layer, key)
+  return false if traces.empty?
+  has_key = false
+
+  traces.each do |t|
+    if t["Layer"] == layer and t.has_key?(key)
+      has_key = true
+
+      (t["Backtrace"].length > 0).must_equal true
+    end
+  end
+
+  has_key.must_equal true
+end
+
+##
+# layer_doesnt_have_key
+#
+# Checks an array of trace events to assure that a specific layer
+# (regardless of event type) doesn't have the specified key
+#
+def layer_doesnt_have_key(traces, layer, key)
+  return false if traces.empty?
+  has_key = false
+
+  traces.each do |t|
+    has_key = true if t["Layer"] == layer and t.has_key?(key)
+  end
+
+  has_key.must_equal false
+end
+
+##
+# Sinatra and Padrino Related Helpers
+#
+# Taken from padrino-core gem
+#
+
+class Sinatra::Base
+  # Allow assertions in request context
+  include MiniTest::Assertions
+end
+
+
+class MiniTest::Spec
+  include Rack::Test::Methods
+
+  # Sets up a Sinatra::Base subclass defined with the block
+  # given. Used in setup or individual spec methods to establish
+  # the application.
+  def mock_app(base=Padrino::Application, &block)
+    @app = Sinatra.new(base, &block)
+  end
+
+  def app
+    Rack::Lint.new(@app)
+  end
+end
+