traceview 3.0.2-java → 3.0.3-java

data/test/instrumentation/redis_keys_test.rb

@@ -1,318 +1,321 @@
+# Copyright (c) 2015 AppNeta, Inc.
+# All rights reserved.
+
 require 'minitest_helper'
-require "redis"
 
-describe "Redis Keys" do
-  attr_reader :entry_kvs, :exit_kvs, :redis, :redis_version
+if defined?(::Redis)
+  describe "Redis Keys" 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"
+    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
-  end
 
-  before do
-    clear_all_traces
+    before do
+      clear_all_traces
 
-    @redis ||= Redis.new
+      @redis ||= Redis.new
 
-    @redis_version ||= @redis.info["redis_version"]
+      @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
+      # 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 'Stock Redis should be loaded, defined and ready' do
-    defined?(::Redis).wont_match nil
-  end
+    it 'Stock Redis should be loaded, defined and ready' do
+      defined?(::Redis).wont_match nil
+    end
 
-  it "should trace del" do
-    @redis.setex("del_test", 60, "blah")
+    it "should trace del" do
+      @redis.setex("del_test", 60, "blah")
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.del("del_test")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.del("del_test")
+      end
+
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "del"
+      traces[2]['KVKey'].must_equal "del_test"
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "del"
-    traces[2]['KVKey'].must_equal "del_test"
-  end
+    it "should trace del of multiple keys" do
+      @redis.setex("del_test", 60, "blah")
 
-  it "should trace del of multiple keys" do
-    @redis.setex("del_test", 60, "blah")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.del(["del_test", "noexist", "maybe"])
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.del(["del_test", "noexist", "maybe"])
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "del"
+      traces[2].has_key?('KVKey').must_equal false
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "del"
-    traces[2].has_key?('KVKey').must_equal false
-  end
+    it "should trace dump" do
+      min_server_version(2.6)
 
-  it "should trace dump" do
-    min_server_version(2.6)
+      @redis.setex("dump_test", 60, "blah")
 
-    @redis.setex("dump_test", 60, "blah")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.dump("del_test")
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.dump("del_test")
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "dump"
+      traces[2]['KVKey'].must_equal "del_test"
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "dump"
-    traces[2]['KVKey'].must_equal "del_test"
-  end
+    it "should trace exists" do
+      @redis.setex("talking_heads", 60, "burning down the house")
 
-  it "should trace exists" do
-    @redis.setex("talking_heads", 60, "burning down the house")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @it_exists = @redis.exists("talking_heads")
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @it_exists = @redis.exists("talking_heads")
-    end
+      @it_exists.must_equal true
 
-    @it_exists.must_equal true
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "exists"
+      traces[2]['KVKey'].must_equal "talking_heads"
+    end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "exists"
-    traces[2]['KVKey'].must_equal "talking_heads"
-  end
+    it "should trace expire" do
+      @redis.set("expire_please", "burning down the house")
 
-  it "should trace expire" do
-    @redis.set("expire_please", "burning down the house")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.expire("expire_please", 120)
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.expire("expire_please", 120)
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "expire"
+      traces[2]['KVKey'].must_equal "expire_please"
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "expire"
-    traces[2]['KVKey'].must_equal "expire_please"
-  end
+    it "should trace expireat" do
+      @redis.set("expireat_please", "burning down the house")
 
-  it "should trace expireat" do
-    @redis.set("expireat_please", "burning down the house")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.expireat("expireat_please", Time.now.to_i)
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.expireat("expireat_please", Time.now.to_i)
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "expireat"
+      traces[2]['KVKey'].must_equal "expireat_please"
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "expireat"
-    traces[2]['KVKey'].must_equal "expireat_please"
-  end
+    it "should trace keys" do
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.keys("del*")
+      end
 
-  it "should trace keys" do
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.keys("del*")
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "keys"
+      traces[2]['pattern'].must_equal "del*"
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "keys"
-    traces[2]['pattern'].must_equal "del*"
-  end
+    it "should trace basic move" do
+      @redis.set("piano", Time.now)
 
-  it "should trace basic move" do
-    @redis.set("piano", Time.now)
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.move("piano", 1)
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.move("piano", 1)
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "move"
+      traces[2]['KVKey'].must_equal "piano"
+      traces[2]['db'].must_equal 1
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "move"
-    traces[2]['KVKey'].must_equal "piano"
-    traces[2]['db'].must_equal 1
-  end
+    it "should trace persist" do
+      min_server_version(2.2)
 
-  it "should trace persist" do
-    min_server_version(2.2)
+      @redis.setex("mine", 60, "blah")
 
-    @redis.setex("mine", 60, "blah")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.persist("mine")
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.persist("mine")
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "persist"
+      traces[2]['KVKey'].must_equal "mine"
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "persist"
-    traces[2]['KVKey'].must_equal "mine"
-  end
+    it "should trace pexpire" do
+      min_server_version(2.6)
 
-  it "should trace pexpire" do
-    min_server_version(2.6)
+      @redis.set("sand", "blah")
 
-    @redis.set("sand", "blah")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @rv = @redis.pexpire("sand", 8000)
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @rv = @redis.pexpire("sand", 8000)
+      @rv.must_equal true
+
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "pexpire"
+      traces[2]['KVKey'].must_equal "sand"
+      traces[2]['milliseconds'].must_equal 8000
     end
 
-    @rv.must_equal true
+    it "should trace pexpireat" do
+      min_server_version(2.6)
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "pexpire"
-    traces[2]['KVKey'].must_equal "sand"
-    traces[2]['milliseconds'].must_equal 8000
-  end
+      @redis.set("sand", "blah")
 
-  it "should trace pexpireat" do
-    min_server_version(2.6)
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @rv = @redis.pexpireat("sand", 8000)
+      end
 
-    @redis.set("sand", "blah")
+      @rv.must_equal true
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @rv = @redis.pexpireat("sand", 8000)
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "pexpireat"
+      traces[2]['KVKey'].must_equal "sand"
+      traces[2]['milliseconds'].must_equal 8000
     end
 
-    @rv.must_equal true
+    it "should trace pttl" do
+      min_server_version(2.6)
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "pexpireat"
-    traces[2]['KVKey'].must_equal "sand"
-    traces[2]['milliseconds'].must_equal 8000
-  end
-
-  it "should trace pttl" do
-    min_server_version(2.6)
+      @redis.setex("sand", 120, "blah")
 
-    @redis.setex("sand", 120, "blah")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.pttl("sand")
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.pttl("sand")
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "pttl"
+      traces[2]['KVKey'].must_equal "sand"
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "pttl"
-    traces[2]['KVKey'].must_equal "sand"
-  end
+    it "should trace randomkey" do
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.randomkey()
+      end
 
-  it "should trace randomkey" do
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.randomkey()
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "randomkey"
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "randomkey"
-  end
+    it "should trace rename" do
+      @redis.setex("sand", 120, "blah")
 
-  it "should trace rename" do
-    @redis.setex("sand", 120, "blah")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.rename("sand", "sandy")
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.rename("sand", "sandy")
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "rename"
+      traces[2]['KVKey'].must_equal "sand"
+      traces[2]['newkey'].must_equal "sandy"
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "rename"
-    traces[2]['KVKey'].must_equal "sand"
-    traces[2]['newkey'].must_equal "sandy"
-  end
+    it "should trace renamenx" do
+      @redis.setex("sand", 120, "blah")
 
-  it "should trace renamenx" do
-    @redis.setex("sand", 120, "blah")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.renamenx("sand", "sandy")
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.renamenx("sand", "sandy")
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "renamenx"
+      traces[2]['KVKey'].must_equal "sand"
+      traces[2]['newkey'].must_equal "sandy"
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "renamenx"
-    traces[2]['KVKey'].must_equal "sand"
-    traces[2]['newkey'].must_equal "sandy"
-  end
+    it "should trace restore" do
+      min_server_version(2.6)
 
-  it "should trace restore" do
-    min_server_version(2.6)
+      @redis.setex("qubit", 60, "zero")
+      x = @redis.dump("qubit")
+      @redis.del "blue"
 
-    @redis.setex("qubit", 60, "zero")
-    x = @redis.dump("qubit")
-    @redis.del "blue"
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.restore("blue", 0, x)
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.restore("blue", 0, x)
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "restore"
+      traces[2]['KVKey'].must_equal "blue"
+      traces[2]['ttl'].must_equal 0
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "restore"
-    traces[2]['KVKey'].must_equal "blue"
-    traces[2]['ttl'].must_equal 0
-  end
+    it "should trace sort" do
+      min_server_version(2.2)
 
-  it "should trace sort" do
-    min_server_version(2.2)
+      @redis.rpush("penguin", "one")
+      @redis.rpush("penguin", "two")
+      @redis.rpush("penguin", "three")
+      @redis.rpush("penguin", "four")
 
-    @redis.rpush("penguin", "one")
-    @redis.rpush("penguin", "two")
-    @redis.rpush("penguin", "three")
-    @redis.rpush("penguin", "four")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.sort("penguin", :order => "desc alpha", :store => "target")
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.sort("penguin", :order => "desc alpha", :store => "target")
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "sort"
+      traces[2]['KVKey'].must_equal "penguin"
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "sort"
-    traces[2]['KVKey'].must_equal "penguin"
-  end
+    it "should trace ttl" do
+      min_server_version(2.6)
 
-  it "should trace ttl" do
-    min_server_version(2.6)
+      @redis.setex("sand", 120, "blah")
 
-    @redis.setex("sand", 120, "blah")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.ttl("sand")
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.ttl("sand")
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "ttl"
+      traces[2]['KVKey'].must_equal "sand"
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "ttl"
-    traces[2]['KVKey'].must_equal "sand"
-  end
+    it "should trace type" do
+      min_server_version(2.6)
 
-  it "should trace type" do
-    min_server_version(2.6)
+      @redis.setex("sand", 120, "blah")
 
-    @redis.setex("sand", 120, "blah")
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.type("sand")
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.type("sand")
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "type"
+      traces[2]['KVKey'].must_equal "sand"
     end
 
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "type"
-    traces[2]['KVKey'].must_equal "sand"
-  end
+    it "should trace scan" do
+      min_server_version(2.8)
 
-  it "should trace scan" do
-    min_server_version(2.8)
+      TraceView::API.start_trace('redis_test', '', {}) do
+        @redis.scan(0)
+      end
 
-    TraceView::API.start_trace('redis_test', '', {}) do
-      @redis.scan(0)
+      traces = get_all_traces
+      traces.count.must_equal 4
+      traces[2]['KVOp'].must_equal "scan"
     end
-
-    traces = get_all_traces
-    traces.count.must_equal 4
-    traces[2]['KVOp'].must_equal "scan"
   end
 end
-