oboe 2.4.0.1 → 2.5.0.7

data/test/instrumentation/redis_sets_test.rb

@@ -0,0 +1,293 @@
+require 'minitest_helper'
+require "redis"
+    
+describe Oboe::Inst::Redis, :sets 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 sadd" do
+    min_server_version(1.0)
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.sadd("shrimp", "fried")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "sadd"
+    traces[2]['KVKey'].must_equal "shrimp"
+  end
+  
+  it "should trace scard" do
+    min_server_version(1.0)
+      
+    @redis.sadd("mother sauces", "bechamel")
+    @redis.sadd("mother sauces", "veloute")
+    @redis.sadd("mother sauces", "espagnole")
+    @redis.sadd("mother sauces", "hollandaise")
+    @redis.sadd("mother sauces", "classic tomate")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.scard("mother sauces")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "scard"
+    traces[2]['KVKey'].must_equal "mother sauces"
+  end
+  
+  it "should trace sdiff" do
+    min_server_version(1.0)
+
+    @redis.sadd("abc", "a")
+    @redis.sadd("abc", "b")
+    @redis.sadd("abc", "c")
+    @redis.sadd("ab", "a")
+    @redis.sadd("ab", "b")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.sdiff("abc", "ab")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "sdiff"
+    traces[2].has_key?('KVKey').must_equal false
+  end
+  
+  it "should trace sdiffstore" do
+    min_server_version(1.0)
+
+    @redis.sadd("abc", "a")
+    @redis.sadd("abc", "b")
+    @redis.sadd("abc", "c")
+    @redis.sadd("ab", "a")
+    @redis.sadd("ab", "b")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.sdiffstore("dest", "abc", "ab")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "sdiffstore"
+    traces[2]['destination'].must_equal "dest"
+  end
+  
+  it "should trace sinter" do
+    min_server_version(1.0)
+
+    @redis.sadd("abc", "a")
+    @redis.sadd("abc", "b")
+    @redis.sadd("abc", "c")
+    @redis.sadd("ab", "a")
+    @redis.sadd("ab", "b")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.sinter("abc", "ab")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "sinter"
+    traces[2].has_key?('KVKey').must_equal false
+  end
+  
+  it "should trace sinterstore" do
+    min_server_version(1.0)
+
+    @redis.sadd("abc", "a")
+    @redis.sadd("abc", "b")
+    @redis.sadd("abc", "c")
+    @redis.sadd("ab", "a")
+    @redis.sadd("ab", "b")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.sinterstore("dest", "abc", "ab")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "sinterstore"
+    traces[2]['destination'].must_equal "dest"
+  end
+  
+  it "should trace sismember" do
+    min_server_version(1.0)
+
+    @redis.sadd("fibonacci", "0")
+    @redis.sadd("fibonacci", "1")
+    @redis.sadd("fibonacci", "1")
+    @redis.sadd("fibonacci", "2")
+    @redis.sadd("fibonacci", "3")
+    @redis.sadd("fibonacci", "5")
+    @redis.sadd("fibonacci", "8")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.sismember("fibonacci", "5")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "sismember"
+    traces[2]['KVKey'].must_equal "fibonacci"
+  end
+  
+  it "should trace smembers" do
+    min_server_version(1.0)
+
+    @redis.sadd("fibonacci", "0")
+    @redis.sadd("fibonacci", "1")
+    @redis.sadd("fibonacci", "1")
+    @redis.sadd("fibonacci", "2")
+    @redis.sadd("fibonacci", "3")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.smembers("fibonacci")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "smembers"
+    traces[2]['KVKey'].must_equal "fibonacci"
+  end
+  
+  it "should trace smove" do
+    min_server_version(1.0)
+
+    @redis.sadd("numbers", "1")
+    @redis.sadd("numbers", "2")
+    @redis.sadd("alpha", "two")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.smove("alpha", "numbers", "two")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "smove"
+    traces[2]['source'].must_equal "alpha"
+    traces[2]['destination'].must_equal "numbers"
+  end
+  
+  it "should trace spop" do
+    min_server_version(1.0)
+
+    @redis.sadd("fibonacci", "0")
+    @redis.sadd("fibonacci", "1")
+    @redis.sadd("fibonacci", "1")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.spop("fibonacci")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "spop"
+    traces[2]['KVKey'].must_equal "fibonacci"
+  end
+  
+  it "should trace srandmember" do
+    min_server_version(1.0)
+
+    @redis.sadd("fibonacci", "0")
+    @redis.sadd("fibonacci", "1")
+    @redis.sadd("fibonacci", "1")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.srandmember("fibonacci")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "srandmember"
+    traces[2]['KVKey'].must_equal "fibonacci"
+  end
+  
+  it "should trace srem" do
+    min_server_version(1.0)
+
+    @redis.sadd("fibonacci", "0")
+    @redis.sadd("fibonacci", "1")
+    @redis.sadd("fibonacci", "1")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.srem("fibonacci", "0")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "srem"
+    traces[2]['KVKey'].must_equal "fibonacci"
+  end
+  
+  it "should trace sunion" do
+    min_server_version(1.0)
+
+    @redis.sadd("group1", "moe")
+    @redis.sadd("group1", "curly")
+    @redis.sadd("group2", "larry")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.sunion("group1", "group2")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "sunion"
+    traces[2].has_key?('KVKey').must_equal false
+  end
+  
+  it "should trace sunionstore" do
+    min_server_version(1.0)
+
+    @redis.sadd("group1", "moe")
+    @redis.sadd("group1", "curly")
+    @redis.sadd("group2", "larry")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.sunionstore("dest", "group1", "group2")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "sunionstore"
+    traces[2]['destination'].must_equal "dest"
+    traces[2].has_key?('KVKey').must_equal false
+  end
+  
+  it "should trace sscan" do
+    min_server_version(2.8)
+
+    @redis.sadd("group1", "moe")
+    @redis.sadd("group1", "curly")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.sscan("group1", 1)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "sscan"
+    traces[2]['KVKey'].must_equal "group1"
+  end
+end
+

data/test/instrumentation/redis_sortedsets_test.rb

@@ -0,0 +1,325 @@
+require 'minitest_helper'
+require "redis"
+    
+describe Oboe::Inst::Redis, :sortedsets 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 zadd" do
+    min_server_version(1.2)
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zadd("time", 0, "past")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zadd"
+    traces[2]['KVKey'].must_equal "time"
+  end
+  
+  it "should trace zcard" do
+    min_server_version(1.2)
+      
+    @redis.zadd("sauce", 0, "bechamel")
+    @redis.zadd("sauce", 1, "veloute")
+    @redis.zadd("sauce", 2, "espagnole")
+    @redis.zadd("sauce", 3, "hollandaise")
+    @redis.zadd("sauce", 4, "classic tomate")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zcard("sauce")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zcard"
+    traces[2]['KVKey'].must_equal "sauce"
+  end
+  
+  it "should trace zcount" do
+    min_server_version(2.0)
+      
+    @redis.zadd("sauce", 0, "bechamel")
+    @redis.zadd("sauce", 1, "veloute")
+    @redis.zadd("sauce", 2, "espagnole")
+    @redis.zadd("sauce", 3, "hollandaise")
+    @redis.zadd("sauce", 4, "classic tomate")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zcount("sauce", 1, 3)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zcount"
+    traces[2]['KVKey'].must_equal "sauce"
+  end
+  
+  it "should trace zincrby" do
+    min_server_version(1.2)
+      
+    @redis.zadd("sauce", 0, "bechamel")
+    @redis.zadd("sauce", 1, "veloute")
+    @redis.zadd("sauce", 2, "espagnole")
+    @redis.zadd("sauce", 3, "hollandaise")
+    @redis.zadd("sauce", 4, "classic tomate")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zincrby("sauce", 1, "veloute")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zincrby"
+    traces[2]['KVKey'].must_equal "sauce"
+  end
+  
+  it "should trace zinterstore" do
+    min_server_version(2.0)
+      
+    @redis.zadd("sauce", 0, "bechamel")
+    @redis.zadd("sauce", 1, "veloute")
+    @redis.zadd("sauce", 2, "espagnole")
+    
+    @redis.zadd("beverage", 0, "milkshake")
+    @redis.zadd("beverage", 1, "soda")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zinterstore("zinterstore_dest", [ "sauce", "beverage" ], :weights => [2, 3])
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zinterstore"
+    traces[2]['destination'].must_equal "zinterstore_dest"
+    traces[2].has_key?('KVKey').must_equal false
+  end
+  
+  it "should trace zrange" do
+    min_server_version(1.2)
+      
+    @redis.zadd("sauce", 0, "bechamel")
+    @redis.zadd("sauce", 1, "veloute")
+    @redis.zadd("sauce", 2, "espagnole")
+    @redis.zadd("sauce", 3, "hollandaise")
+    @redis.zadd("sauce", 4, "classic tomate")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zrange("sauce", 1, 3)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zrange"
+    traces[2]['KVKey'].must_equal "sauce"
+  end
+
+  it "should trace zrangebyscore" do
+    min_server_version(1.0)
+      
+    @redis.zadd("sauce", 0, "bechamel")
+    @redis.zadd("sauce", 1, "veloute")
+    @redis.zadd("sauce", 2, "espagnole")
+    @redis.zadd("sauce", 3, "hollandaise")
+    @redis.zadd("sauce", 4, "classic tomate")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zrangebyscore("sauce", "5", "(100")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zrangebyscore"
+    traces[2]['KVKey'].must_equal "sauce"
+  end
+  
+  it "should trace zrank" do
+    min_server_version(2.0)
+      
+    @redis.zadd("sauce", 0, "bechamel")
+    @redis.zadd("sauce", 1, "veloute")
+    @redis.zadd("sauce", 2, "espagnole")
+    @redis.zadd("sauce", 3, "hollandaise")
+    @redis.zadd("sauce", 4, "classic tomate")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zrank("sauce", "veloute")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zrank"
+    traces[2]['KVKey'].must_equal "sauce"
+  end
+  
+  it "should trace zrem" do
+    min_server_version(1.2)
+      
+    @redis.zadd("sauce", 0, "bechamel")
+    @redis.zadd("sauce", 1, "veloute")
+    @redis.zadd("sauce", 2, "espagnole")
+    @redis.zadd("sauce", 3, "hollandaise")
+    @redis.zadd("sauce", 4, "classic tomate")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zrem("sauce", "veloute")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zrem"
+    traces[2]['KVKey'].must_equal "sauce"
+  end
+
+  it "should trace zremrangebyrank" do
+    min_server_version(2.0)
+      
+    @redis.zadd("sauce", 0, "bechamel")
+    @redis.zadd("sauce", 1, "veloute")
+    @redis.zadd("sauce", 2, "espagnole")
+    @redis.zadd("sauce", 3, "hollandaise")
+    @redis.zadd("sauce", 4, "classic tomate")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zremrangebyrank("sauce", -5, -1)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zremrangebyrank"
+    traces[2]['KVKey'].must_equal "sauce"
+    traces[2]['start'].must_equal "-5"
+    traces[2]['stop'].must_equal "-1"
+  end
+
+  it "should trace zremrangebyscore" do
+    min_server_version(1.2)
+      
+    @redis.zadd("sauce", 0, "bechamel")
+    @redis.zadd("sauce", 1, "veloute")
+    @redis.zadd("sauce", 2, "espagnole")
+    @redis.zadd("sauce", 3, "hollandaise")
+    @redis.zadd("sauce", 4, "classic tomate")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zremrangebyscore("sauce", -5, -1)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zremrangebyscore"
+    traces[2]['KVKey'].must_equal "sauce"
+  end
+
+  it "should trace zrevrange" do
+    min_server_version(1.2)
+      
+    @redis.zadd("sauce", 0, "bechamel")
+    @redis.zadd("sauce", 1, "veloute")
+    @redis.zadd("sauce", 2, "espagnole")
+    @redis.zadd("sauce", 3, "hollandaise")
+    @redis.zadd("sauce", 4, "classic tomate")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zrevrange("sauce", 0, -1)
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zrevrange"
+    traces[2]['KVKey'].must_equal "sauce"
+    traces[2]['start'].must_equal "0"
+    traces[2]['stop'].must_equal "-1"
+  end
+
+  it "should trace zrevrangebyscore" do
+    min_server_version(1.2)
+      
+    @redis.zadd("sauce", 0, "bechamel")
+    @redis.zadd("sauce", 1, "veloute")
+    @redis.zadd("sauce", 2, "espagnole")
+    @redis.zadd("sauce", 3, "hollandaise")
+    @redis.zadd("sauce", 4, "classic tomate")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zrevrangebyscore("sauce", "(100", "5")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zrevrangebyscore"
+    traces[2]['KVKey'].must_equal "sauce"
+  end
+
+  it "should trace zrevrank" do
+    min_server_version(2.0)
+
+    @redis.zadd("letters", 0, "a")
+    @redis.zadd("letters", 1, "b")
+    @redis.zadd("letters", 1, "c")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zrevrank("letters", "c")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zrevrank"
+    traces[2]['KVKey'].must_equal "letters"
+  end
+  
+  it "should trace zscore" do
+    min_server_version(1.2)
+
+    @redis.zadd("elements", 0, "fire")
+    @redis.zadd("elements", 1, "water")
+    @redis.zadd("elements", 1, "earth")
+    @redis.zadd("elements", 1, "air")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zscore("elements", "earth")
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zscore"
+    traces[2]['KVKey'].must_equal "elements"
+  end
+
+  it "should trace zunionstore" do
+    min_server_version(1.0)
+
+    @redis.zadd("colors", 0, "blueish")
+    @redis.zadd("colors", 1, "yellowish")
+    @redis.zadd("codes", 0, "0xff")
+
+    Oboe::API.start_trace('redis_test', '', {}) do
+      @redis.zunionstore("zdest", ["colors", "codes"])
+    end
+
+    traces = get_all_traces
+    traces.count.must_equal 4
+    traces[2]['KVOp'].must_equal "zunionstore"
+    traces[2]['destination'].must_equal "zdest"
+    traces[2].has_key?('KVKey').must_equal false
+  end
+end
+