appoptics_apm 4.0.2

Sign up to get free protection for your applications and to get access to all the features.
Files changed (226) hide show
  1. checksums.yaml +7 -0
  2. data/.codeclimate.yml +43 -0
  3. data/.dockerignore +5 -0
  4. data/.gitignore +23 -0
  5. data/.rubocop.yml +5 -0
  6. data/.travis.yml +82 -0
  7. data/CHANGELOG.md +769 -0
  8. data/CONFIG.md +33 -0
  9. data/Dockerfile +41 -0
  10. data/Dockerfile_test +66 -0
  11. data/Gemfile +41 -0
  12. data/LICENSE +193 -0
  13. data/README.md +351 -0
  14. data/Rakefile +202 -0
  15. data/Vagrantfile +67 -0
  16. data/appoptics_apm.gemspec +55 -0
  17. data/build_gems.sh +15 -0
  18. data/docker-compose.yml +73 -0
  19. data/examples/DNT.md +35 -0
  20. data/examples/carrying_context.rb +220 -0
  21. data/examples/instrumenting_metal_controller.rb +8 -0
  22. data/examples/puma_on_heroku_config.rb +17 -0
  23. data/examples/tracing_async_threads.rb +124 -0
  24. data/examples/tracing_background_jobs.rb +53 -0
  25. data/examples/tracing_forked_processes.rb +99 -0
  26. data/examples/unicorn_on_heroku_config.rb +28 -0
  27. data/ext/oboe_metal/extconf.rb +54 -0
  28. data/ext/oboe_metal/lib/.keep +0 -0
  29. data/ext/oboe_metal/lib/liboboe-1.0.so.0.0.0 +0 -0
  30. data/ext/oboe_metal/noop/noop.c +7 -0
  31. data/ext/oboe_metal/src/VERSION +1 -0
  32. data/ext/oboe_metal/src/bson/bson.h +221 -0
  33. data/ext/oboe_metal/src/bson/platform_hacks.h +91 -0
  34. data/ext/oboe_metal/src/oboe.h +883 -0
  35. data/ext/oboe_metal/src/oboe.hpp +793 -0
  36. data/ext/oboe_metal/src/oboe_debug.h +50 -0
  37. data/ext/oboe_metal/src/oboe_wrap.cxx +6088 -0
  38. data/ext/oboe_metal/tests/test.rb +11 -0
  39. data/gemfiles/delayed_job.gemfile +36 -0
  40. data/gemfiles/frameworks.gemfile +44 -0
  41. data/gemfiles/instrumentation_mocked.gemfile +29 -0
  42. data/gemfiles/libraries.gemfile +85 -0
  43. data/gemfiles/rails23.gemfile +39 -0
  44. data/gemfiles/rails30.gemfile +42 -0
  45. data/gemfiles/rails31.gemfile +44 -0
  46. data/gemfiles/rails32.gemfile +54 -0
  47. data/gemfiles/rails40.gemfile +27 -0
  48. data/gemfiles/rails41.gemfile +27 -0
  49. data/gemfiles/rails42.gemfile +35 -0
  50. data/gemfiles/rails50.gemfile +44 -0
  51. data/gemfiles/rails51.gemfile +44 -0
  52. data/get_version.rb +5 -0
  53. data/init.rb +4 -0
  54. data/lib/appoptics_apm/api/layerinit.rb +39 -0
  55. data/lib/appoptics_apm/api/logging.rb +359 -0
  56. data/lib/appoptics_apm/api/memcache.rb +34 -0
  57. data/lib/appoptics_apm/api/profiling.rb +201 -0
  58. data/lib/appoptics_apm/api/tracing.rb +152 -0
  59. data/lib/appoptics_apm/api/util.rb +128 -0
  60. data/lib/appoptics_apm/api.rb +18 -0
  61. data/lib/appoptics_apm/base.rb +252 -0
  62. data/lib/appoptics_apm/config.rb +281 -0
  63. data/lib/appoptics_apm/frameworks/grape.rb +93 -0
  64. data/lib/appoptics_apm/frameworks/padrino/templates.rb +58 -0
  65. data/lib/appoptics_apm/frameworks/padrino.rb +52 -0
  66. data/lib/appoptics_apm/frameworks/rails/inst/action_controller.rb +106 -0
  67. data/lib/appoptics_apm/frameworks/rails/inst/action_controller2.rb +61 -0
  68. data/lib/appoptics_apm/frameworks/rails/inst/action_controller3.rb +58 -0
  69. data/lib/appoptics_apm/frameworks/rails/inst/action_controller4.rb +48 -0
  70. data/lib/appoptics_apm/frameworks/rails/inst/action_controller5.rb +50 -0
  71. data/lib/appoptics_apm/frameworks/rails/inst/action_controller_api.rb +50 -0
  72. data/lib/appoptics_apm/frameworks/rails/inst/action_view.rb +58 -0
  73. data/lib/appoptics_apm/frameworks/rails/inst/action_view_2x.rb +56 -0
  74. data/lib/appoptics_apm/frameworks/rails/inst/action_view_30.rb +50 -0
  75. data/lib/appoptics_apm/frameworks/rails/inst/active_record.rb +27 -0
  76. data/lib/appoptics_apm/frameworks/rails/inst/connection_adapters/mysql.rb +43 -0
  77. data/lib/appoptics_apm/frameworks/rails/inst/connection_adapters/mysql2.rb +28 -0
  78. data/lib/appoptics_apm/frameworks/rails/inst/connection_adapters/postgresql.rb +30 -0
  79. data/lib/appoptics_apm/frameworks/rails/inst/connection_adapters/utils.rb +120 -0
  80. data/lib/appoptics_apm/frameworks/rails/inst/connection_adapters/utils5x.rb +101 -0
  81. data/lib/appoptics_apm/frameworks/rails.rb +116 -0
  82. data/lib/appoptics_apm/frameworks/sinatra/templates.rb +56 -0
  83. data/lib/appoptics_apm/frameworks/sinatra.rb +71 -0
  84. data/lib/appoptics_apm/inst/bunny-client.rb +148 -0
  85. data/lib/appoptics_apm/inst/bunny-consumer.rb +92 -0
  86. data/lib/appoptics_apm/inst/curb.rb +329 -0
  87. data/lib/appoptics_apm/inst/dalli.rb +85 -0
  88. data/lib/appoptics_apm/inst/delayed_job.rb +92 -0
  89. data/lib/appoptics_apm/inst/em-http-request.rb +105 -0
  90. data/lib/appoptics_apm/inst/excon.rb +130 -0
  91. data/lib/appoptics_apm/inst/faraday.rb +77 -0
  92. data/lib/appoptics_apm/inst/http.rb +83 -0
  93. data/lib/appoptics_apm/inst/httpclient.rb +176 -0
  94. data/lib/appoptics_apm/inst/memcache.rb +102 -0
  95. data/lib/appoptics_apm/inst/memcached.rb +94 -0
  96. data/lib/appoptics_apm/inst/mongo.rb +242 -0
  97. data/lib/appoptics_apm/inst/mongo2.rb +225 -0
  98. data/lib/appoptics_apm/inst/moped.rb +466 -0
  99. data/lib/appoptics_apm/inst/rack.rb +146 -0
  100. data/lib/appoptics_apm/inst/redis.rb +275 -0
  101. data/lib/appoptics_apm/inst/resque.rb +151 -0
  102. data/lib/appoptics_apm/inst/rest-client.rb +50 -0
  103. data/lib/appoptics_apm/inst/sequel.rb +178 -0
  104. data/lib/appoptics_apm/inst/sidekiq-client.rb +53 -0
  105. data/lib/appoptics_apm/inst/sidekiq-worker.rb +67 -0
  106. data/lib/appoptics_apm/inst/twitter-cassandra.rb +294 -0
  107. data/lib/appoptics_apm/inst/typhoeus.rb +113 -0
  108. data/lib/appoptics_apm/instrumentation.rb +22 -0
  109. data/lib/appoptics_apm/legacy_method_profiling.rb +97 -0
  110. data/lib/appoptics_apm/loading.rb +66 -0
  111. data/lib/appoptics_apm/logger.rb +41 -0
  112. data/lib/appoptics_apm/method_profiling.rb +33 -0
  113. data/lib/appoptics_apm/ruby.rb +35 -0
  114. data/lib/appoptics_apm/support.rb +135 -0
  115. data/lib/appoptics_apm/test.rb +94 -0
  116. data/lib/appoptics_apm/thread_local.rb +26 -0
  117. data/lib/appoptics_apm/util.rb +312 -0
  118. data/lib/appoptics_apm/version.rb +15 -0
  119. data/lib/appoptics_apm/xtrace.rb +103 -0
  120. data/lib/appoptics_apm.rb +72 -0
  121. data/lib/joboe_metal.rb +214 -0
  122. data/lib/oboe/README +2 -0
  123. data/lib/oboe/backward_compatibility.rb +80 -0
  124. data/lib/oboe/inst/rack.rb +11 -0
  125. data/lib/oboe.rb +7 -0
  126. data/lib/oboe_metal.rb +187 -0
  127. data/lib/rails/generators/appoptics_apm/install_generator.rb +45 -0
  128. data/lib/rails/generators/appoptics_apm/templates/appoptics_apm_initializer.rb +222 -0
  129. data/ruby_setup.sh +47 -0
  130. data/run_docker_build_gem_upload_to_packagecloud.sh +20 -0
  131. data/run_tests_docker.rb +32 -0
  132. data/test/benchmark/README.md +65 -0
  133. data/test/benchmark/logging_bench.rb +54 -0
  134. data/test/benchmark/with_libraries_gemfile/bunny_bench.rb +69 -0
  135. data/test/benchmark/with_rails5x_gemfile/action_controller5x_bench.rb +43 -0
  136. data/test/frameworks/apps/grape_nested.rb +33 -0
  137. data/test/frameworks/apps/grape_simple.rb +80 -0
  138. data/test/frameworks/apps/padrino_simple.rb +80 -0
  139. data/test/frameworks/apps/sinatra_simple.rb +55 -0
  140. data/test/frameworks/grape_test.rb +286 -0
  141. data/test/frameworks/padrino_test.rb +222 -0
  142. data/test/frameworks/rails3x_test.rb +554 -0
  143. data/test/frameworks/rails4x_test.rb +570 -0
  144. data/test/frameworks/rails5x_api_test.rb +210 -0
  145. data/test/frameworks/rails5x_test.rb +376 -0
  146. data/test/frameworks/rails_shared_tests.rb +172 -0
  147. data/test/frameworks/sinatra_test.rb +140 -0
  148. data/test/instrumentation/bunny_client_test.rb +276 -0
  149. data/test/instrumentation/bunny_consumer_test.rb +204 -0
  150. data/test/instrumentation/curb_test.rb +398 -0
  151. data/test/instrumentation/dalli_test.rb +177 -0
  152. data/test/instrumentation/em_http_request_test.rb +89 -0
  153. data/test/instrumentation/excon_test.rb +231 -0
  154. data/test/instrumentation/faraday_test.rb +228 -0
  155. data/test/instrumentation/http_test.rb +143 -0
  156. data/test/instrumentation/httpclient_test.rb +320 -0
  157. data/test/instrumentation/memcache_test.rb +260 -0
  158. data/test/instrumentation/memcached_test.rb +229 -0
  159. data/test/instrumentation/mongo_v1_test.rb +479 -0
  160. data/test/instrumentation/mongo_v2_index_test.rb +124 -0
  161. data/test/instrumentation/mongo_v2_test.rb +584 -0
  162. data/test/instrumentation/mongo_v2_view_test.rb +435 -0
  163. data/test/instrumentation/moped_test.rb +517 -0
  164. data/test/instrumentation/rack_test.rb +165 -0
  165. data/test/instrumentation/redis_hashes_test.rb +268 -0
  166. data/test/instrumentation/redis_keys_test.rb +321 -0
  167. data/test/instrumentation/redis_lists_test.rb +310 -0
  168. data/test/instrumentation/redis_misc_test.rb +163 -0
  169. data/test/instrumentation/redis_sets_test.rb +296 -0
  170. data/test/instrumentation/redis_sortedsets_test.rb +328 -0
  171. data/test/instrumentation/redis_strings_test.rb +349 -0
  172. data/test/instrumentation/resque_test.rb +185 -0
  173. data/test/instrumentation/rest-client_test.rb +288 -0
  174. data/test/instrumentation/sequel_mysql2_test.rb +353 -0
  175. data/test/instrumentation/sequel_mysql_test.rb +334 -0
  176. data/test/instrumentation/sequel_pg_test.rb +336 -0
  177. data/test/instrumentation/sidekiq-client_test.rb +159 -0
  178. data/test/instrumentation/sidekiq-worker_test.rb +180 -0
  179. data/test/instrumentation/twitter-cassandra_test.rb +424 -0
  180. data/test/instrumentation/typhoeus_test.rb +284 -0
  181. data/test/jobs/delayed_job/db_worker_job.rb +29 -0
  182. data/test/jobs/delayed_job/error_worker_job.rb +10 -0
  183. data/test/jobs/delayed_job/remote_call_worker_job.rb +20 -0
  184. data/test/jobs/resque/db_worker_job.rb +29 -0
  185. data/test/jobs/resque/error_worker_job.rb +10 -0
  186. data/test/jobs/resque/remote_call_worker_job.rb +20 -0
  187. data/test/jobs/sidekiq/db_worker_job.rb +29 -0
  188. data/test/jobs/sidekiq/error_worker_job.rb +10 -0
  189. data/test/jobs/sidekiq/remote_call_worker_job.rb +20 -0
  190. data/test/minitest_helper.rb +276 -0
  191. data/test/mocked/curb_mocked_test.rb +311 -0
  192. data/test/mocked/excon_mocked_test.rb +166 -0
  193. data/test/mocked/faraday_mocked_test.rb +93 -0
  194. data/test/mocked/http_mocked_test.rb +129 -0
  195. data/test/mocked/httpclient_mocked_test.rb +245 -0
  196. data/test/mocked/rest_client_mocked_test.rb +103 -0
  197. data/test/mocked/typhoeus_mocked_test.rb +192 -0
  198. data/test/models/widget.rb +36 -0
  199. data/test/profiling/legacy_method_profiling_test.rb +201 -0
  200. data/test/profiling/method_profiling_test.rb +631 -0
  201. data/test/queues/delayed_job-client_test.rb +95 -0
  202. data/test/queues/delayed_job-worker_test.rb +91 -0
  203. data/test/reporter/reporter_test.rb +14 -0
  204. data/test/servers/delayed_job.rb +107 -0
  205. data/test/servers/rackapp_8101.rb +29 -0
  206. data/test/servers/rails3x_8140.rb +96 -0
  207. data/test/servers/rails4x_8140.rb +96 -0
  208. data/test/servers/rails5x_8140.rb +95 -0
  209. data/test/servers/rails5x_api_8150.rb +78 -0
  210. data/test/servers/sidekiq.rb +29 -0
  211. data/test/servers/sidekiq.yml +7 -0
  212. data/test/servers/sidekiq_initializer.rb +25 -0
  213. data/test/settings +0 -0
  214. data/test/support/auto_tracing_test.rb +50 -0
  215. data/test/support/backcompat_test.rb +276 -0
  216. data/test/support/config_test.rb +149 -0
  217. data/test/support/dnt_test.rb +98 -0
  218. data/test/support/init_report_test.rb +25 -0
  219. data/test/support/liboboe_settings_test.rb +110 -0
  220. data/test/support/logging_test.rb +130 -0
  221. data/test/support/noop_test.rb +88 -0
  222. data/test/support/sql_sanitize_test.rb +55 -0
  223. data/test/support/tracing_mode_test.rb +33 -0
  224. data/test/support/tvalias_test.rb +15 -0
  225. data/test/support/xtrace_test.rb +41 -0
  226. metadata +475 -0
@@ -0,0 +1,479 @@
1
+ # Copyright (c) 2016 SolarWinds, LLC.
2
+ # All rights reserved.
3
+
4
+ require 'minitest_helper'
5
+
6
+ unless ENV['APPOPTICS_MONGO_SERVER']
7
+ ENV['APPOPTICS_MONGO_SERVER'] = "127.0.0.1:27017"
8
+ end
9
+
10
+ if Gem.loaded_specs['mongo'].version.to_s < '2.0.0'
11
+ describe "Mongo" do
12
+ before do
13
+ clear_all_traces
14
+
15
+ @mongo_server = ENV['APPOPTICS_MONGO_SERVER'].split(':')[0]
16
+ @mongo_port = ENV['APPOPTICS_MONGO_SERVER'].split(':')[1]
17
+
18
+ @connection = Mongo::Connection.new(@mongo_server, @mongo_port, :slave_ok => true)
19
+ @db = @connection.db("test-#{ENV['RACK_ENV']}")
20
+
21
+ @collections = @db.collection_names
22
+ @db.create_collection("testCollection") unless @collections.include? "testCollection"
23
+
24
+ # These are standard entry/exit KVs that are passed up with all mongo operations
25
+ @entry_kvs = {
26
+ 'Layer' => 'mongo',
27
+ 'Label' => 'entry',
28
+ 'Flavor' => 'mongodb',
29
+ 'Database' => 'test-test',
30
+ 'RemoteHost' => @mongo_server,
31
+ 'RemotePort' => @mongo_port }
32
+
33
+ @exit_kvs = { 'Layer' => 'mongo', 'Label' => 'exit' }
34
+ @collect_backtraces = AppOpticsAPM::Config[:mongo][:collect_backtraces]
35
+ end
36
+
37
+ after do
38
+ AppOpticsAPM::Config[:mongo][:collect_backtraces] = @collect_backtraces
39
+ end
40
+
41
+ it 'Stock Mongo should be loaded, defined and ready' do
42
+ defined?(::Mongo).wont_match nil
43
+ defined?(::Mongo::DB).wont_match nil
44
+ defined?(::Mongo::Cursor).wont_match nil
45
+ defined?(::Mongo::Collection).wont_match nil
46
+ end
47
+
48
+ it 'reports mongo gem version in init' do
49
+ init_kvs = ::AppOpticsAPM::Util.build_init_report
50
+ assert init_kvs.key?('Ruby.mongo.Version')
51
+ assert_equal Gem.loaded_specs['mongo'].version.to_s, init_kvs['Ruby.mongo.Version']
52
+ end
53
+
54
+ it 'Mongo should have appoptics_apm methods defined' do
55
+ AppOpticsAPM::Inst::Mongo::DB_OPS.each do |m|
56
+ ::Mongo::DB.method_defined?("#{m}_with_appoptics").must_equal true
57
+ end
58
+ AppOpticsAPM::Inst::Mongo::CURSOR_OPS.each do |m|
59
+ ::Mongo::Cursor.method_defined?("#{m}_with_appoptics").must_equal true
60
+ end
61
+ AppOpticsAPM::Inst::Mongo::COLL_WRITE_OPS.each do |m|
62
+ ::Mongo::Collection.method_defined?("#{m}_with_appoptics").must_equal true
63
+ end
64
+ AppOpticsAPM::Inst::Mongo::COLL_QUERY_OPS.each do |m|
65
+ ::Mongo::Collection.method_defined?("#{m}_with_appoptics").must_equal true
66
+ end
67
+ AppOpticsAPM::Inst::Mongo::COLL_INDEX_OPS.each do |m|
68
+ ::Mongo::Collection.method_defined?("#{m}_with_appoptics").must_equal true
69
+ end
70
+ ::Mongo::Collection.method_defined?(:appoptics_collect).must_equal true
71
+ end
72
+
73
+ it "should trace create_collection" do
74
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
75
+ @db.create_collection("create_and_drop_collection_test")
76
+ end
77
+
78
+ traces = get_all_traces
79
+ traces.count.must_equal 4
80
+
81
+ validate_outer_layers(traces, 'mongo_test')
82
+ validate_event_keys(traces[1], @entry_kvs)
83
+ validate_event_keys(traces[2], @exit_kvs)
84
+
85
+ traces[1]['QueryOp'].must_equal "create_collection"
86
+ traces[1]['New_Collection_Name'].must_equal "create_and_drop_collection_test"
87
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
88
+ end
89
+
90
+ it "should trace drop_collection" do
91
+ # Create a collection so we have one to drop
92
+ @db.create_collection("create_and_drop_collection_test")
93
+
94
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
95
+ @db.drop_collection("create_and_drop_collection_test")
96
+ end
97
+
98
+ traces = get_all_traces
99
+ traces.count.must_equal 4
100
+
101
+ validate_outer_layers(traces, 'mongo_test')
102
+ validate_event_keys(traces[1], @entry_kvs)
103
+ validate_event_keys(traces[2], @exit_kvs)
104
+
105
+ traces[1]['QueryOp'].must_equal "drop_collection"
106
+ traces[1]['Collection'].must_equal "create_and_drop_collection_test"
107
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
108
+ end
109
+
110
+ it "should trace count" do
111
+ coll = @db.collection("testCollection")
112
+
113
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
114
+ coll.count(:query => {:name => "MyName"})
115
+ end
116
+
117
+ traces = get_all_traces
118
+ traces.count.must_equal 6
119
+
120
+ validate_outer_layers(traces, 'mongo_test')
121
+ validate_event_keys(traces[1], @entry_kvs)
122
+ validate_event_keys(traces[2], @exit_kvs)
123
+
124
+ traces[3]['QueryOp'].must_equal "count"
125
+ end
126
+
127
+ it "should trace find_and_modify" do
128
+ coll = @db.collection("testCollection")
129
+
130
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
131
+ coll.find_and_modify({ :query => { :name => "MyName" }, :update => { :count => 203 }})
132
+ end
133
+
134
+ traces = get_all_traces
135
+ traces.count.must_equal 4
136
+
137
+ validate_outer_layers(traces, 'mongo_test')
138
+ validate_event_keys(traces[1], @entry_kvs)
139
+ validate_event_keys(traces[2], @exit_kvs)
140
+
141
+ traces[1]['Collection'].must_equal "testCollection"
142
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
143
+ traces[1]['QueryOp'].must_equal "find_and_modify"
144
+ traces[1]['Update_Document'].must_equal "{:count=>203}"
145
+ end
146
+
147
+ it "should trace insert" do
148
+ coll = @db.collection("testCollection")
149
+
150
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
151
+ doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
152
+ coll.insert(doc)
153
+ end
154
+
155
+ traces = get_all_traces
156
+ traces.count.must_equal 4
157
+
158
+ validate_outer_layers(traces, 'mongo_test')
159
+ validate_event_keys(traces[1], @entry_kvs)
160
+ validate_event_keys(traces[2], @exit_kvs)
161
+
162
+ traces[1]['Collection'].must_equal "testCollection"
163
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
164
+ traces[1]['QueryOp'].must_equal "insert"
165
+ # Don't test exact hash value since to_json hash ordering varies between 1.8.7 and 1.9+
166
+ traces[1].has_key?('Query').must_equal true
167
+ end
168
+
169
+ it "should trace map_reduce" do
170
+ coll = @db.collection("testCollection")
171
+
172
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
173
+ map = "function() { emit(this.name, 1); }"
174
+ reduce = "function(k, vals) { var sum = 0; for(var i in vals) sum += vals[i]; return sum; }"
175
+ coll.map_reduce(map, reduce, { :out => "mr_results", :limit => 100, :read => :primary })
176
+ end
177
+
178
+ traces = get_all_traces
179
+ traces.count.must_equal 4
180
+
181
+ validate_outer_layers(traces, 'mongo_test')
182
+ validate_event_keys(traces[1], @entry_kvs)
183
+ validate_event_keys(traces[2], @exit_kvs)
184
+
185
+ traces[1]['Collection'].must_equal "testCollection"
186
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
187
+ traces[1]['QueryOp'].must_equal "map_reduce"
188
+ traces[1]['Map_Function'].must_equal "function() { emit(this.name, 1); }"
189
+ traces[1]['Reduce_Function'].must_equal "function(k, vals) { var sum = 0; for(var i in vals) sum += vals[i]; return sum; }"
190
+ traces[1]['Limit'].must_equal 100
191
+ end
192
+
193
+ it "should trace remove" do
194
+ coll = @db.collection("testCollection")
195
+
196
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
197
+ coll.remove(:name => "SaveOp")
198
+ end
199
+
200
+ traces = get_all_traces
201
+ traces.count.must_equal 4
202
+
203
+ validate_outer_layers(traces, 'mongo_test')
204
+ validate_event_keys(traces[1], @entry_kvs)
205
+ validate_event_keys(traces[2], @exit_kvs)
206
+
207
+ traces[1]['Collection'].must_equal "testCollection"
208
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
209
+ traces[1]['QueryOp'].must_equal "remove"
210
+ traces[1]['Query'].must_equal "{\"name\":\"SaveOp\"}"
211
+ end
212
+
213
+ it "should trace rename" do
214
+ coll = @db.collection("testCollection")
215
+ new_name = (0...10).map{ ('a'..'z').to_a[rand(26)] }.join
216
+
217
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
218
+ coll.rename(new_name)
219
+ end
220
+
221
+ traces = get_all_traces
222
+ traces.count.must_equal 4
223
+
224
+ validate_outer_layers(traces, 'mongo_test')
225
+ validate_event_keys(traces[1], @entry_kvs)
226
+ validate_event_keys(traces[2], @exit_kvs)
227
+
228
+ traces[1]['Collection'].must_equal "testCollection"
229
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
230
+ traces[1]['QueryOp'].must_equal "rename"
231
+ traces[1]['New_Collection_Name'].must_equal new_name
232
+
233
+ # Clean up after test and set collection name back to original
234
+ coll.rename("testCollection")
235
+ end
236
+
237
+ it "should trace update" do
238
+ coll = @db.collection("testCollection")
239
+
240
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
241
+ # Two types of update calls
242
+ coll.update({"_id" => 1}, { "$set" => {"name" => "MongoDB Ruby"}}, :multi => true)
243
+
244
+ doc = {"name" => "MyOtherName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
245
+ coll.update({"_id" => 1}, doc)
246
+ end
247
+
248
+ traces = get_all_traces
249
+ traces.count.must_equal 6
250
+
251
+ validate_outer_layers(traces, 'mongo_test')
252
+ validate_event_keys(traces[1], @entry_kvs)
253
+ validate_event_keys(traces[2], @exit_kvs)
254
+
255
+ traces[1]['Collection'].must_equal "testCollection"
256
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
257
+ traces[1]['QueryOp'].must_equal "update"
258
+ traces[1]['Query'].must_equal "{\"_id\":1}"
259
+
260
+ validate_event_keys(traces[3], @entry_kvs)
261
+ validate_event_keys(traces[4], @exit_kvs)
262
+
263
+ traces[3].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
264
+ traces[3]['QueryOp'].must_equal "update"
265
+ traces[3]['Query'].must_equal "{\"_id\":1}"
266
+ end
267
+
268
+ it "should trace distinct" do
269
+ coll = @db.collection("testCollection")
270
+
271
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
272
+ coll.distinct("count")
273
+ end
274
+
275
+ traces = get_all_traces
276
+ traces.count.must_equal 4
277
+
278
+ validate_outer_layers(traces, 'mongo_test')
279
+ validate_event_keys(traces[1], @entry_kvs)
280
+ validate_event_keys(traces[2], @exit_kvs)
281
+
282
+ traces[1]['Collection'].must_equal "testCollection"
283
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
284
+ traces[1]['QueryOp'].must_equal "distinct"
285
+ end
286
+
287
+ it "should trace find" do
288
+ coll = @db.collection("testCollection")
289
+ result = nil
290
+
291
+ # Insert a doc to assure we get a result
292
+ doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
293
+ coll.insert(doc)
294
+
295
+ # If given an optional block +find+ will yield a Cursor to that block,
296
+ # close the cursor, and then return nil. This guarantees that partially
297
+ # evaluated cursors will be closed. If given no block +find+ returns a
298
+ # cursor.
299
+ # https://github.com/mongodb/mongo-ruby-driver/blob/1.10.1/lib/mongo/collection.rb#L178
300
+
301
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
302
+ result = coll.find(:name => "MyName", :limit => 1)
303
+ end
304
+
305
+ traces = get_all_traces
306
+ traces.count.must_equal 4
307
+
308
+ validate_outer_layers(traces, 'mongo_test')
309
+ validate_event_keys(traces[1], @entry_kvs)
310
+ validate_event_keys(traces[2], @exit_kvs)
311
+
312
+ result.wont_match nil
313
+ result.is_a?(Mongo::Cursor).must_equal true
314
+ traces[1]['Collection'].must_equal "testCollection"
315
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
316
+ traces[1]['QueryOp'].must_equal "find"
317
+ traces[1].has_key?('Query').must_equal true
318
+ traces[1]['Limit'].must_equal 1
319
+ end
320
+
321
+ it "should trace find (with block)" do
322
+ coll = @db.collection("testCollection")
323
+ result = []
324
+
325
+ # Insert a doc to assure we get a result
326
+ doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
327
+ coll.insert(doc)
328
+
329
+ # If given an optional block +find+ will yield a Cursor to that block,
330
+ # close the cursor, and then return nil. This guarantees that partially
331
+ # evaluated cursors will be closed. If given no block +find+ returns a
332
+ # cursor.
333
+ # https://github.com/mongodb/mongo-ruby-driver/blob/1.10.1/lib/mongo/collection.rb#L178
334
+
335
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
336
+ blk = lambda { |x| x }
337
+ result = coll.find(:name => "MyName", :limit => 10, &blk)
338
+ end
339
+
340
+ traces = get_all_traces
341
+ traces.count.must_equal 4
342
+
343
+ validate_outer_layers(traces, 'mongo_test')
344
+ validate_event_keys(traces[1], @entry_kvs)
345
+ validate_event_keys(traces[2], @exit_kvs)
346
+
347
+ result.must_equal nil
348
+ traces[1]['Collection'].must_equal "testCollection"
349
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
350
+ traces[1]['QueryOp'].must_equal "find"
351
+ traces[1].has_key?('Query').must_equal true
352
+ end
353
+
354
+ it "should trace group" do
355
+ coll = @db.collection("testCollection")
356
+
357
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
358
+ coll.group( :key => :type,
359
+ :cond => { :count => 1 },
360
+ :initial => { :count => 0 },
361
+ :reduce => 'function(obj,prev) { prev.count += obj.c; }')
362
+ end
363
+
364
+ traces = get_all_traces
365
+ traces.count.must_equal 4
366
+
367
+ validate_outer_layers(traces, 'mongo_test')
368
+ validate_event_keys(traces[1], @entry_kvs)
369
+ validate_event_keys(traces[2], @exit_kvs)
370
+
371
+ traces[1]['Collection'].must_equal "testCollection"
372
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
373
+ traces[1]['QueryOp'].must_equal "group"
374
+ # Don't test exact hash value since to_json hash ordering varies between 1.8.7 and 1.9+
375
+ traces[1].has_key?('Query').must_equal true
376
+ end
377
+
378
+ it "should trace create, ensure and drop index" do
379
+ coll = @db.collection("testCollection")
380
+
381
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
382
+ coll.create_index("i")
383
+ coll.ensure_index("i")
384
+ coll.drop_index("i_1")
385
+ end
386
+
387
+ traces = get_all_traces
388
+ traces.count.must_equal 8
389
+
390
+ validate_outer_layers(traces, 'mongo_test')
391
+ validate_event_keys(traces[1], @entry_kvs)
392
+ validate_event_keys(traces[2], @exit_kvs)
393
+
394
+ traces[1]['Collection'].must_equal "testCollection"
395
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
396
+ traces[1]['QueryOp'].must_equal "create_index"
397
+
398
+ validate_event_keys(traces[3], @entry_kvs)
399
+ validate_event_keys(traces[4], @exit_kvs)
400
+
401
+ traces[3]['Collection'].must_equal "testCollection"
402
+ traces[3].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
403
+ traces[3]['QueryOp'].must_equal "ensure_index"
404
+
405
+ validate_event_keys(traces[5], @entry_kvs)
406
+ validate_event_keys(traces[6], @exit_kvs)
407
+
408
+ traces[5]['Collection'].must_equal "testCollection"
409
+ traces[5].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
410
+ traces[5]['QueryOp'].must_equal "drop_index"
411
+ end
412
+
413
+ it "should trace drop_indexes" do
414
+ coll = @db.collection("testCollection")
415
+
416
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
417
+ coll.drop_indexes
418
+ end
419
+
420
+ traces = get_all_traces
421
+ traces.count.must_equal 4
422
+
423
+ validate_outer_layers(traces, 'mongo_test')
424
+ validate_event_keys(traces[1], @entry_kvs)
425
+ validate_event_keys(traces[2], @exit_kvs)
426
+
427
+ traces[1]['Collection'].must_equal "testCollection"
428
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
429
+ traces[1]['QueryOp'].must_equal "drop_indexes"
430
+ end
431
+
432
+ it "should trace index_information" do
433
+ coll = @db.collection("testCollection")
434
+
435
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
436
+ coll.index_information
437
+ end
438
+
439
+ traces = get_all_traces
440
+ traces.count.must_equal 4
441
+
442
+ validate_outer_layers(traces, 'mongo_test')
443
+ validate_event_keys(traces[1], @entry_kvs)
444
+ validate_event_keys(traces[2], @exit_kvs)
445
+
446
+ traces[1]['Collection'].must_equal "testCollection"
447
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
448
+ traces[1]['QueryOp'].must_equal "index_information"
449
+ end
450
+
451
+ it "should obey :collect_backtraces setting when true" do
452
+ AppOpticsAPM::Config[:mongo][:collect_backtraces] = true
453
+
454
+ coll = @db.collection("testCollection")
455
+
456
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
457
+ doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
458
+ coll.insert(doc)
459
+ end
460
+
461
+ traces = get_all_traces
462
+ layer_has_key(traces, 'mongo', 'Backtrace')
463
+ end
464
+
465
+ it "should obey :collect_backtraces setting when false" do
466
+ AppOpticsAPM::Config[:mongo][:collect_backtraces] = false
467
+
468
+ coll = @db.collection("testCollection")
469
+
470
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
471
+ doc = {"name" => "MyName", "type" => "MyType", "count" => 1, "info" => {"x" => 203, "y" => '102'}}
472
+ coll.insert(doc)
473
+ end
474
+
475
+ traces = get_all_traces
476
+ layer_doesnt_have_key(traces, 'mongo', 'Backtrace')
477
+ end
478
+ end
479
+ end
@@ -0,0 +1,124 @@
1
+ # Copyright (c) 2016 SolarWinds, LLC.
2
+ # All rights reserved.
3
+
4
+ require 'minitest_helper'
5
+
6
+ unless ENV['APPOPTICS_MONGO_SERVER']
7
+ ENV['APPOPTICS_MONGO_SERVER'] = "127.0.0.1:27017"
8
+ end
9
+
10
+ if defined?(::Mongo::VERSION) && Mongo::VERSION >= '2.0.0'
11
+ describe "MongoIndex" do
12
+ before do
13
+ clear_all_traces
14
+
15
+ @client = Mongo::Client.new([ ENV['APPOPTICS_MONGO_SERVER'] ], :database => "appoptics_apm-#{ENV['RACK_ENV']}")
16
+ if Mongo::VERSION < '2.2'
17
+ Mongo::Logger.logger.level = Logger::INFO
18
+ else
19
+ @client.logger.level = Logger::INFO
20
+ end
21
+ @db = @client.database
22
+
23
+ @collections = @db.collection_names
24
+ @testCollection = @client[:test_collection]
25
+ @testCollection.create unless @collections.include? "test_collection"
26
+
27
+ # These are standard entry/exit KVs that are passed up with all mongo operations
28
+ @entry_kvs = {
29
+ 'Layer' => 'mongo',
30
+ 'Label' => 'entry',
31
+ 'Flavor' => 'mongodb',
32
+ 'Database' => 'appoptics_apm-test',
33
+ 'RemoteHost' => ENV['APPOPTICS_MONGO_SERVER'] }
34
+
35
+ @exit_kvs = { 'Layer' => 'mongo', 'Label' => 'exit' }
36
+ @collect_backtraces = AppOpticsAPM::Config[:mongo][:collect_backtraces]
37
+ end
38
+
39
+ after do
40
+ AppOpticsAPM::Config[:mongo][:collect_backtraces] = @collect_backtraces
41
+ end
42
+
43
+ it "should trace create_one" do
44
+ coll = @db[:test_collection]
45
+ coll.indexes.drop_all
46
+
47
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
48
+ coll.indexes.create_one({ :name => 1 })
49
+ end
50
+
51
+ traces = get_all_traces
52
+ traces.count.must_equal 4
53
+
54
+ validate_outer_layers(traces, 'mongo_test')
55
+ validate_event_keys(traces[1], @entry_kvs)
56
+ validate_event_keys(traces[2], @exit_kvs)
57
+
58
+ traces[1]['Collection'].must_equal "test_collection"
59
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
60
+ traces[1]['QueryOp'].must_equal "create_one"
61
+ end
62
+
63
+ it "should trace create_many" do
64
+ coll = @db[:test_collection]
65
+ coll.indexes.drop_all
66
+
67
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
68
+ coll.indexes.create_many([ { :key => {:asdf => 1}, :unique => false },
69
+ { :key => {:age => -1}, :background => true} ])
70
+ end
71
+
72
+ traces = get_all_traces
73
+ traces.count.must_equal 4
74
+
75
+ validate_outer_layers(traces, 'mongo_test')
76
+ validate_event_keys(traces[1], @entry_kvs)
77
+ validate_event_keys(traces[2], @exit_kvs)
78
+
79
+ traces[1]['Collection'].must_equal "test_collection"
80
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
81
+ traces[1]['QueryOp'].must_equal "create_many"
82
+ end
83
+
84
+ it "should trace drop_one" do
85
+ coll = @db[:test_collection]
86
+ coll.indexes.create_one({ :name => 1 })
87
+
88
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
89
+ coll.indexes.drop_one('name_1')
90
+ end
91
+
92
+ traces = get_all_traces
93
+ traces.count.must_equal 4
94
+
95
+ validate_outer_layers(traces, 'mongo_test')
96
+ validate_event_keys(traces[1], @entry_kvs)
97
+ validate_event_keys(traces[2], @exit_kvs)
98
+
99
+ traces[1]['Collection'].must_equal "test_collection"
100
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
101
+ traces[1]['QueryOp'].must_equal "drop_one"
102
+ end
103
+
104
+ it "should trace drop_all" do
105
+ coll = @db[:test_collection]
106
+ coll.indexes.create_one({ :name => 1 })
107
+
108
+ AppOpticsAPM::API.start_trace('mongo_test', '', {}) do
109
+ coll.indexes.drop_all
110
+ end
111
+
112
+ traces = get_all_traces
113
+ traces.count.must_equal 4
114
+
115
+ validate_outer_layers(traces, 'mongo_test')
116
+ validate_event_keys(traces[1], @entry_kvs)
117
+ validate_event_keys(traces[2], @exit_kvs)
118
+
119
+ traces[1]['Collection'].must_equal "test_collection"
120
+ traces[1].has_key?('Backtrace').must_equal AppOpticsAPM::Config[:mongo][:collect_backtraces]
121
+ traces[1]['QueryOp'].must_equal "drop_all"
122
+ end
123
+ end
124
+ end