traceview 3.7.0-java → 3.7.1-java

data/test/instrumentation/{mongo_test.rb → mongo_v1_test.rb}

@@ -7,7 +7,7 @@ unless ENV['TV_MONGO_SERVER']
   ENV['TV_MONGO_SERVER'] = "127.0.0.1:27017"
 end
 
-if defined?(::BSON::VERSION) and (BSON::VERSION < "2.0")
+if Gem.loaded_specs['mongo'].version.to_s < '2.0.0'
   describe "Mongo" do
     before do
       clear_all_traces
@@ -45,6 +45,12 @@ if defined?(::BSON::VERSION) and (BSON::VERSION < "2.0")
       defined?(::Mongo::Collection).wont_match nil
     end
 
+    it 'reports mongo gem version in init' do
+      init_kvs = ::TraceView::Util.build_init_report
+      assert init_kvs.key?('Ruby.mongo.Version')
+      assert_equal Gem.loaded_specs['mongo'].version.to_s, init_kvs['Ruby.mongo.Version']
+    end
+
     it 'Mongo should have traceview methods defined' do
       TraceView::Inst::Mongo::DB_OPS.each do |m|
         ::Mongo::DB.method_defined?("#{m}_with_traceview").must_equal true
@@ -181,7 +187,7 @@ if defined?(::BSON::VERSION) and (BSON::VERSION < "2.0")
       traces[1]['QueryOp'].must_equal "map_reduce"
       traces[1]['Map_Function'].must_equal "function() { emit(this.name, 1); }"
       traces[1]['Reduce_Function'].must_equal "function(k, vals) { var sum = 0; for(var i in vals) sum += vals[i]; return sum; }"
-      traces[1]['Limit'].must_equal "100"
+      traces[1]['Limit'].must_equal 100
     end
 
     it "should trace remove" do
@@ -309,7 +315,7 @@ if defined?(::BSON::VERSION) and (BSON::VERSION < "2.0")
       traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
       traces[1]['QueryOp'].must_equal "find"
       traces[1].has_key?('Query').must_equal true
-      traces[1]['Limit'].must_equal "1"
+      traces[1]['Limit'].must_equal 1
     end
 
     it "should trace find (with block)" do

data/test/instrumentation/mongo_v2_index_test.rb

@@ -0,0 +1,124 @@
+# Copyright (c) 2015 AppNeta, Inc.
+# All rights reserved.
+
+require 'minitest_helper'
+
+unless ENV['TV_MONGO_SERVER']
+  ENV['TV_MONGO_SERVER'] = "127.0.0.1:27017"
+end
+
+if defined?(::Mongo::VERSION) && Mongo::VERSION >= '2.0.0'
+  describe "MongoIndex" do
+    before do
+      clear_all_traces
+
+      @client = Mongo::Client.new([ ENV['TV_MONGO_SERVER'] ], :database => "traceview-#{ENV['RACK_ENV']}")
+      if Mongo::VERSION < '2.2'
+        Mongo::Logger.logger.level = Logger::INFO
+      else
+        @client.logger.level = Logger::INFO
+      end
+      @db = @client.database
+
+      @collections = @db.collection_names
+      @testCollection = @client[:test_collection]
+      @testCollection.create unless @collections.include? "test_collection"
+
+      # These are standard entry/exit KVs that are passed up with all mongo operations
+      @entry_kvs = {
+        'Layer' => 'mongo',
+        'Label' => 'entry',
+        'Flavor' => 'mongodb',
+        'Database' => 'traceview-test',
+        'RemoteHost' => ENV['TV_MONGO_SERVER'] }
+
+      @exit_kvs = { 'Layer' => 'mongo', 'Label' => 'exit' }
+      @collect_backtraces = TraceView::Config[:mongo][:collect_backtraces]
+    end
+
+    after do
+      TraceView::Config[:mongo][:collect_backtraces] = @collect_backtraces
+    end
+
+    it "should trace create_one" do
+      coll = @db[:test_collection]
+      coll.indexes.drop_all
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        coll.indexes.create_one({ :name => 1 })
+      end
+
+      traces = get_all_traces
+      traces.count.must_equal 4
+
+      validate_outer_layers(traces, 'mongo_test')
+      validate_event_keys(traces[1], @entry_kvs)
+      validate_event_keys(traces[2], @exit_kvs)
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "create_one"
+    end
+
+    it "should trace create_many" do
+      coll = @db[:test_collection]
+      coll.indexes.drop_all
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        coll.indexes.create_many([ { :key => {:asdf => 1}, :unique => false },
+                                   { :key => {:age => -1}, :background => true} ])
+      end
+
+      traces = get_all_traces
+      traces.count.must_equal 4
+
+      validate_outer_layers(traces, 'mongo_test')
+      validate_event_keys(traces[1], @entry_kvs)
+      validate_event_keys(traces[2], @exit_kvs)
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "create_many"
+    end
+
+    it "should trace drop_one" do
+      coll = @db[:test_collection]
+      coll.indexes.create_one({ :name => 1 })
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        coll.indexes.drop_one('name_1')
+      end
+
+      traces = get_all_traces
+      traces.count.must_equal 4
+
+      validate_outer_layers(traces, 'mongo_test')
+      validate_event_keys(traces[1], @entry_kvs)
+      validate_event_keys(traces[2], @exit_kvs)
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "drop_one"
+    end
+
+    it "should trace drop_all" do
+      coll = @db[:test_collection]
+      coll.indexes.create_one({ :name => 1 })
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        coll.indexes.drop_all
+      end
+
+      traces = get_all_traces
+      traces.count.must_equal 4
+
+      validate_outer_layers(traces, 'mongo_test')
+      validate_event_keys(traces[1], @entry_kvs)
+      validate_event_keys(traces[2], @exit_kvs)
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "drop_all"
+    end
+  end
+end