traceview 3.7.0-java → 3.7.1-java

data/test/instrumentation/mongo_v2_test.rb

@@ -0,0 +1,585 @@
+# 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 "MongoCollection" do
+    before do
+      clear_all_traces
+
+      @client = Mongo::Client.new([ ENV['TV_MONGO_SERVER'] ], :database => "traceview-#{ENV['RACK_ENV']}")
+      @db = @client.database
+
+      if Mongo::VERSION < '2.2'
+        Mongo::Logger.logger.level = Logger::INFO
+      else
+        @client.logger.level = Logger::INFO
+      end
+
+      @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
+
+      if @db.collection_names.include?("temp_collection")
+        @db[:temp_collection].drop
+      end
+    end
+
+    it "should trace collection creation" do
+      r = nil
+      collection = @db[:temp_collection]
+      TraceView::API.start_trace('mongo_test', nil, {}) do
+        r = collection.create
+      end
+
+      traces = get_all_traces
+      traces.count.must_equal 4
+
+      r.must_be_instance_of ::Mongo::Operation::Result
+      if Mongo::VERSION < '2.2'
+        r.successful?.must_equal true
+      else
+        r.ok?.must_equal true
+      end
+
+      validate_outer_layers(traces, 'mongo_test')
+      validate_event_keys(traces[1], @entry_kvs)
+      validate_event_keys(traces[2], @exit_kvs)
+
+      traces[1]['QueryOp'].must_equal "create"
+      traces[1]['New_Collection_Name'].must_equal "temp_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+    end
+
+    it "should trace drop_collection" do
+      r = nil
+      collection = @db[:deleteme_collection]
+
+      # Create something to drop unless it already exists
+      unless @db.collection_names.include?("deleteme_collection")
+        collection.create
+      end
+
+      TraceView::API.start_trace('mongo_test', nil, {}) do
+        r = collection.drop
+      end
+
+      traces = get_all_traces
+      traces.count.must_equal 4
+
+      r.must_be_instance_of ::Mongo::Operation::Result
+      if Mongo::VERSION < '2.2'
+        r.successful?.must_equal true
+      else
+        r.ok?.must_equal true
+      end
+
+      validate_outer_layers(traces, 'mongo_test')
+      validate_event_keys(traces[1], @entry_kvs)
+      validate_event_keys(traces[2], @exit_kvs)
+
+      traces[1]['QueryOp'].must_equal "drop"
+      traces[1]['Collection'].must_equal "deleteme_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+    end
+
+    it "should capture collection creation errors" do
+      collection = @db[:temp_collection]
+      collection.create
+
+      begin
+        TraceView::API.start_trace('mongo_test', nil, {}) do
+          collection.create
+        end
+      rescue
+      end
+
+      traces = get_all_traces
+      traces.count.must_equal 5
+
+      validate_outer_layers(traces, 'mongo_test')
+      validate_event_keys(traces[1], @entry_kvs)
+      validate_event_keys(traces[3], @exit_kvs)
+
+      traces[1]['QueryOp'].must_equal "create"
+      traces[1]['New_Collection_Name'].must_equal "temp_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+
+      traces[2]['Layer'].must_equal "mongo"
+      traces[2]['Label'].must_equal "error"
+      traces[2]['ErrorClass'].must_equal "Mongo::Error::OperationFailure"
+      traces[2]['ErrorMsg'].must_equal "collection already exists ()"
+      traces[2].has_key?('Backtrace').must_equal true
+    end
+
+    it "should trace insert_one" do
+      r = nil
+      collection = @db[:tv_collection]
+
+      TraceView::API.start_trace('mongo_test', nil, {}) do
+        r = collection.insert_one({ name => 'Rabel Lasen' })
+      end
+
+      traces = get_all_traces
+      traces.count.must_equal 4
+
+      r.must_be_instance_of Mongo::Operation::Write::Insert::Result
+      if Mongo::VERSION < '2.2'
+        r.successful?.must_equal true
+      else
+        r.ok?.must_equal true
+      end
+
+      validate_outer_layers(traces, 'mongo_test')
+      validate_event_keys(traces[1], @entry_kvs)
+      validate_event_keys(traces[2], @exit_kvs)
+
+      traces[1]['QueryOp'].must_equal "insert_one"
+      traces[1]['Collection'].must_equal "tv_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+    end
+
+    it "should trace insert_many" do
+      r = nil
+      collection = @db[:tv_collection]
+
+      TraceView::API.start_trace('mongo_test', nil, {}) do
+        r = collection.insert_many([
+          { :name => 'Rabel Lasen' },
+          { :name => 'Louval Raiden' }])
+      end
+
+      traces = get_all_traces
+      traces.count.must_equal 4
+
+      if Mongo::VERSION < '2.1'
+        r.must_be_instance_of Mongo::Operation::Write::Insert::Result
+      else
+        r.must_be_instance_of Mongo::BulkWrite::Result
+        r.inserted_count.must_equal 2
+      end
+
+      validate_outer_layers(traces, 'mongo_test')
+      validate_event_keys(traces[1], @entry_kvs)
+      validate_event_keys(traces[2], @exit_kvs)
+
+      traces[1]['QueryOp'].must_equal "insert_many"
+      traces[1]['Collection'].must_equal "tv_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+    end
+
+    it "should trace find" do
+      coll = @db[:test_collection]
+      r = nil
+
+      # Insert a doc to assure we get a result
+      doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
+      coll.insert_one(doc)
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        r = coll.find(:name => "MyName", :limit => 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)
+
+      r.must_be_instance_of Mongo::Collection::View
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "find"
+      traces[1]['Query'].must_equal "{\"name\":\"MyName\",\"limit\":1}"
+      traces[1].has_key?('Query').must_equal true
+    end
+
+    it "should trace find_one_and_delete" do
+      skip unless Mongo::VERSION >= '2.1'
+
+      coll = @db[:test_collection]
+      r = nil
+
+      # Insert a doc to assure we get a result
+      doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
+      coll.insert_one(doc)
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        r = coll.find_one_and_delete(:name => "MyName")
+      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)
+
+      r.must_be_instance_of BSON::Document
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "find_one_and_delete"
+      traces[1]['Query'].must_equal "{\"name\":\"MyName\"}"
+      traces[1].has_key?('Query').must_equal true
+    end
+
+    it "should trace find_one_and_update" do
+      skip unless Mongo::VERSION >= '2.1'
+      coll = @db[:test_collection]
+      r = nil
+
+      # Insert a doc to assure we get a result
+      doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
+      coll.insert_one(doc)
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        r = coll.find_one_and_update({ :name => 'MyName' }, { "$set" => { :name => 'test1' }}, :return_document => :after)
+      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)
+
+      r.must_be_instance_of BSON::Document
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "find_one_and_update"
+      traces[1]['Query'].must_equal "{\"name\":\"MyName\"}"
+      traces[1].has_key?('Query').must_equal true
+    end
+
+    it "should trace find_one_and_replace" do
+      skip unless Mongo::VERSION >= '2.1'
+      coll = @db[:test_collection]
+      r = nil
+
+      # Insert a doc to assure we get a result
+      doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
+      coll.insert_one(doc)
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        r = coll.find_one_and_replace({ :name => 'MyName' }, { "$set" => { :name => 'test1' }}, :return_document => :after)
+      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)
+
+      r.must_be_instance_of BSON::Document
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "find_one_and_replace"
+      traces[1]['Query'].must_equal "{\"name\":\"MyName\"}"
+      traces[1].has_key?('Query').must_equal true
+    end
+
+    it "should trace update_one" do
+      skip unless Mongo::VERSION >= '2.1'
+
+      coll = @db[:test_collection]
+      r = nil
+
+      # Insert a doc to assure we get a result
+      doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
+      coll.insert_one(doc)
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        r = coll.update_one({ :name => 'MyName' }, { "$set" => { :name => 'test1' }}, :return_document => :after)
+      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)
+
+      r.class.ancestors.include?(Mongo::Operation::Result).must_equal true
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "update_one"
+      traces[1]['Query'].must_equal "{\"name\":\"MyName\"}"
+      traces[1].has_key?('Query').must_equal true
+    end
+
+    it "should trace update_many" do
+      skip unless Mongo::VERSION > '2.1'
+      coll = @db[:test_collection]
+      r = nil
+
+      # Insert a doc to assure we get a result
+      doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
+      coll.insert_one(doc)
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        r = coll.update_many({ :name => 'MyName' }, { "$set" => { :name => 'test1' }}, :return_document => :after)
+      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)
+
+      r.class.ancestors.include?(Mongo::Operation::Result).must_equal true
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "update_many"
+      traces[1]['Query'].must_equal "{\"name\":\"MyName\"}"
+      traces[1].has_key?('Query').must_equal true
+    end
+
+    it "should trace delete_one" do
+      skip unless Mongo::VERSION > '2.1'
+      coll = @db[:test_collection]
+      r = nil
+
+      # Insert a doc to assure we get a result
+      doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
+      coll.insert_one(doc)
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        r = coll.delete_one({ :name => 'MyName' })
+      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)
+
+      r.class.ancestors.include?(Mongo::Operation::Result).must_equal true
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "delete_one"
+      traces[1]['Query'].must_equal "{\"name\":\"MyName\"}"
+      traces[1].has_key?('Query').must_equal true
+    end
+
+    it "should trace delete_many" do
+      skip unless Mongo::VERSION > '2.1'
+      coll = @db[:test_collection]
+      r = nil
+
+      # Insert a doc to assure we get a result
+      doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
+      coll.insert_one(doc)
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        r = coll.delete_many({ :name => 'MyName' })
+      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)
+
+      r.class.ancestors.include?(Mongo::Operation::Result).must_equal true
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "delete_many"
+      traces[1]['Query'].must_equal "{\"name\":\"MyName\"}"
+      traces[1].has_key?('Query').must_equal true
+    end
+
+    it "should trace replace_one" do
+      skip unless Mongo::VERSION > '2.1'
+      coll = @db[:test_collection]
+      r = nil
+
+      # Insert a doc to assure we get a result
+      doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
+      coll.insert_one(doc)
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        r = coll.replace_one({ :name => 'test' }, { :name => 'test1' })
+      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)
+
+      r.class.ancestors.include?(Mongo::Operation::Result).must_equal true
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "replace_one"
+      traces[1]['Query'].must_equal "{\"name\":\"test\"}"
+      traces[1].has_key?('Query').must_equal true
+    end
+
+    it "should trace count" do
+      skip unless Mongo::VERSION > '2.1'
+      coll = @db[:test_collection]
+      r = nil
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        r = coll.count({ :name => 'MyName' })
+      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)
+
+      r.is_a?(Numeric).must_equal true
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "count"
+      traces[1]['Query'].must_equal "{\"name\":\"MyName\"}"
+      traces[1].has_key?('Query').must_equal true
+    end
+
+    it "should trace distinct" do
+      skip unless Mongo::VERSION > '2.1'
+      coll = @db[:test_collection]
+      r = nil
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        r = coll.distinct('name', { :name => 'MyName' })
+      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)
+
+      r.is_a?(Array).must_equal true
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "distinct"
+      traces[1]['Query'].must_equal "{\"name\":\"MyName\"}"
+      traces[1].has_key?('Query').must_equal true
+    end
+
+    it "should trace aggregate" do
+      skip unless Mongo::VERSION >= '2.1'
+      coll = @db[:test_collection]
+      r = nil
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        r = coll.aggregate([ { "$group" => { "_id" => "$city", "tpop" => { "$sum" => "$pop" }}} ])
+      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)
+
+      r.must_be_instance_of Mongo::Collection::View::Aggregation
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "aggregate"
+      traces[1].key?('Query').must_equal false
+    end
+
+    it "should trace bulk_write" do
+      coll = @db[:test_collection]
+      r = nil
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        r = coll.bulk_write([ { :insert_one => { :x => 1} },
+                              { :insert_one => { :x => 3} } ],
+                              :ordered => 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)
+
+      if Mongo::VERSION < '2.1'
+        r.must_be_instance_of Hash
+        r[:n_inserted].must_equal 2
+      else
+        r.must_be_instance_of Mongo::BulkWrite::Result
+      end
+
+      traces[1]['Collection'].must_equal "test_collection"
+      traces[1].has_key?('Backtrace').must_equal TraceView::Config[:mongo][:collect_backtraces]
+      traces[1]['QueryOp'].must_equal "bulk_write"
+      traces[1].key?('Query').must_equal false
+    end
+
+    it "should obey :collect_backtraces setting when true" do
+      TraceView::Config[:mongo][:collect_backtraces] = true
+
+      coll = @db[:test_collection]
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
+        coll.insert_one(doc)
+      end
+
+      traces = get_all_traces
+      layer_has_key(traces, 'mongo', 'Backtrace')
+    end
+
+    it "should obey :collect_backtraces setting when false" do
+      TraceView::Config[:mongo][:collect_backtraces] = false
+
+      coll = @db[:test_collection]
+
+      TraceView::API.start_trace('mongo_test', '', {}) do
+        doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
+        coll.insert_one(doc)
+      end
+
+      traces = get_all_traces
+      layer_doesnt_have_key(traces, 'mongo', 'Backtrace')
+    end
+  end
+end