mongo 2.12.4 → 2.14.0.rc1

Sign up to get free protection for your applications and to get access to all the features.
Files changed (900) hide show
  1. checksums.yaml +4 -4
  2. checksums.yaml.gz.sig +0 -0
  3. data.tar.gz.sig +0 -0
  4. data/CONTRIBUTING.md +8 -36
  5. data/LICENSE +1 -1
  6. data/README.md +54 -54
  7. data/Rakefile +16 -10
  8. data/lib/mongo.rb +17 -3
  9. data/lib/mongo/active_support.rb +1 -1
  10. data/lib/mongo/address.rb +40 -13
  11. data/lib/mongo/address/ipv4.rb +32 -5
  12. data/lib/mongo/address/ipv6.rb +32 -5
  13. data/lib/mongo/address/unix.rb +3 -3
  14. data/lib/mongo/address/validator.rb +1 -1
  15. data/lib/mongo/auth.rb +36 -13
  16. data/lib/mongo/auth/aws.rb +37 -0
  17. data/lib/mongo/auth/aws/conversation.rb +128 -0
  18. data/lib/mongo/auth/aws/credentials_retriever.rb +219 -0
  19. data/lib/mongo/auth/aws/request.rb +283 -0
  20. data/lib/mongo/auth/base.rb +129 -0
  21. data/lib/mongo/auth/conversation_base.rb +52 -0
  22. data/lib/mongo/auth/cr.rb +9 -36
  23. data/lib/mongo/auth/cr/conversation.rb +24 -69
  24. data/lib/mongo/auth/credential_cache.rb +1 -1
  25. data/lib/mongo/auth/gssapi.rb +38 -0
  26. data/lib/mongo/auth/gssapi/conversation.rb +108 -0
  27. data/lib/mongo/auth/ldap.rb +9 -34
  28. data/lib/mongo/auth/ldap/conversation.rb +3 -43
  29. data/lib/mongo/auth/roles.rb +1 -1
  30. data/lib/mongo/auth/sasl_conversation_base.rb +111 -0
  31. data/lib/mongo/auth/scram.rb +39 -51
  32. data/lib/mongo/auth/scram/conversation.rb +12 -506
  33. data/lib/mongo/auth/scram256.rb +31 -0
  34. data/lib/mongo/auth/scram256/conversation.rb +63 -0
  35. data/lib/mongo/auth/scram_conversation_base.rb +402 -0
  36. data/lib/mongo/auth/stringprep.rb +5 -4
  37. data/lib/mongo/auth/stringprep/profiles/sasl.rb +2 -1
  38. data/lib/mongo/auth/stringprep/tables.rb +2 -1
  39. data/lib/mongo/auth/stringprep/unicode_normalize/normalize.rb +1 -0
  40. data/lib/mongo/auth/stringprep/unicode_normalize/tables.rb +1 -0
  41. data/lib/mongo/auth/user.rb +2 -2
  42. data/lib/mongo/auth/user/view.rb +1 -1
  43. data/lib/mongo/auth/x509.rb +14 -32
  44. data/lib/mongo/auth/x509/conversation.rb +15 -42
  45. data/lib/mongo/background_thread.rb +11 -2
  46. data/lib/mongo/bson.rb +1 -1
  47. data/lib/mongo/bulk_write.rb +79 -26
  48. data/lib/mongo/bulk_write/combineable.rb +20 -8
  49. data/lib/mongo/bulk_write/ordered_combiner.rb +1 -1
  50. data/lib/mongo/bulk_write/result.rb +1 -1
  51. data/lib/mongo/bulk_write/result_combiner.rb +1 -1
  52. data/lib/mongo/bulk_write/transformable.rb +9 -10
  53. data/lib/mongo/bulk_write/unordered_combiner.rb +1 -1
  54. data/lib/mongo/bulk_write/validatable.rb +5 -1
  55. data/lib/mongo/caching_cursor.rb +74 -0
  56. data/lib/mongo/client.rb +261 -39
  57. data/lib/mongo/client_encryption.rb +1 -1
  58. data/lib/mongo/cluster.rb +180 -72
  59. data/lib/mongo/cluster/periodic_executor.rb +1 -1
  60. data/lib/mongo/cluster/reapers/cursor_reaper.rb +1 -1
  61. data/lib/mongo/cluster/reapers/socket_reaper.rb +1 -1
  62. data/lib/mongo/cluster/sdam_flow.rb +36 -17
  63. data/lib/mongo/cluster/topology.rb +19 -2
  64. data/lib/mongo/cluster/topology/base.rb +1 -1
  65. data/lib/mongo/cluster/topology/no_replica_set_options.rb +1 -1
  66. data/lib/mongo/cluster/topology/replica_set_no_primary.rb +4 -3
  67. data/lib/mongo/cluster/topology/replica_set_with_primary.rb +1 -1
  68. data/lib/mongo/cluster/topology/sharded.rb +2 -2
  69. data/lib/mongo/cluster/topology/single.rb +3 -3
  70. data/lib/mongo/cluster/topology/unknown.rb +1 -1
  71. data/lib/mongo/cluster_time.rb +1 -1
  72. data/lib/mongo/collection.rb +77 -24
  73. data/lib/mongo/collection/view.rb +30 -22
  74. data/lib/mongo/collection/view/aggregation.rb +31 -7
  75. data/lib/mongo/collection/view/builder.rb +1 -1
  76. data/lib/mongo/collection/view/builder/aggregation.rb +1 -1
  77. data/lib/mongo/collection/view/builder/find_command.rb +47 -19
  78. data/lib/mongo/collection/view/builder/flags.rb +1 -1
  79. data/lib/mongo/collection/view/builder/map_reduce.rb +1 -1
  80. data/lib/mongo/collection/view/builder/modifiers.rb +1 -1
  81. data/lib/mongo/collection/view/builder/op_query.rb +1 -1
  82. data/lib/mongo/collection/view/change_stream.rb +3 -6
  83. data/lib/mongo/collection/view/change_stream/retryable.rb +1 -1
  84. data/lib/mongo/collection/view/explainable.rb +28 -9
  85. data/lib/mongo/collection/view/immutable.rb +1 -1
  86. data/lib/mongo/collection/view/iterable.rb +80 -14
  87. data/lib/mongo/collection/view/map_reduce.rb +7 -4
  88. data/lib/mongo/collection/view/readable.rb +22 -4
  89. data/lib/mongo/collection/view/writable.rb +107 -13
  90. data/lib/mongo/crypt.rb +1 -1
  91. data/lib/mongo/crypt/auto_decryption_context.rb +1 -1
  92. data/lib/mongo/crypt/auto_encrypter.rb +1 -1
  93. data/lib/mongo/crypt/auto_encryption_context.rb +1 -1
  94. data/lib/mongo/crypt/binary.rb +1 -1
  95. data/lib/mongo/crypt/binding.rb +1 -1
  96. data/lib/mongo/crypt/context.rb +1 -1
  97. data/lib/mongo/crypt/data_key_context.rb +1 -1
  98. data/lib/mongo/crypt/encryption_io.rb +45 -28
  99. data/lib/mongo/crypt/explicit_decryption_context.rb +1 -1
  100. data/lib/mongo/crypt/explicit_encrypter.rb +1 -1
  101. data/lib/mongo/crypt/explicit_encryption_context.rb +1 -1
  102. data/lib/mongo/crypt/handle.rb +1 -1
  103. data/lib/mongo/crypt/hooks.rb +1 -1
  104. data/lib/mongo/crypt/kms_context.rb +1 -1
  105. data/lib/mongo/crypt/status.rb +1 -1
  106. data/lib/mongo/cursor.rb +4 -3
  107. data/lib/mongo/cursor/builder.rb +1 -1
  108. data/lib/mongo/cursor/builder/get_more_command.rb +1 -1
  109. data/lib/mongo/cursor/builder/kill_cursors_command.rb +1 -1
  110. data/lib/mongo/cursor/builder/op_get_more.rb +1 -1
  111. data/lib/mongo/cursor/builder/op_kill_cursors.rb +1 -1
  112. data/lib/mongo/database.rb +41 -5
  113. data/lib/mongo/database/view.rb +21 -6
  114. data/lib/mongo/dbref.rb +1 -1
  115. data/lib/mongo/distinguishing_semaphore.rb +55 -0
  116. data/lib/mongo/error.rb +30 -2
  117. data/lib/mongo/error/auth_error.rb +1 -1
  118. data/lib/mongo/error/bulk_write_error.rb +1 -1
  119. data/lib/mongo/error/change_stream_resumable.rb +1 -1
  120. data/lib/mongo/error/closed_stream.rb +1 -1
  121. data/lib/mongo/error/connection_check_out_timeout.rb +1 -1
  122. data/lib/mongo/error/connection_perished.rb +23 -0
  123. data/lib/mongo/error/credential_check_error.rb +26 -0
  124. data/lib/mongo/error/extra_file_chunk.rb +1 -1
  125. data/lib/mongo/error/file_not_found.rb +1 -1
  126. data/lib/mongo/error/handshake_error.rb +1 -1
  127. data/lib/mongo/error/insufficient_iteration_count.rb +1 -1
  128. data/lib/mongo/error/invalid_address.rb +1 -1
  129. data/lib/mongo/error/invalid_application_name.rb +1 -1
  130. data/lib/mongo/error/invalid_bulk_operation.rb +1 -1
  131. data/lib/mongo/error/invalid_bulk_operation_type.rb +1 -1
  132. data/lib/mongo/error/invalid_collection_name.rb +1 -1
  133. data/lib/mongo/error/invalid_cursor_operation.rb +1 -1
  134. data/lib/mongo/error/invalid_database_name.rb +1 -1
  135. data/lib/mongo/error/invalid_document.rb +1 -1
  136. data/lib/mongo/error/invalid_file.rb +1 -1
  137. data/lib/mongo/error/invalid_file_revision.rb +1 -1
  138. data/lib/mongo/error/invalid_min_pool_size.rb +1 -1
  139. data/lib/mongo/error/invalid_nonce.rb +2 -2
  140. data/lib/mongo/error/invalid_read_concern.rb +28 -0
  141. data/lib/mongo/error/invalid_read_option.rb +1 -1
  142. data/lib/mongo/error/invalid_replacement_document.rb +1 -1
  143. data/lib/mongo/error/invalid_server_auth_host.rb +22 -0
  144. data/lib/mongo/error/invalid_server_auth_response.rb +23 -0
  145. data/lib/mongo/error/invalid_server_preference.rb +6 -1
  146. data/lib/mongo/error/invalid_session.rb +3 -2
  147. data/lib/mongo/error/invalid_signature.rb +1 -1
  148. data/lib/mongo/error/invalid_transaction_operation.rb +1 -1
  149. data/lib/mongo/error/invalid_txt_record.rb +1 -1
  150. data/lib/mongo/error/invalid_update_document.rb +1 -1
  151. data/lib/mongo/error/invalid_uri.rb +1 -1
  152. data/lib/mongo/error/invalid_write_concern.rb +1 -1
  153. data/lib/mongo/error/lint_error.rb +1 -1
  154. data/lib/mongo/error/max_bson_size.rb +1 -1
  155. data/lib/mongo/error/max_message_size.rb +1 -1
  156. data/lib/mongo/error/mismatched_domain.rb +1 -1
  157. data/lib/mongo/error/missing_file_chunk.rb +1 -1
  158. data/lib/mongo/error/missing_password.rb +1 -1
  159. data/lib/mongo/error/missing_resume_token.rb +1 -1
  160. data/lib/mongo/error/missing_scram_server_signature.rb +27 -0
  161. data/lib/mongo/error/multi_index_drop.rb +1 -1
  162. data/lib/mongo/error/need_primary_server.rb +1 -1
  163. data/lib/mongo/error/no_server_available.rb +1 -1
  164. data/lib/mongo/error/no_srv_records.rb +1 -1
  165. data/lib/mongo/error/notable.rb +18 -3
  166. data/lib/mongo/error/operation_failure.rb +63 -44
  167. data/lib/mongo/error/parser.rb +16 -5
  168. data/lib/mongo/error/pool_closed_error.rb +1 -1
  169. data/lib/mongo/error/raise_original_error.rb +29 -0
  170. data/lib/mongo/error/server_certificate_revoked.rb +22 -0
  171. data/lib/mongo/error/session_ended.rb +1 -1
  172. data/lib/mongo/error/sessions_not_supported.rb +35 -0
  173. data/lib/mongo/error/socket_error.rb +1 -1
  174. data/lib/mongo/error/socket_timeout_error.rb +1 -1
  175. data/lib/mongo/error/unchangeable_collection_option.rb +1 -1
  176. data/lib/mongo/error/unexpected_chunk_length.rb +1 -1
  177. data/lib/mongo/error/unexpected_response.rb +1 -1
  178. data/lib/mongo/error/unknown_payload_type.rb +1 -1
  179. data/lib/mongo/error/unsupported_array_filters.rb +7 -2
  180. data/lib/mongo/error/unsupported_collation.rb +7 -2
  181. data/lib/mongo/error/unsupported_features.rb +1 -1
  182. data/lib/mongo/error/unsupported_message_type.rb +1 -1
  183. data/lib/mongo/error/unsupported_option.rb +101 -0
  184. data/lib/mongo/error/write_retryable.rb +1 -1
  185. data/lib/mongo/event.rb +1 -1
  186. data/lib/mongo/event/base.rb +7 -1
  187. data/lib/mongo/event/listeners.rb +1 -1
  188. data/lib/mongo/event/publisher.rb +1 -1
  189. data/lib/mongo/event/subscriber.rb +1 -1
  190. data/lib/mongo/grid.rb +1 -1
  191. data/lib/mongo/grid/file.rb +1 -1
  192. data/lib/mongo/grid/file/chunk.rb +1 -1
  193. data/lib/mongo/grid/file/info.rb +1 -1
  194. data/lib/mongo/grid/fs_bucket.rb +9 -4
  195. data/lib/mongo/grid/stream.rb +1 -1
  196. data/lib/mongo/grid/stream/read.rb +1 -1
  197. data/lib/mongo/grid/stream/write.rb +1 -1
  198. data/lib/mongo/id.rb +1 -1
  199. data/lib/mongo/index.rb +2 -1
  200. data/lib/mongo/index/view.rb +61 -11
  201. data/lib/mongo/lint.rb +11 -0
  202. data/lib/mongo/loggable.rb +1 -1
  203. data/lib/mongo/logger.rb +4 -4
  204. data/lib/mongo/monitoring.rb +39 -1
  205. data/lib/mongo/monitoring/cmap_log_subscriber.rb +1 -1
  206. data/lib/mongo/monitoring/command_log_subscriber.rb +20 -5
  207. data/lib/mongo/monitoring/event.rb +1 -1
  208. data/lib/mongo/monitoring/event/cmap.rb +1 -1
  209. data/lib/mongo/monitoring/event/cmap/base.rb +1 -1
  210. data/lib/mongo/monitoring/event/cmap/connection_check_out_failed.rb +1 -1
  211. data/lib/mongo/monitoring/event/cmap/connection_check_out_started.rb +1 -1
  212. data/lib/mongo/monitoring/event/cmap/connection_checked_in.rb +1 -1
  213. data/lib/mongo/monitoring/event/cmap/connection_checked_out.rb +1 -1
  214. data/lib/mongo/monitoring/event/cmap/connection_closed.rb +1 -1
  215. data/lib/mongo/monitoring/event/cmap/connection_created.rb +1 -1
  216. data/lib/mongo/monitoring/event/cmap/connection_ready.rb +1 -1
  217. data/lib/mongo/monitoring/event/cmap/pool_cleared.rb +1 -1
  218. data/lib/mongo/monitoring/event/cmap/pool_closed.rb +1 -1
  219. data/lib/mongo/monitoring/event/cmap/pool_created.rb +1 -1
  220. data/lib/mongo/monitoring/event/command_failed.rb +12 -1
  221. data/lib/mongo/monitoring/event/command_started.rb +44 -3
  222. data/lib/mongo/monitoring/event/command_succeeded.rb +12 -1
  223. data/lib/mongo/monitoring/event/secure.rb +8 -2
  224. data/lib/mongo/monitoring/event/server_closed.rb +2 -2
  225. data/lib/mongo/monitoring/event/server_description_changed.rb +28 -5
  226. data/lib/mongo/monitoring/event/server_heartbeat_failed.rb +10 -3
  227. data/lib/mongo/monitoring/event/server_heartbeat_started.rb +10 -3
  228. data/lib/mongo/monitoring/event/server_heartbeat_succeeded.rb +10 -3
  229. data/lib/mongo/monitoring/event/server_opening.rb +2 -2
  230. data/lib/mongo/monitoring/event/topology_changed.rb +2 -2
  231. data/lib/mongo/monitoring/event/topology_closed.rb +2 -2
  232. data/lib/mongo/monitoring/event/topology_opening.rb +2 -2
  233. data/lib/mongo/monitoring/publishable.rb +9 -9
  234. data/lib/mongo/monitoring/sdam_log_subscriber.rb +1 -1
  235. data/lib/mongo/monitoring/server_closed_log_subscriber.rb +1 -1
  236. data/lib/mongo/monitoring/server_description_changed_log_subscriber.rb +10 -2
  237. data/lib/mongo/monitoring/server_opening_log_subscriber.rb +1 -1
  238. data/lib/mongo/monitoring/topology_changed_log_subscriber.rb +2 -2
  239. data/lib/mongo/monitoring/topology_closed_log_subscriber.rb +1 -1
  240. data/lib/mongo/monitoring/topology_opening_log_subscriber.rb +1 -1
  241. data/lib/mongo/monitoring/unified_sdam_log_subscriber.rb +1 -1
  242. data/lib/mongo/operation.rb +3 -0
  243. data/lib/mongo/operation/aggregate.rb +1 -1
  244. data/lib/mongo/operation/aggregate/command.rb +5 -5
  245. data/lib/mongo/operation/aggregate/op_msg.rb +1 -1
  246. data/lib/mongo/operation/aggregate/result.rb +10 -9
  247. data/lib/mongo/operation/collections_info.rb +3 -36
  248. data/lib/mongo/operation/collections_info/command.rb +40 -0
  249. data/lib/mongo/operation/collections_info/result.rb +3 -1
  250. data/lib/mongo/operation/command.rb +1 -1
  251. data/lib/mongo/operation/command/command.rb +3 -3
  252. data/lib/mongo/operation/command/op_msg.rb +1 -1
  253. data/lib/mongo/operation/count.rb +1 -1
  254. data/lib/mongo/operation/count/command.rb +3 -3
  255. data/lib/mongo/operation/count/op_msg.rb +1 -1
  256. data/lib/mongo/operation/create.rb +1 -1
  257. data/lib/mongo/operation/create/command.rb +3 -3
  258. data/lib/mongo/operation/create/op_msg.rb +1 -1
  259. data/lib/mongo/operation/create_index.rb +1 -1
  260. data/lib/mongo/operation/create_index/command.rb +4 -4
  261. data/lib/mongo/operation/create_index/op_msg.rb +6 -3
  262. data/lib/mongo/operation/create_user.rb +1 -1
  263. data/lib/mongo/operation/create_user/command.rb +4 -4
  264. data/lib/mongo/operation/create_user/op_msg.rb +2 -2
  265. data/lib/mongo/operation/delete.rb +1 -1
  266. data/lib/mongo/operation/delete/bulk_result.rb +3 -1
  267. data/lib/mongo/operation/delete/command.rb +4 -4
  268. data/lib/mongo/operation/delete/legacy.rb +2 -2
  269. data/lib/mongo/operation/delete/op_msg.rb +4 -4
  270. data/lib/mongo/operation/delete/result.rb +5 -2
  271. data/lib/mongo/operation/distinct.rb +1 -1
  272. data/lib/mongo/operation/distinct/command.rb +3 -3
  273. data/lib/mongo/operation/distinct/op_msg.rb +1 -1
  274. data/lib/mongo/operation/drop.rb +1 -1
  275. data/lib/mongo/operation/drop/command.rb +3 -3
  276. data/lib/mongo/operation/drop/op_msg.rb +1 -1
  277. data/lib/mongo/operation/drop_database.rb +1 -1
  278. data/lib/mongo/operation/drop_database/command.rb +3 -3
  279. data/lib/mongo/operation/drop_database/op_msg.rb +1 -1
  280. data/lib/mongo/operation/drop_index.rb +1 -1
  281. data/lib/mongo/operation/drop_index/command.rb +4 -4
  282. data/lib/mongo/operation/drop_index/op_msg.rb +2 -2
  283. data/lib/mongo/operation/explain.rb +1 -1
  284. data/lib/mongo/operation/explain/command.rb +7 -3
  285. data/lib/mongo/operation/explain/legacy.rb +7 -3
  286. data/lib/mongo/operation/explain/op_msg.rb +7 -1
  287. data/lib/mongo/operation/explain/result.rb +4 -1
  288. data/lib/mongo/operation/find.rb +1 -1
  289. data/lib/mongo/operation/find/command.rb +3 -3
  290. data/lib/mongo/operation/find/legacy.rb +3 -3
  291. data/lib/mongo/operation/find/legacy/result.rb +3 -1
  292. data/lib/mongo/operation/find/op_msg.rb +2 -12
  293. data/lib/mongo/operation/find/result.rb +4 -1
  294. data/lib/mongo/operation/get_more.rb +1 -1
  295. data/lib/mongo/operation/get_more/command.rb +3 -3
  296. data/lib/mongo/operation/get_more/legacy.rb +2 -2
  297. data/lib/mongo/operation/get_more/op_msg.rb +2 -12
  298. data/lib/mongo/operation/get_more/result.rb +4 -1
  299. data/lib/mongo/operation/indexes.rb +1 -1
  300. data/lib/mongo/operation/indexes/command.rb +3 -3
  301. data/lib/mongo/operation/indexes/legacy.rb +4 -4
  302. data/lib/mongo/operation/indexes/op_msg.rb +1 -1
  303. data/lib/mongo/operation/indexes/result.rb +6 -1
  304. data/lib/mongo/operation/insert.rb +3 -2
  305. data/lib/mongo/operation/insert/bulk_result.rb +11 -2
  306. data/lib/mongo/operation/insert/command.rb +7 -7
  307. data/lib/mongo/operation/insert/legacy.rb +9 -5
  308. data/lib/mongo/operation/insert/op_msg.rb +6 -6
  309. data/lib/mongo/operation/insert/result.rb +12 -4
  310. data/lib/mongo/operation/kill_cursors.rb +1 -1
  311. data/lib/mongo/operation/kill_cursors/command.rb +3 -3
  312. data/lib/mongo/operation/kill_cursors/legacy.rb +2 -2
  313. data/lib/mongo/operation/kill_cursors/op_msg.rb +1 -1
  314. data/lib/mongo/operation/list_collections.rb +1 -1
  315. data/lib/mongo/operation/list_collections/command.rb +4 -4
  316. data/lib/mongo/operation/list_collections/op_msg.rb +2 -2
  317. data/lib/mongo/operation/list_collections/result.rb +6 -1
  318. data/lib/mongo/operation/map_reduce.rb +1 -1
  319. data/lib/mongo/operation/map_reduce/command.rb +3 -3
  320. data/lib/mongo/operation/map_reduce/op_msg.rb +1 -1
  321. data/lib/mongo/operation/map_reduce/result.rb +11 -1
  322. data/lib/mongo/operation/op_msg_base.rb +3 -3
  323. data/lib/mongo/operation/parallel_scan.rb +1 -1
  324. data/lib/mongo/operation/parallel_scan/command.rb +5 -5
  325. data/lib/mongo/operation/parallel_scan/op_msg.rb +2 -2
  326. data/lib/mongo/operation/parallel_scan/result.rb +5 -1
  327. data/lib/mongo/operation/remove_user.rb +1 -1
  328. data/lib/mongo/operation/remove_user/command.rb +4 -4
  329. data/lib/mongo/operation/remove_user/op_msg.rb +2 -2
  330. data/lib/mongo/operation/result.rb +63 -10
  331. data/lib/mongo/operation/shared/bypass_document_validation.rb +11 -4
  332. data/lib/mongo/operation/shared/causal_consistency_supported.rb +4 -3
  333. data/lib/mongo/operation/shared/collections_info_or_list_collections.rb +58 -0
  334. data/lib/mongo/operation/shared/executable.rb +47 -28
  335. data/lib/mongo/operation/shared/executable_no_validate.rb +3 -3
  336. data/lib/mongo/operation/shared/executable_transaction_label.rb +1 -1
  337. data/lib/mongo/operation/shared/idable.rb +3 -2
  338. data/lib/mongo/operation/shared/limited.rb +11 -2
  339. data/lib/mongo/operation/shared/object_id_generator.rb +2 -1
  340. data/lib/mongo/operation/shared/op_msg_or_command.rb +7 -5
  341. data/lib/mongo/operation/shared/op_msg_or_find_command.rb +8 -6
  342. data/lib/mongo/operation/shared/op_msg_or_list_indexes_command.rb +8 -6
  343. data/lib/mongo/operation/shared/polymorphic_lookup.rb +1 -1
  344. data/lib/mongo/operation/shared/polymorphic_result.rb +1 -1
  345. data/lib/mongo/operation/shared/read_preference_supported.rb +19 -16
  346. data/lib/mongo/operation/shared/response_handling.rb +83 -8
  347. data/lib/mongo/operation/shared/result/aggregatable.rb +2 -1
  348. data/lib/mongo/operation/shared/result/use_legacy_error_parser.rb +1 -1
  349. data/lib/mongo/operation/shared/sessions_supported.rb +51 -27
  350. data/lib/mongo/operation/shared/specifiable.rb +33 -20
  351. data/lib/mongo/operation/shared/write.rb +26 -19
  352. data/lib/mongo/operation/shared/write_concern_supported.rb +7 -6
  353. data/lib/mongo/operation/update.rb +1 -1
  354. data/lib/mongo/operation/update/bulk_result.rb +1 -1
  355. data/lib/mongo/operation/update/command.rb +4 -4
  356. data/lib/mongo/operation/update/legacy.rb +2 -2
  357. data/lib/mongo/operation/update/legacy/result.rb +8 -1
  358. data/lib/mongo/operation/update/op_msg.rb +4 -4
  359. data/lib/mongo/operation/update/result.rb +10 -2
  360. data/lib/mongo/operation/update_user.rb +1 -1
  361. data/lib/mongo/operation/update_user/command.rb +4 -4
  362. data/lib/mongo/operation/update_user/op_msg.rb +2 -2
  363. data/lib/mongo/operation/users_info.rb +1 -1
  364. data/lib/mongo/operation/users_info/command.rb +4 -4
  365. data/lib/mongo/operation/users_info/op_msg.rb +2 -2
  366. data/lib/mongo/operation/users_info/result.rb +4 -1
  367. data/lib/mongo/options.rb +1 -1
  368. data/lib/mongo/options/mapper.rb +1 -1
  369. data/lib/mongo/options/redacted.rb +1 -1
  370. data/lib/mongo/protocol/bit_vector.rb +1 -1
  371. data/lib/mongo/protocol/compressed.rb +1 -1
  372. data/lib/mongo/protocol/delete.rb +1 -1
  373. data/lib/mongo/protocol/get_more.rb +1 -1
  374. data/lib/mongo/protocol/insert.rb +1 -1
  375. data/lib/mongo/protocol/kill_cursors.rb +1 -1
  376. data/lib/mongo/protocol/message.rb +35 -7
  377. data/lib/mongo/protocol/msg.rb +18 -4
  378. data/lib/mongo/protocol/query.rb +1 -1
  379. data/lib/mongo/protocol/registry.rb +1 -1
  380. data/lib/mongo/protocol/reply.rb +1 -1
  381. data/lib/mongo/protocol/serializers.rb +1 -1
  382. data/lib/mongo/protocol/update.rb +1 -1
  383. data/lib/mongo/query_cache.rb +242 -0
  384. data/lib/mongo/retryable.rb +31 -11
  385. data/lib/mongo/semaphore.rb +1 -1
  386. data/lib/mongo/server.rb +45 -9
  387. data/lib/mongo/server/app_metadata.rb +44 -7
  388. data/lib/mongo/server/connection.rb +37 -133
  389. data/lib/mongo/server/connection_base.rb +22 -7
  390. data/lib/mongo/server/connection_common.rb +75 -1
  391. data/lib/mongo/server/connection_pool.rb +23 -1
  392. data/lib/mongo/server/connection_pool/populator.rb +1 -1
  393. data/lib/mongo/server/context.rb +1 -1
  394. data/lib/mongo/server/description.rb +49 -2
  395. data/lib/mongo/server/description/features.rb +12 -3
  396. data/lib/mongo/server/monitor.rb +145 -69
  397. data/lib/mongo/server/monitor/app_metadata.rb +1 -1
  398. data/lib/mongo/server/monitor/connection.rb +110 -80
  399. data/lib/mongo/server/pending_connection.rb +215 -3
  400. data/lib/mongo/server/push_monitor.rb +173 -0
  401. data/{spec/runners/transactions/context.rb → lib/mongo/server/push_monitor/connection.rb} +9 -14
  402. data/lib/mongo/server/round_trip_time_averager.rb +12 -3
  403. data/lib/mongo/server_selector.rb +2 -2
  404. data/lib/mongo/server_selector/{selectable.rb → base.rb} +164 -87
  405. data/lib/mongo/server_selector/nearest.rb +26 -21
  406. data/lib/mongo/server_selector/primary.rb +24 -28
  407. data/lib/mongo/server_selector/primary_preferred.rb +32 -25
  408. data/lib/mongo/server_selector/secondary.rb +26 -21
  409. data/lib/mongo/server_selector/secondary_preferred.rb +29 -34
  410. data/lib/mongo/session.rb +17 -1
  411. data/lib/mongo/session/server_session.rb +1 -1
  412. data/lib/mongo/session/session_pool.rb +1 -1
  413. data/lib/mongo/socket.rb +127 -45
  414. data/lib/mongo/socket/ocsp_cache.rb +97 -0
  415. data/lib/mongo/socket/ocsp_verifier.rb +368 -0
  416. data/lib/mongo/socket/ssl.rb +141 -57
  417. data/lib/mongo/socket/tcp.rb +39 -31
  418. data/lib/mongo/socket/unix.rb +14 -6
  419. data/lib/mongo/srv.rb +1 -1
  420. data/lib/mongo/srv/monitor.rb +8 -14
  421. data/lib/mongo/srv/resolver.rb +15 -11
  422. data/lib/mongo/srv/result.rb +1 -1
  423. data/lib/mongo/timeout.rb +11 -9
  424. data/lib/mongo/topology_version.rb +89 -0
  425. data/lib/mongo/uri.rb +52 -407
  426. data/lib/mongo/uri/options_mapper.rb +582 -0
  427. data/lib/mongo/uri/srv_protocol.rb +12 -3
  428. data/lib/mongo/utils.rb +73 -0
  429. data/lib/mongo/version.rb +2 -2
  430. data/lib/mongo/write_concern.rb +1 -1
  431. data/lib/mongo/write_concern/acknowledged.rb +1 -1
  432. data/lib/mongo/write_concern/base.rb +1 -1
  433. data/lib/mongo/write_concern/unacknowledged.rb +1 -1
  434. data/mongo.gemspec +1 -1
  435. data/spec/NOTES.aws-auth.md +296 -0
  436. data/spec/README.aws-auth.md +318 -0
  437. data/spec/README.md +120 -18
  438. data/spec/integration/auth_spec.rb +29 -9
  439. data/spec/integration/awaited_ismaster_spec.rb +28 -0
  440. data/spec/integration/aws_auth_request_spec.rb +74 -0
  441. data/spec/integration/aws_credentials_retriever_spec.rb +103 -0
  442. data/spec/integration/bulk_write_spec.rb +48 -0
  443. data/spec/integration/change_stream_examples_spec.rb +6 -2
  444. data/spec/integration/change_stream_spec.rb +123 -51
  445. data/spec/integration/check_clean_slate_spec.rb +16 -0
  446. data/spec/integration/client_authentication_options_spec.rb +55 -28
  447. data/spec/integration/client_construction_aws_auth_spec.rb +191 -0
  448. data/spec/integration/client_construction_spec.rb +2 -1
  449. data/spec/integration/collection_indexes_prose_spec.rb +55 -0
  450. data/spec/integration/command_monitoring_spec.rb +30 -6
  451. data/spec/integration/command_spec.rb +11 -9
  452. data/spec/integration/connect_single_rs_name_spec.rb +10 -5
  453. data/spec/integration/connection_pool_populator_spec.rb +3 -1
  454. data/spec/integration/connection_spec.rb +7 -3
  455. data/spec/integration/crud_spec.rb +32 -4
  456. data/spec/integration/cursor_reaping_spec.rb +67 -27
  457. data/spec/integration/docs_examples_spec.rb +6 -0
  458. data/spec/integration/fork_reconnect_spec.rb +143 -0
  459. data/spec/integration/get_more_spec.rb +10 -3
  460. data/spec/integration/heartbeat_events_spec.rb +5 -24
  461. data/spec/integration/ocsp_connectivity_spec.rb +26 -0
  462. data/spec/integration/ocsp_verifier_cache_spec.rb +188 -0
  463. data/spec/integration/ocsp_verifier_spec.rb +334 -0
  464. data/spec/integration/query_cache_spec.rb +1045 -0
  465. data/spec/integration/query_cache_transactions_spec.rb +179 -0
  466. data/spec/integration/read_concern_spec.rb +1 -1
  467. data/spec/integration/read_preference_spec.rb +41 -11
  468. data/spec/integration/reconnect_spec.rb +2 -3
  469. data/spec/integration/retryable_errors_spec.rb +33 -14
  470. data/spec/integration/{retryable_writes_spec.rb → retryable_writes/retryable_writes_36_and_older_spec.rb} +55 -51
  471. data/spec/integration/retryable_writes/retryable_writes_40_and_newer_spec.rb +402 -0
  472. data/spec/integration/retryable_writes/shared/adds_diagnostics.rb +15 -0
  473. data/spec/integration/retryable_writes/shared/does_not_support_retries.rb +24 -0
  474. data/spec/integration/retryable_writes/shared/only_supports_legacy_retries.rb +25 -0
  475. data/spec/integration/retryable_writes/shared/performs_legacy_retries.rb +217 -0
  476. data/spec/integration/retryable_writes/shared/performs_modern_retries.rb +232 -0
  477. data/spec/integration/retryable_writes/shared/performs_no_retries.rb +110 -0
  478. data/spec/integration/retryable_writes/shared/supports_legacy_retries.rb +19 -0
  479. data/spec/integration/retryable_writes/shared/supports_modern_retries.rb +25 -0
  480. data/spec/integration/retryable_writes/shared/supports_retries.rb +16 -0
  481. data/spec/integration/sdam_error_handling_spec.rb +210 -22
  482. data/spec/integration/sdam_events_spec.rb +77 -5
  483. data/spec/integration/sdam_prose_spec.rb +64 -0
  484. data/spec/integration/server_monitor_spec.rb +25 -1
  485. data/spec/integration/server_selection_spec.rb +36 -0
  486. data/spec/integration/server_spec.rb +42 -26
  487. data/spec/integration/size_limit_spec.rb +3 -3
  488. data/spec/integration/srv_monitoring_spec.rb +38 -3
  489. data/spec/integration/srv_spec.rb +56 -0
  490. data/spec/integration/ssl_uri_options_spec.rb +2 -2
  491. data/spec/integration/step_down_spec.rb +15 -15
  492. data/spec/integration/transactions_api_examples_spec.rb +59 -0
  493. data/spec/integration/transactions_examples_spec.rb +5 -2
  494. data/spec/integration/x509_auth_spec.rb +109 -0
  495. data/spec/kerberos/kerberos_spec.rb +10 -6
  496. data/spec/lite_spec_helper.rb +34 -23
  497. data/spec/mongo/address/ipv4_spec.rb +1 -1
  498. data/spec/mongo/address_spec.rb +3 -3
  499. data/spec/mongo/auth/aws/request_region_spec.rb +42 -0
  500. data/spec/mongo/auth/aws/request_spec.rb +76 -0
  501. data/spec/mongo/auth/cr_spec.rb +7 -7
  502. data/spec/mongo/auth/gssapi/conversation_spec.rb +121 -0
  503. data/spec/mongo/auth/invalid_mechanism_spec.rb +1 -1
  504. data/spec/mongo/auth/ldap/conversation_spec.rb +1 -1
  505. data/spec/mongo/auth/ldap_spec.rb +3 -3
  506. data/spec/mongo/auth/scram/conversation_spec.rb +119 -334
  507. data/spec/mongo/auth/scram256/conversation_spec.rb +171 -0
  508. data/spec/mongo/auth/{scram/negotiation_spec.rb → scram_negotiation_spec.rb} +13 -8
  509. data/spec/mongo/auth/scram_spec.rb +29 -69
  510. data/spec/mongo/auth/user_spec.rb +1 -1
  511. data/spec/mongo/auth/x509/conversation_spec.rb +1 -1
  512. data/spec/mongo/auth/x509_spec.rb +8 -8
  513. data/spec/mongo/auth_spec.rb +4 -4
  514. data/spec/mongo/bulk_write_spec.rb +206 -2
  515. data/spec/mongo/caching_cursor_spec.rb +70 -0
  516. data/spec/mongo/client_construction_spec.rb +674 -87
  517. data/spec/mongo/client_spec.rb +99 -5
  518. data/spec/mongo/cluster/topology/replica_set_spec.rb +52 -9
  519. data/spec/mongo/cluster/topology/single_spec.rb +18 -7
  520. data/spec/mongo/cluster_spec.rb +40 -36
  521. data/spec/mongo/collection/view/aggregation_spec.rb +6 -2
  522. data/spec/mongo/collection/view/builder/find_command_spec.rb +17 -6
  523. data/spec/mongo/collection/view/change_stream_resume_spec.rb +392 -0
  524. data/spec/mongo/collection/view/change_stream_spec.rb +0 -318
  525. data/spec/mongo/collection/view/explainable_spec.rb +87 -4
  526. data/spec/mongo/collection/view/iterable_spec.rb +38 -0
  527. data/spec/mongo/collection/view/map_reduce_spec.rb +8 -2
  528. data/spec/mongo/collection/view/readable_spec.rb +15 -1
  529. data/spec/mongo/collection/view/writable_spec.rb +208 -1
  530. data/spec/mongo/collection_spec.rb +567 -43
  531. data/spec/mongo/crypt/auto_decryption_context_spec.rb +1 -1
  532. data/spec/mongo/crypt/auto_encryption_context_spec.rb +1 -1
  533. data/spec/mongo/crypt/explicit_decryption_context_spec.rb +1 -1
  534. data/spec/mongo/crypt/explicit_encryption_context_spec.rb +1 -1
  535. data/spec/mongo/cursor/builder/get_more_command_spec.rb +6 -1
  536. data/spec/mongo/cursor/builder/op_get_more_spec.rb +6 -1
  537. data/spec/mongo/cursor/builder/op_kill_cursors_spec.rb +6 -1
  538. data/spec/mongo/cursor_spec.rb +9 -1
  539. data/spec/mongo/database_spec.rb +295 -10
  540. data/spec/mongo/distinguishing_semaphore_spec.rb +63 -0
  541. data/spec/mongo/error/no_server_available_spec.rb +1 -1
  542. data/spec/mongo/error/operation_failure_heavy_spec.rb +58 -0
  543. data/spec/mongo/error/operation_failure_spec.rb +167 -69
  544. data/spec/mongo/error/unsupported_option_spec.rb +54 -0
  545. data/spec/mongo/grid/fs_bucket_spec.rb +18 -0
  546. data/spec/mongo/grid/stream/write_spec.rb +32 -0
  547. data/spec/mongo/index/view_spec.rb +312 -0
  548. data/spec/mongo/logger_spec.rb +13 -11
  549. data/spec/mongo/monitoring/event/server_description_changed_spec.rb +1 -4
  550. data/spec/mongo/operation/aggregate/result_spec.rb +6 -1
  551. data/spec/mongo/operation/delete/bulk_spec.rb +18 -6
  552. data/spec/mongo/operation/delete/op_msg_spec.rb +22 -14
  553. data/spec/mongo/operation/find/legacy_spec.rb +27 -7
  554. data/spec/mongo/operation/get_more_spec.rb +6 -1
  555. data/spec/mongo/operation/insert/bulk_spec.rb +21 -7
  556. data/spec/mongo/operation/insert/command_spec.rb +4 -0
  557. data/spec/mongo/operation/insert/op_msg_spec.rb +22 -14
  558. data/spec/mongo/operation/limited_spec.rb +5 -3
  559. data/spec/mongo/operation/read_preference_legacy_spec.rb +16 -4
  560. data/spec/mongo/operation/read_preference_op_msg_spec.rb +115 -5
  561. data/spec/mongo/operation/result_spec.rb +6 -1
  562. data/spec/mongo/operation/update/bulk_spec.rb +18 -6
  563. data/spec/mongo/operation/update/command_spec.rb +4 -0
  564. data/spec/mongo/operation/update/op_msg_spec.rb +22 -14
  565. data/spec/mongo/protocol/msg_spec.rb +10 -0
  566. data/spec/mongo/query_cache_spec.rb +279 -0
  567. data/spec/mongo/retryable_spec.rb +71 -70
  568. data/spec/mongo/semaphore_spec.rb +51 -0
  569. data/spec/mongo/server/app_metadata_shared.rb +136 -0
  570. data/spec/mongo/server/app_metadata_spec.rb +8 -1
  571. data/spec/mongo/server/connection_auth_spec.rb +33 -14
  572. data/spec/mongo/server/connection_pool_spec.rb +7 -34
  573. data/spec/mongo/server/connection_spec.rb +131 -77
  574. data/spec/mongo/server/monitor/app_metadata_spec.rb +8 -1
  575. data/spec/mongo/server/monitor/connection_spec.rb +1 -82
  576. data/spec/mongo/server/monitor_spec.rb +76 -17
  577. data/spec/mongo/server/round_trip_time_averager_spec.rb +5 -3
  578. data/spec/mongo/server_selector/nearest_spec.rb +24 -23
  579. data/spec/mongo/server_selector/primary_preferred_spec.rb +27 -26
  580. data/spec/mongo/server_selector/primary_spec.rb +27 -9
  581. data/spec/mongo/server_selector/secondary_preferred_spec.rb +40 -23
  582. data/spec/mongo/server_selector/secondary_spec.rb +19 -18
  583. data/spec/mongo/server_selector_spec.rb +4 -5
  584. data/spec/mongo/session/session_pool_spec.rb +7 -3
  585. data/spec/mongo/session_spec.rb +35 -0
  586. data/spec/mongo/socket/ssl_spec.rb +3 -3
  587. data/spec/mongo/socket/tcp_spec.rb +2 -2
  588. data/spec/mongo/socket/unix_spec.rb +2 -2
  589. data/spec/mongo/socket_spec.rb +10 -10
  590. data/spec/mongo/timeout_spec.rb +22 -68
  591. data/spec/mongo/uri/srv_protocol_spec.rb +64 -33
  592. data/spec/mongo/uri_option_parsing_spec.rb +11 -11
  593. data/spec/mongo/uri_spec.rb +85 -43
  594. data/spec/mongo/utils_spec.rb +39 -0
  595. data/spec/runners/auth.rb +8 -6
  596. data/spec/runners/change_streams/outcome.rb +42 -0
  597. data/spec/runners/change_streams/spec.rb +57 -0
  598. data/spec/runners/change_streams/test.rb +229 -0
  599. data/spec/runners/cmap.rb +1 -1
  600. data/spec/runners/cmap/verifier.rb +1 -1
  601. data/spec/runners/command_monitoring.rb +4 -35
  602. data/spec/runners/connection_string.rb +36 -124
  603. data/spec/runners/crud.rb +2 -2
  604. data/spec/runners/crud/context.rb +10 -6
  605. data/spec/runners/crud/operation.rb +177 -55
  606. data/spec/runners/crud/outcome.rb +1 -1
  607. data/spec/runners/crud/spec.rb +0 -7
  608. data/spec/runners/crud/test.rb +8 -26
  609. data/spec/runners/crud/test_base.rb +47 -0
  610. data/spec/runners/crud/verifier.rb +21 -3
  611. data/spec/runners/gridfs.rb +1 -1
  612. data/spec/runners/{server_discovery_and_monitoring.rb → sdam.rb} +41 -22
  613. data/spec/runners/sdam/verifier.rb +26 -8
  614. data/spec/runners/server_selection.rb +242 -28
  615. data/spec/runners/transactions.rb +13 -14
  616. data/spec/runners/transactions/operation.rb +155 -25
  617. data/spec/runners/transactions/spec.rb +1 -1
  618. data/spec/runners/transactions/test.rb +103 -65
  619. data/spec/spec_tests/auth_spec.rb +2 -0
  620. data/spec/spec_tests/change_streams_spec.rb +39 -4
  621. data/spec/spec_tests/client_side_encryption_spec.rb +3 -0
  622. data/spec/spec_tests/cmap_spec.rb +12 -3
  623. data/spec/spec_tests/command_monitoring_spec.rb +25 -12
  624. data/spec/spec_tests/connection_string_spec.rb +2 -0
  625. data/spec/spec_tests/crud_spec.rb +3 -1
  626. data/spec/spec_tests/data/auth/connection-string.yml +57 -1
  627. data/spec/spec_tests/data/change_streams/change-streams-errors.yml +27 -2
  628. data/spec/spec_tests/data/change_streams/change-streams-resume-errorLabels.yml +1105 -0
  629. data/spec/spec_tests/data/change_streams/change-streams-resume-whitelist.yml +1173 -0
  630. data/spec/spec_tests/data/change_streams/change-streams.yml +5 -5
  631. data/spec/spec_tests/data/cmap/pool-checkout-connection.yml +6 -2
  632. data/spec/spec_tests/data/cmap/pool-create-min-size.yml +3 -0
  633. data/spec/spec_tests/data/connection_string/valid-warnings.yml +24 -0
  634. data/spec/spec_tests/data/crud_v2/aggregate-merge.yml +1 -1
  635. data/spec/spec_tests/data/crud_v2/bulkWrite-arrayFilters.yml +33 -11
  636. data/spec/spec_tests/data/crud_v2/bulkWrite-delete-hint-clientError.yml +63 -0
  637. data/spec/spec_tests/data/crud_v2/bulkWrite-delete-hint-serverError.yml +92 -0
  638. data/spec/spec_tests/data/crud_v2/bulkWrite-delete-hint.yml +103 -0
  639. data/spec/spec_tests/data/crud_v2/bulkWrite-update-hint-clientError.yml +90 -0
  640. data/spec/spec_tests/data/crud_v2/bulkWrite-update-hint-serverError.yml +147 -0
  641. data/spec/spec_tests/data/crud_v2/bulkWrite-update-hint.yml +164 -0
  642. data/spec/spec_tests/data/crud_v2/deleteMany-hint-clientError.yml +43 -0
  643. data/spec/spec_tests/data/crud_v2/deleteMany-hint-serverError.yml +62 -0
  644. data/spec/spec_tests/data/crud_v2/deleteMany-hint.yml +58 -0
  645. data/spec/spec_tests/data/crud_v2/deleteOne-hint-clientError.yml +41 -0
  646. data/spec/spec_tests/data/crud_v2/deleteOne-hint-serverError.yml +60 -0
  647. data/spec/spec_tests/data/crud_v2/deleteOne-hint.yml +57 -0
  648. data/spec/spec_tests/data/crud_v2/find-allowdiskuse-clientError.yml +28 -0
  649. data/spec/spec_tests/data/crud_v2/find-allowdiskuse-serverError.yml +44 -0
  650. data/spec/spec_tests/data/crud_v2/find-allowdiskuse.yml +50 -0
  651. data/spec/spec_tests/data/crud_v2/findOneAndDelete-hint-clientError.yml +45 -0
  652. data/spec/spec_tests/data/crud_v2/findOneAndDelete-hint-serverError.yml +60 -0
  653. data/spec/spec_tests/data/crud_v2/findOneAndDelete-hint.yml +56 -0
  654. data/spec/spec_tests/data/crud_v2/findOneAndReplace-hint-clientError.yml +40 -0
  655. data/spec/spec_tests/data/crud_v2/findOneAndReplace-hint-serverError.yml +59 -0
  656. data/spec/spec_tests/data/crud_v2/findOneAndReplace-hint.yml +55 -0
  657. data/spec/spec_tests/data/crud_v2/findOneAndUpdate-hint-clientError.yml +40 -0
  658. data/spec/spec_tests/data/crud_v2/findOneAndUpdate-hint-serverError.yml +58 -0
  659. data/spec/spec_tests/data/crud_v2/findOneAndUpdate-hint.yml +55 -0
  660. data/spec/spec_tests/data/crud_v2/replaceOne-hint.yml +61 -0
  661. data/spec/spec_tests/data/crud_v2/unacknowledged-bulkWrite-delete-hint-clientError.yml +60 -0
  662. data/spec/spec_tests/data/crud_v2/unacknowledged-bulkWrite-update-hint-clientError.yml +88 -0
  663. data/spec/spec_tests/data/crud_v2/unacknowledged-deleteMany-hint-clientError.yml +40 -0
  664. data/spec/spec_tests/data/crud_v2/unacknowledged-deleteOne-hint-clientError.yml +38 -0
  665. data/spec/spec_tests/data/crud_v2/unacknowledged-findOneAndDelete-hint-clientError.yml +42 -0
  666. data/spec/spec_tests/data/crud_v2/unacknowledged-findOneAndReplace-hint-clientError.yml +40 -0
  667. data/spec/spec_tests/data/crud_v2/unacknowledged-findOneAndUpdate-hint-clientError.yml +40 -0
  668. data/spec/spec_tests/data/crud_v2/unacknowledged-replaceOne-hint-clientError.yml +40 -0
  669. data/spec/spec_tests/data/crud_v2/unacknowledged-updateMany-hint-clientError.yml +43 -0
  670. data/spec/spec_tests/data/crud_v2/unacknowledged-updateOne-hint-clientError.yml +40 -0
  671. data/spec/spec_tests/data/crud_v2/updateMany-hint-clientError.yml +45 -0
  672. data/spec/spec_tests/data/crud_v2/updateMany-hint-serverError.yml +66 -0
  673. data/spec/spec_tests/data/crud_v2/updateMany-hint.yml +65 -0
  674. data/spec/spec_tests/data/crud_v2/updateOne-hint-clientError.yml +43 -0
  675. data/spec/spec_tests/data/crud_v2/updateOne-hint-serverError.yml +62 -0
  676. data/spec/spec_tests/data/crud_v2/updateOne-hint.yml +61 -0
  677. data/spec/spec_tests/data/crud_v2/updateWithPipelines.yml +65 -0
  678. data/spec/spec_tests/data/dns_seedlist_discovery/direct-connection-false.yml +10 -0
  679. data/spec/spec_tests/data/dns_seedlist_discovery/direct-connection-true.yml +5 -0
  680. data/spec/spec_tests/data/max_staleness/ReplicaSetNoPrimary/MaxStalenessTooSmall.yml +15 -0
  681. data/spec/spec_tests/data/max_staleness/ReplicaSetNoPrimary/NoKnownServers.yml +4 -3
  682. data/spec/spec_tests/data/max_staleness/Unknown/SmallMaxStaleness.yml +1 -0
  683. data/spec/spec_tests/data/read_write_concern/operation/default-write-concern-2.6.yml +215 -0
  684. data/spec/spec_tests/data/read_write_concern/operation/default-write-concern-3.2.yml +58 -0
  685. data/spec/spec_tests/data/read_write_concern/operation/default-write-concern-3.4.yml +95 -0
  686. data/spec/spec_tests/data/read_write_concern/operation/default-write-concern-4.2.yml +36 -0
  687. data/spec/spec_tests/data/retryable_writes/bulkWrite-errorLabels.yml +77 -0
  688. data/spec/spec_tests/data/retryable_writes/bulkWrite-serverErrors.yml +37 -0
  689. data/spec/spec_tests/data/retryable_writes/deleteOne-errorLabels.yml +48 -0
  690. data/spec/spec_tests/data/retryable_writes/deleteOne-serverErrors.yml +22 -0
  691. data/spec/spec_tests/data/retryable_writes/findOneAndDelete-errorLabels.yml +49 -0
  692. data/spec/spec_tests/data/retryable_writes/findOneAndDelete-serverErrors.yml +23 -0
  693. data/spec/spec_tests/data/retryable_writes/findOneAndReplace-errorLabels.yml +52 -0
  694. data/spec/spec_tests/data/retryable_writes/findOneAndReplace-serverErrors.yml +25 -0
  695. data/spec/spec_tests/data/retryable_writes/findOneAndUpdate-errorLabels.yml +52 -0
  696. data/spec/spec_tests/data/retryable_writes/findOneAndUpdate-serverErrors.yml +24 -0
  697. data/spec/spec_tests/data/retryable_writes/insertMany-errorLabels.yml +54 -0
  698. data/spec/spec_tests/data/retryable_writes/insertMany-serverErrors.yml +24 -0
  699. data/spec/spec_tests/data/retryable_writes/insertOne-errorLabels.yml +44 -0
  700. data/spec/spec_tests/data/retryable_writes/insertOne-serverErrors.yml +69 -0
  701. data/spec/spec_tests/data/retryable_writes/replaceOne-errorLabels.yml +53 -0
  702. data/spec/spec_tests/data/retryable_writes/replaceOne-serverErrors.yml +23 -0
  703. data/spec/spec_tests/data/retryable_writes/updateOne-errorLabels.yml +53 -0
  704. data/spec/spec_tests/data/retryable_writes/updateOne-serverErrors.yml +23 -0
  705. data/spec/spec_tests/data/sdam/errors/error_handling_handshake.yml +54 -0
  706. data/spec/spec_tests/data/sdam/errors/non-stale-network-error.yml +46 -0
  707. data/spec/spec_tests/data/sdam/errors/non-stale-network-timeout-error.yml +37 -0
  708. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-greater-InterruptedAtShutdown.yml +60 -0
  709. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-greater-InterruptedDueToReplStateChange.yml +60 -0
  710. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-greater-NotMaster.yml +60 -0
  711. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-greater-NotMasterNoSlaveOk.yml +60 -0
  712. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-greater-NotMasterOrSecondary.yml +60 -0
  713. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-greater-PrimarySteppedDown.yml +60 -0
  714. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-greater-ShutdownInProgress.yml +60 -0
  715. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-missing-InterruptedAtShutdown.yml +51 -0
  716. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-missing-InterruptedDueToReplStateChange.yml +51 -0
  717. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-missing-NotMaster.yml +51 -0
  718. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-missing-NotMasterNoSlaveOk.yml +51 -0
  719. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-missing-NotMasterOrSecondary.yml +51 -0
  720. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-missing-PrimarySteppedDown.yml +51 -0
  721. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-missing-ShutdownInProgress.yml +51 -0
  722. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-proccessId-changed-InterruptedAtShutdown.yml +60 -0
  723. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-proccessId-changed-InterruptedDueToReplStateChange.yml +60 -0
  724. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-proccessId-changed-NotMaster.yml +60 -0
  725. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-proccessId-changed-NotMasterNoSlaveOk.yml +60 -0
  726. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-proccessId-changed-NotMasterOrSecondary.yml +60 -0
  727. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-proccessId-changed-PrimarySteppedDown.yml +60 -0
  728. data/spec/spec_tests/data/sdam/errors/non-stale-topologyVersion-proccessId-changed-ShutdownInProgress.yml +60 -0
  729. data/spec/spec_tests/data/sdam/errors/post-42-InterruptedAtShutdown.yml +46 -0
  730. data/spec/spec_tests/data/sdam/errors/post-42-InterruptedDueToReplStateChange.yml +46 -0
  731. data/spec/spec_tests/data/sdam/errors/post-42-NotMaster.yml +46 -0
  732. data/spec/spec_tests/data/sdam/errors/post-42-NotMasterNoSlaveOk.yml +46 -0
  733. data/spec/spec_tests/data/sdam/errors/post-42-NotMasterOrSecondary.yml +46 -0
  734. data/spec/spec_tests/data/sdam/errors/post-42-PrimarySteppedDown.yml +46 -0
  735. data/spec/spec_tests/data/sdam/errors/post-42-ShutdownInProgress.yml +46 -0
  736. data/spec/spec_tests/data/sdam/errors/pre-42-InterruptedAtShutdown.yml +46 -0
  737. data/spec/spec_tests/data/sdam/errors/pre-42-InterruptedDueToReplStateChange.yml +46 -0
  738. data/spec/spec_tests/data/sdam/errors/pre-42-NotMaster.yml +46 -0
  739. data/spec/spec_tests/data/sdam/errors/pre-42-NotMasterNoSlaveOk.yml +46 -0
  740. data/spec/spec_tests/data/sdam/errors/pre-42-NotMasterOrSecondary.yml +46 -0
  741. data/spec/spec_tests/data/sdam/errors/pre-42-PrimarySteppedDown.yml +46 -0
  742. data/spec/spec_tests/data/sdam/errors/pre-42-ShutdownInProgress.yml +46 -0
  743. data/spec/spec_tests/data/sdam/errors/stale-generation-InterruptedAtShutdown.yml +89 -0
  744. data/spec/spec_tests/data/sdam/errors/stale-generation-InterruptedDueToReplStateChange.yml +89 -0
  745. data/spec/spec_tests/data/sdam/errors/stale-generation-NotMaster.yml +89 -0
  746. data/spec/spec_tests/data/sdam/errors/stale-generation-NotMasterNoSlaveOk.yml +89 -0
  747. data/spec/spec_tests/data/sdam/errors/stale-generation-NotMasterOrSecondary.yml +89 -0
  748. data/spec/spec_tests/data/sdam/errors/stale-generation-PrimarySteppedDown.yml +89 -0
  749. data/spec/spec_tests/data/sdam/errors/stale-generation-ShutdownInProgress.yml +89 -0
  750. data/spec/spec_tests/data/sdam/errors/stale-generation-afterHandshakeCompletes-InterruptedAtShutdown.yml +89 -0
  751. data/spec/spec_tests/data/sdam/errors/stale-generation-afterHandshakeCompletes-InterruptedDueToReplStateChange.yml +89 -0
  752. data/spec/spec_tests/data/sdam/errors/stale-generation-afterHandshakeCompletes-NotMaster.yml +89 -0
  753. data/spec/spec_tests/data/sdam/errors/stale-generation-afterHandshakeCompletes-NotMasterNoSlaveOk.yml +89 -0
  754. data/spec/spec_tests/data/sdam/errors/stale-generation-afterHandshakeCompletes-NotMasterOrSecondary.yml +89 -0
  755. data/spec/spec_tests/data/sdam/errors/stale-generation-afterHandshakeCompletes-PrimarySteppedDown.yml +89 -0
  756. data/spec/spec_tests/data/sdam/errors/stale-generation-afterHandshakeCompletes-ShutdownInProgress.yml +89 -0
  757. data/spec/spec_tests/data/sdam/errors/stale-generation-afterHandshakeCompletes-network.yml +80 -0
  758. data/spec/spec_tests/data/sdam/errors/stale-generation-afterHandshakeCompletes-timeout.yml +80 -0
  759. data/spec/spec_tests/data/sdam/errors/stale-generation-beforeHandshakeCompletes-InterruptedAtShutdown.yml +89 -0
  760. data/spec/spec_tests/data/sdam/errors/stale-generation-beforeHandshakeCompletes-InterruptedDueToReplStateChange.yml +89 -0
  761. data/spec/spec_tests/data/sdam/errors/stale-generation-beforeHandshakeCompletes-NotMaster.yml +89 -0
  762. data/spec/spec_tests/data/sdam/errors/stale-generation-beforeHandshakeCompletes-NotMasterNoSlaveOk.yml +89 -0
  763. data/spec/spec_tests/data/sdam/errors/stale-generation-beforeHandshakeCompletes-NotMasterOrSecondary.yml +89 -0
  764. data/spec/spec_tests/data/sdam/errors/stale-generation-beforeHandshakeCompletes-PrimarySteppedDown.yml +89 -0
  765. data/spec/spec_tests/data/sdam/errors/stale-generation-beforeHandshakeCompletes-ShutdownInProgress.yml +89 -0
  766. data/spec/spec_tests/data/sdam/errors/stale-generation-beforeHandshakeCompletes-network.yml +80 -0
  767. data/spec/spec_tests/data/sdam/errors/stale-generation-beforeHandshakeCompletes-timeout.yml +80 -0
  768. data/spec/spec_tests/data/sdam/errors/stale-topologyVersion-InterruptedAtShutdown.yml +64 -0
  769. data/spec/spec_tests/data/sdam/errors/stale-topologyVersion-InterruptedDueToReplStateChange.yml +64 -0
  770. data/spec/spec_tests/data/sdam/errors/stale-topologyVersion-NotMaster.yml +64 -0
  771. data/spec/spec_tests/data/sdam/errors/stale-topologyVersion-NotMasterNoSlaveOk.yml +64 -0
  772. data/spec/spec_tests/data/sdam/errors/stale-topologyVersion-NotMasterOrSecondary.yml +64 -0
  773. data/spec/spec_tests/data/sdam/errors/stale-topologyVersion-PrimarySteppedDown.yml +64 -0
  774. data/spec/spec_tests/data/sdam/errors/stale-topologyVersion-ShutdownInProgress.yml +64 -0
  775. data/spec/spec_tests/data/sdam/rs/compatible.yml +2 -0
  776. data/spec/spec_tests/data/sdam/rs/compatible_unknown.yml +2 -0
  777. data/spec/spec_tests/data/sdam/rs/discover_arbiters.yml +2 -2
  778. data/spec/spec_tests/data/sdam/rs/discover_arbiters_replicaset.yml +43 -0
  779. data/spec/spec_tests/data/sdam/rs/discover_ghost.yml +35 -0
  780. data/spec/spec_tests/data/sdam/rs/{ghost_discovered.yml → discover_ghost_replicaset.yml} +1 -1
  781. data/spec/spec_tests/data/sdam/rs/discover_hidden.yml +50 -0
  782. data/spec/spec_tests/data/sdam/rs/discover_hidden_replicaset.yml +50 -0
  783. data/spec/spec_tests/data/sdam/rs/discover_passives.yml +2 -2
  784. data/spec/spec_tests/data/sdam/rs/discover_passives_replicaset.yml +81 -0
  785. data/spec/spec_tests/data/sdam/rs/discover_primary.yml +2 -2
  786. data/spec/spec_tests/data/sdam/rs/discover_primary_replicaset.yml +42 -0
  787. data/spec/spec_tests/data/sdam/rs/discover_rsother.yml +49 -0
  788. data/spec/spec_tests/data/sdam/rs/{rsother_discovered.yml → discover_rsother_replicaset.yml} +1 -1
  789. data/spec/spec_tests/data/sdam/rs/discover_secondary.yml +2 -2
  790. data/spec/spec_tests/data/sdam/rs/discover_secondary_replicaset.yml +43 -0
  791. data/spec/spec_tests/data/sdam/rs/incompatible_arbiter.yml +2 -0
  792. data/spec/spec_tests/data/sdam/rs/incompatible_ghost.yml +2 -0
  793. data/spec/spec_tests/data/sdam/rs/incompatible_other.yml +2 -0
  794. data/spec/spec_tests/data/sdam/rs/primary_mismatched_me.yml +23 -27
  795. data/spec/spec_tests/data/sdam/rs/primary_to_no_primary_mismatched_me.yml +79 -55
  796. data/spec/spec_tests/data/sdam/rs/replicaset_rsnp.yml +20 -0
  797. data/spec/spec_tests/data/sdam/rs/secondary_mismatched_me.yml +3 -2
  798. data/spec/spec_tests/data/sdam/rs/too_new.yml +2 -0
  799. data/spec/spec_tests/data/sdam/rs/topology_version_equal.yml +66 -0
  800. data/spec/spec_tests/data/sdam/rs/topology_version_greater.yml +189 -0
  801. data/spec/spec_tests/data/sdam/rs/topology_version_less.yml +62 -0
  802. data/spec/spec_tests/data/sdam/sharded/discover_single_mongos.yml +23 -0
  803. data/spec/spec_tests/data/sdam/single/direct_connection_external_ip.yml +1 -1
  804. data/spec/spec_tests/data/sdam/single/direct_connection_mongos.yml +2 -2
  805. data/spec/spec_tests/data/sdam/single/direct_connection_replicaset.yml +22 -0
  806. data/spec/spec_tests/data/sdam/single/direct_connection_rsarbiter.yml +2 -2
  807. data/spec/spec_tests/data/sdam/single/direct_connection_rsprimary.yml +2 -2
  808. data/spec/spec_tests/data/sdam/single/direct_connection_rssecondary.yml +2 -2
  809. data/spec/spec_tests/data/sdam/single/direct_connection_slave.yml +1 -1
  810. data/spec/spec_tests/data/sdam/single/direct_connection_standalone.yml +2 -2
  811. data/spec/spec_tests/data/sdam/single/{unavailable_seed.yml → direct_connection_unavailable_seed.yml} +2 -2
  812. data/spec/spec_tests/data/sdam/single/direct_connection_wrong_set_name.yml +38 -0
  813. data/spec/spec_tests/data/sdam/single/discover_standalone.yml +34 -0
  814. data/spec/spec_tests/data/sdam/single/discover_unavailable_seed.yml +28 -0
  815. data/spec/spec_tests/data/sdam/single/too_old_then_upgraded.yml +46 -0
  816. data/spec/spec_tests/data/sdam_integration/cancel-server-check.yml +96 -0
  817. data/spec/spec_tests/data/sdam_integration/connectTimeoutMS.yml +88 -0
  818. data/spec/spec_tests/data/sdam_integration/find-network-error.yml +83 -0
  819. data/spec/spec_tests/data/sdam_integration/find-shutdown-error.yml +116 -0
  820. data/spec/spec_tests/data/sdam_integration/insert-network-error.yml +86 -0
  821. data/spec/spec_tests/data/sdam_integration/insert-shutdown-error.yml +115 -0
  822. data/spec/spec_tests/data/sdam_integration/isMaster-command-error.yml +168 -0
  823. data/spec/spec_tests/data/sdam_integration/isMaster-network-error.yml +162 -0
  824. data/spec/spec_tests/data/sdam_integration/isMaster-timeout.yml +229 -0
  825. data/spec/spec_tests/data/sdam_integration/rediscover-quickly-after-step-down.yml +87 -0
  826. data/spec/spec_tests/data/sdam_monitoring/discovered_standalone.yml +1 -3
  827. data/spec/spec_tests/data/sdam_monitoring/standalone.yml +2 -2
  828. data/spec/spec_tests/data/sdam_monitoring/standalone_repeated.yml +2 -2
  829. data/spec/spec_tests/data/sdam_monitoring/standalone_suppress_equal_description_changes.yml +2 -2
  830. data/spec/spec_tests/data/sdam_monitoring/standalone_to_rs_with_me_mismatch.yml +2 -2
  831. data/spec/spec_tests/data/transactions/create-collection.yml +131 -0
  832. data/spec/spec_tests/data/transactions/create-index.yml +152 -0
  833. data/spec/spec_tests/data/transactions/error-labels.yml +87 -21
  834. data/spec/spec_tests/data/transactions/mongos-recovery-token.yml +1 -0
  835. data/spec/spec_tests/data/transactions/retryable-abort-errorLabels.yml +124 -0
  836. data/spec/spec_tests/data/transactions/retryable-abort.yml +17 -2
  837. data/spec/spec_tests/data/transactions/retryable-commit-errorLabels.yml +132 -0
  838. data/spec/spec_tests/data/transactions/retryable-commit.yml +24 -9
  839. data/spec/spec_tests/data/uri_options/auth-options.yml +25 -0
  840. data/spec/spec_tests/data/uri_options/compression-options.yml +6 -3
  841. data/spec/spec_tests/data/uri_options/connection-options.yml +43 -0
  842. data/spec/spec_tests/data/uri_options/read-preference-options.yml +24 -0
  843. data/spec/spec_tests/data/uri_options/ruby-auth-options.yml +12 -0
  844. data/spec/spec_tests/data/uri_options/ruby-connection-options.yml +58 -0
  845. data/spec/spec_tests/data/uri_options/tls-options.yml +160 -4
  846. data/spec/spec_tests/dns_seedlist_discovery_spec.rb +12 -2
  847. data/spec/spec_tests/gridfs_spec.rb +2 -0
  848. data/spec/spec_tests/max_staleness_spec.rb +5 -141
  849. data/spec/spec_tests/read_write_concern_connection_string_spec.rb +2 -0
  850. data/spec/spec_tests/read_write_concern_operaton_spec.rb +10 -0
  851. data/spec/spec_tests/retryable_reads_spec.rb +4 -2
  852. data/spec/spec_tests/retryable_writes_spec.rb +8 -1
  853. data/spec/spec_tests/sdam_integration_spec.rb +13 -0
  854. data/spec/spec_tests/sdam_monitoring_spec.rb +3 -2
  855. data/spec/spec_tests/sdam_spec.rb +70 -1
  856. data/spec/spec_tests/server_selection_rtt_spec.rb +2 -0
  857. data/spec/spec_tests/server_selection_spec.rb +5 -115
  858. data/spec/spec_tests/transactions_api_spec.rb +5 -0
  859. data/spec/spec_tests/transactions_spec.rb +5 -0
  860. data/spec/spec_tests/uri_options_spec.rb +33 -33
  861. data/spec/stress/cleanup_spec.rb +58 -0
  862. data/spec/stress/connection_pool_stress_spec.rb +11 -13
  863. data/spec/stress/connection_pool_timing_spec.rb +3 -6
  864. data/spec/stress/fork_reconnect_stress_spec.rb +109 -0
  865. data/spec/support/authorization.rb +1 -11
  866. data/spec/support/aws_utils.rb +62 -0
  867. data/spec/support/aws_utils/base.rb +134 -0
  868. data/spec/support/aws_utils/inspector.rb +224 -0
  869. data/spec/support/aws_utils/orchestrator.rb +370 -0
  870. data/spec/support/aws_utils/provisioner.rb +360 -0
  871. data/spec/support/background_thread_registry.rb +6 -2
  872. data/spec/support/certificates/atlas-ocsp-ca.crt +28 -0
  873. data/spec/support/certificates/atlas-ocsp.crt +41 -0
  874. data/spec/support/child_process_helper.rb +78 -0
  875. data/spec/support/client_registry.rb +6 -24
  876. data/spec/support/client_registry_macros.rb +11 -2
  877. data/spec/support/cluster_config.rb +5 -0
  878. data/spec/support/common_shortcuts.rb +88 -0
  879. data/spec/support/constraints.rb +50 -7
  880. data/spec/support/crypt.rb +1 -1
  881. data/spec/support/event_subscriber.rb +184 -84
  882. data/spec/support/keyword_struct.rb +26 -0
  883. data/spec/support/lite_constraints.rb +71 -0
  884. data/spec/support/matchers.rb +16 -0
  885. data/spec/support/ocsp +1 -0
  886. data/spec/support/session_registry.rb +52 -0
  887. data/spec/support/shared/scram_conversation.rb +100 -0
  888. data/spec/support/shared/server_selector.rb +93 -1
  889. data/spec/support/shared/session.rb +29 -21
  890. data/spec/support/spec_config.rb +98 -21
  891. data/spec/support/spec_organizer.rb +129 -0
  892. data/spec/support/spec_setup.rb +9 -5
  893. data/spec/support/utils.rb +180 -25
  894. metadata +1221 -745
  895. metadata.gz.sig +0 -0
  896. data/lib/mongo/server/connectable.rb +0 -107
  897. data/spec/runners/change_streams.rb +0 -262
  898. data/spec/runners/change_streams/operation.rb +0 -89
  899. data/spec/runners/sdam_monitoring.rb +0 -89
  900. data/spec/spec_tests/data/sdam/sharded/ruby_discovered_single_mongos.yml +0 -27
@@ -0,0 +1,224 @@
1
+ module AwsUtils
2
+ class Inspector < Base
3
+
4
+ def list_key_pairs
5
+ ec2_client.describe_key_pairs.key_pairs.each do |key_pair|
6
+ puts key_pair.key_name
7
+ end
8
+ end
9
+
10
+ def assume_role_arn
11
+ assume_role = detect_object(iam_client.list_roles, :roles, :role_name, AWS_AUTH_ASSUME_ROLE_NAME)
12
+ if assume_role.nil?
13
+ raise 'No user found, please run `aws setup-resources`'
14
+ end
15
+ assume_role.arn
16
+ end
17
+
18
+ def ecs_status(cluster_name: AWS_AUTH_ECS_CLUSTER_NAME,
19
+ service_name: AWS_AUTH_ECS_SERVICE_NAME,
20
+ get_public_ip: true, get_logs: true
21
+ )
22
+ service = ecs_client.describe_services(
23
+ cluster: cluster_name,
24
+ services: [service_name],
25
+ ).services.first
26
+ if service.nil?
27
+ raise "No service #{service_name} in cluster #{cluster_name} - provision first"
28
+ end
29
+
30
+ # When Ruby driver tooling is used, task definition generation is
31
+ # going up on each service launch, and service name is the fixed.
32
+ # When testing in Evergreen, generation is fixed because we do not
33
+ # change the task definition, but service name is different for
34
+ # each test run.
35
+ if service.task_definition =~ /:(\d+)$/
36
+ generation = $1
37
+ puts "Current task definition generation: #{generation} for service: #{service_name}"
38
+ else
39
+ raise 'Could not determine task definition generation'
40
+ end
41
+
42
+ colors = {
43
+ 'running' => :green,
44
+ 'pending' => :yellow,
45
+ 'stopped' => :red,
46
+ }
47
+ # Pending status in the API includes tasks in provisioning status as
48
+ # show in the AWS console.
49
+ #
50
+ # The API returns the tasks unordered, in particular the latest task
51
+ # may be in the middle of the list following relatively ancient tasks.
52
+ # Collect all tasks in a single list and order them by generation.
53
+ # We expect to have a single task per generation.
54
+ tasks = []
55
+ %w(running pending stopped).each do |status|
56
+ resp = ecs_client.list_tasks(
57
+ cluster: cluster_name,
58
+ service_name: service_name,
59
+ desired_status: status,
60
+ )
61
+ task_arns = resp.map(&:task_arns).flatten
62
+ if task_arns.empty?
63
+ next
64
+ end
65
+ ecs_client.describe_tasks(
66
+ cluster: cluster_name,
67
+ tasks: task_arns,
68
+ ).each do |tbatch|
69
+ unless tbatch.failures.empty?
70
+ # The task list endpoint does not raise an exception if it can't
71
+ # find the tasks, but reports "failures".
72
+ puts "Failures for #{task_arns.join(', ')}:"
73
+ tbatch.failures.each do |failure|
74
+ puts "#{failure.arn}: #{failure.reason}"
75
+ next
76
+ end
77
+ end
78
+ tbatch.tasks.each do |task|
79
+ tasks << task
80
+ end
81
+ end
82
+ end
83
+
84
+ tasks.each do |task|
85
+ class << task
86
+ def generation
87
+ @generation ||= if task_definition_arn =~ /:(\d+)$/
88
+ $1.to_i
89
+ else
90
+ raise 'Could not determine generation'
91
+ end
92
+ end
93
+
94
+ def task_uuid
95
+ @uuid ||= task_arn.split('/').last
96
+ end
97
+ end
98
+ end
99
+
100
+ tasks = tasks.sort_by do |task|
101
+ -task.generation
102
+ end.first(3)
103
+
104
+ running_task = nil
105
+ running_private_ip = nil
106
+ running_public_ip = nil
107
+
108
+ if tasks.empty?
109
+ puts 'No tasks in the cluster'
110
+ end
111
+ tasks.each do |task|
112
+ status = task.last_status.downcase
113
+
114
+ status_ext = case status
115
+ when 'stopped'
116
+ ": #{task.stopped_reason}"
117
+ else
118
+ ''
119
+ end
120
+ decorated_status = Paint[status.upcase, colors[status]]
121
+ puts "Task for generation #{task.generation}: #{decorated_status}#{status_ext} (uuid: #{task.task_uuid})"
122
+ if status == 'running'
123
+ puts "Task ARN: #{task.task_arn}"
124
+ running_task ||= task
125
+ end
126
+ task.containers.each do |container|
127
+ if container.reason
128
+ puts container.reason
129
+ end
130
+ end
131
+
132
+ if status == 'running'
133
+ attachment = detect_object([task], :attachments, :type, 'ElasticNetworkInterface')
134
+ ip = detect_object([attachment], :details, :name, 'privateIPv4Address')
135
+ if ip
136
+ private_ip = ip.value
137
+ running_private_ip ||= private_ip
138
+ end
139
+ msg = "Private IP: #{private_ip}"
140
+ if get_public_ip
141
+ niid = detect_object([attachment], :details, :name, 'networkInterfaceId')
142
+ network_interface = ec2_client.describe_network_interfaces(
143
+ network_interface_ids: [niid.value],
144
+ ).network_interfaces.first
145
+ public_ip = network_interface&.association&.public_ip
146
+ running_public_ip ||= public_ip
147
+ msg += ", public IP: #{public_ip}"
148
+ end
149
+ puts msg
150
+ end
151
+ puts
152
+ end
153
+
154
+ puts
155
+ task_ids = []
156
+ max_event_count = 5
157
+ event_count = 0
158
+ service = ecs_client.describe_services(
159
+ cluster: cluster_name,
160
+ services: [service_name],
161
+ ).services.first
162
+ if service.nil?
163
+ puts 'Service is missing'
164
+ else
165
+ if service.events.empty?
166
+ puts 'No events for service'
167
+ else
168
+ puts "Events for #{service.service_arn}:"
169
+ service.events.each do |event|
170
+ event_count += 1
171
+ break if event_count > max_event_count
172
+ if event.message =~ /\(task (\w+)\)/
173
+ task_ids << $1
174
+ end
175
+ puts "#{event.created_at.strftime('%Y-%m-%d %H:%M:%S %z')} #{event.message}"
176
+ end
177
+ end
178
+ end
179
+
180
+ if get_logs && running_task
181
+ puts
182
+ log_stream_name = "task/ssh/#{running_task.task_uuid}"
183
+ log_stream = logs_client.describe_log_streams(
184
+ log_group_name: AWS_AUTH_ECS_LOG_GROUP,
185
+ log_stream_name_prefix: log_stream_name,
186
+ ).log_streams.first
187
+ if log_stream
188
+ log_events = logs_client.get_log_events(
189
+ log_group_name: AWS_AUTH_ECS_LOG_GROUP,
190
+ log_stream_name: log_stream_name,
191
+ end_time: Time.now.to_i * 1000,
192
+ limit: 100,
193
+ ).events
194
+ if log_events.any?
195
+ puts "Task logs for task #{running_task.task_uuid}:"
196
+ log_events.each do |event|
197
+ puts "[#{Time.at(event.timestamp/1000r).strftime('%Y-%m-%d %H:%M:%S %z')}] #{event.message}"
198
+ end
199
+ else
200
+ puts "No CloudWatch events in the log stream for task #{running_task.task_uuid}"
201
+ end
202
+ else
203
+ puts "No CloudWatch log stream for task #{running_task.task_uuid}"
204
+ end
205
+ end
206
+
207
+ if running_public_ip
208
+ puts
209
+ puts "ip=#{running_public_ip}"
210
+ puts "ssh -o StrictHostKeyChecking=false root@#{running_public_ip}"
211
+ end
212
+
213
+ {
214
+ private_ip: running_private_ip,
215
+ }
216
+ end
217
+
218
+ private
219
+
220
+ def ucfirst(str)
221
+ str[0].upcase + str[1...str.length]
222
+ end
223
+ end
224
+ end
@@ -0,0 +1,370 @@
1
+ module AwsUtils
2
+ class Orchestrator < Base
3
+
4
+ def assume_role(role_arn)
5
+ # https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/STS/Client.html#assume_role-instance_method
6
+ resp = sts_client.assume_role(
7
+ role_arn: role_arn,
8
+ role_session_name: "#{NAMESPACE}.test",
9
+ )
10
+ resp.credentials
11
+ end
12
+
13
+ def set_instance_profile(instance_id,
14
+ instance_profile_name: AWS_AUTH_INSTANCE_PROFILE_NAME,
15
+ instance_profile_arn: nil
16
+ )
17
+ clear_instance_profile(instance_id)
18
+
19
+ deadline = Time.now + 30
20
+ begin
21
+ ec2_client.associate_iam_instance_profile(
22
+ iam_instance_profile: {
23
+ name: instance_profile_name,
24
+ arn: instance_profile_arn,
25
+ },
26
+ instance_id: instance_id,
27
+ )
28
+ rescue Aws::EC2::Errors::RequestLimitExceeded => e
29
+ if Time.now >= deadline
30
+ raise
31
+ end
32
+ STDERR.puts("AWS request limit exceeded: #{e.class}: #{e}, will retry")
33
+ sleep 5
34
+ retry
35
+ end
36
+ end
37
+
38
+ def clear_instance_profile(instance_id)
39
+ assoc = detect_object(ec2_client.describe_iam_instance_profile_associations,
40
+ :iam_instance_profile_associations, :instance_id, instance_id)
41
+
42
+ if assoc
43
+ deadline = Time.now + 30
44
+ begin
45
+ ec2_client.disassociate_iam_instance_profile(
46
+ association_id: assoc.association_id,
47
+ )
48
+ rescue Aws::EC2::Errors::RequestLimitExceeded => e
49
+ if Time.now >= deadline
50
+ raise
51
+ end
52
+ STDERR.puts("AWS request limit exceeded: #{e.class}: #{e}, will retry")
53
+ sleep 5
54
+ retry
55
+ end
56
+ end
57
+ end
58
+
59
+ def provision_auth_ec2_instance(key_pair_name: nil, public_key_path: nil,
60
+ distro: 'ubuntu1604'
61
+ )
62
+ security_group_id = ssh_security_group_id!
63
+ reservations = ec2_client.describe_instances(filters: [
64
+ {name: 'tag:name', values: [AWS_AUTH_EC2_INSTANCE_NAME]},
65
+ ]).reservations
66
+ instance = find_running_instance(reservations)
67
+ if instance.nil?
68
+ ami_name = AWS_AUTH_EC2_AMI_NAMES.fetch(distro)
69
+ image = ec2_client.describe_images(
70
+ filters: [{name: 'name', values: [ami_name]}],
71
+ ).images.first
72
+ if public_key_path
73
+ public_key = File.read(public_key_path)
74
+ user_data = Base64.encode64(<<-CMD)
75
+ #!/bin/sh
76
+ for user in `ls /home`; do
77
+ cd /home/$user &&
78
+ mkdir -p .ssh &&
79
+ chmod 0700 .ssh &&
80
+ chown $user:$user .ssh &&
81
+ cat <<-EOT |tee -a .ssh/authorized_keys
82
+ #{public_key}
83
+ EOT
84
+ done
85
+ CMD
86
+ end
87
+ resp = ec2_client.run_instances(
88
+ instance_type: 't3a.small',
89
+ image_id: image.image_id,
90
+ min_count: 1,
91
+ max_count: 1,
92
+ key_name: key_pair_name,
93
+ user_data: user_data,
94
+ tag_specifications: [{
95
+ resource_type: 'instance',
96
+ tags: [{key: 'name', value: AWS_AUTH_EC2_INSTANCE_NAME}],
97
+ }],
98
+ monitoring: {enabled: false},
99
+ credit_specification: {cpu_credits: 'standard'},
100
+ security_group_ids: [security_group_id],
101
+ metadata_options: {
102
+ # This is required for Docker containers on the instance to be able
103
+ # to use the instance metadata endpoints.
104
+ http_put_response_hop_limit: 2,
105
+ },
106
+ ).to_h
107
+ instance_id = resp[:instances].first[:instance_id]
108
+ reservations = ec2_client.describe_instances(instance_ids: [instance_id]).reservations
109
+ instance = find_running_instance(reservations)
110
+ end
111
+ if instance.nil?
112
+ raise "Instance should have been found here"
113
+ end
114
+ if instance.state.name == 'stopped'
115
+ ec2_client.start_instances(instance_ids: [instance.instance_id])
116
+ end
117
+ 10.times do
118
+ if %w(stopped pending).include?(instance.state.name)
119
+ puts "Waiting for instance #{instance.instance_id} to start (current state: #{instance.state.name})"
120
+ sleep 5
121
+ end
122
+ reservations = ec2_client.describe_instances(instance_ids: [instance.instance_id]).reservations
123
+ instance = find_running_instance(reservations)
124
+ end
125
+ puts "Found usable instance #{instance.instance_id} at #{instance.public_ip_address}"
126
+ end
127
+
128
+ def terminate_auth_ec2_instance
129
+ ec2_client.describe_instances(filters: [
130
+ {name: 'tag:name', values: [AWS_AUTH_EC2_INSTANCE_NAME]},
131
+ ]).each do |resp|
132
+ resp.reservations.each do |res|
133
+ res.instances.each do |instance|
134
+ puts "Terminating #{instance.instance_id}"
135
+ ec2_client.terminate_instances(instance_ids: [instance.instance_id])
136
+ end
137
+ end
138
+ end
139
+ end
140
+
141
+ def provision_auth_ecs_task(public_key_path: nil,
142
+ cluster_name: AWS_AUTH_ECS_CLUSTER_NAME,
143
+ service_name: AWS_AUTH_ECS_SERVICE_NAME,
144
+ security_group_id: nil,
145
+ subnet_ids: nil,
146
+ task_definition_ref: AWS_AUTH_ECS_TASK_FAMILY
147
+ )
148
+ security_group_id ||= ssh_vpc_security_group_id!
149
+ subnet_ids ||= [subnet_id!]
150
+
151
+ # https://docs.aws.amazon.com/AmazonECS/latest/developerguide/ECS_AWSCLI_Fargate.html
152
+ resp = ecs_client.describe_clusters(
153
+ clusters: [cluster_name],
154
+ )
155
+ cluster = detect_object(resp, :clusters, :cluster_name, cluster_name)
156
+ if cluster.nil?
157
+ raise 'No cluster found, please run `aws setup-resources`'
158
+ end
159
+
160
+ if public_key_path
161
+ public_key = File.read(public_key_path)
162
+ unless public_key =~ /\Assh-/
163
+ raise "The file at #{public_key_path} does not look like a public key"
164
+ end
165
+ entry_point = ['bash', '-c', <<-CMD]
166
+ apt-get update &&
167
+ apt-get install -y openssh-server &&
168
+ cd /root &&
169
+ mkdir -p .ssh &&
170
+ chmod 0700 .ssh &&
171
+ cat >.ssh/authorized_keys <<-EOT &&
172
+ #{public_key}
173
+ EOT
174
+ service ssh start &&
175
+ sleep 10000000
176
+ #mkdir /run/sshd && /usr/sbin/sshd -d
177
+ CMD
178
+ else
179
+ entry_point = nil
180
+ end
181
+ launch_type = if options[:ec2]
182
+ 'EC2'
183
+ else
184
+ 'FARGATE'
185
+ end
186
+ # When testing in Evergreen, we are given the task definition ARN
187
+ # and we always launch the tasks with that ARN.
188
+ # When testing locally, we repace task definition every time we launch
189
+ # the service.
190
+ if task_definition_ref !~ /^arn:/
191
+ execution_role = detect_object(iam_client.list_roles, :roles, :role_name, AWS_AUTH_ECS_EXECUTION_ROLE_NAME)
192
+ if execution_role.nil?
193
+ raise 'Execution role not configured'
194
+ end
195
+ task_role = detect_object(iam_client.list_roles, :roles, :role_name, AWS_AUTH_ECS_TASK_ROLE_NAME)
196
+ if task_role.nil?
197
+ raise 'Task role not configured'
198
+ end
199
+
200
+ task_definition = ecs_client.register_task_definition(
201
+ family: AWS_AUTH_ECS_TASK_FAMILY,
202
+ container_definitions: [{
203
+ name: 'ssh',
204
+ essential: true,
205
+ entry_point: entry_point,
206
+ image: 'debian:9',
207
+ port_mappings: [{
208
+ container_port: 22,
209
+ protocol: 'tcp',
210
+ }],
211
+ log_configuration: {
212
+ log_driver: 'awslogs',
213
+ options: {
214
+ 'awslogs-group' => AWS_AUTH_ECS_LOG_GROUP,
215
+ 'awslogs-region' => region,
216
+ 'awslogs-stream-prefix' => AWS_AUTH_ECS_LOG_STREAM_PREFIX,
217
+ },
218
+ },
219
+ }],
220
+ requires_compatibilities: [launch_type],
221
+ network_mode: 'awsvpc',
222
+ cpu: '512',
223
+ memory: '2048',
224
+ # This is the ECS task role used for AWS auth testing
225
+ task_role_arn: task_role.arn,
226
+ # The execution role is required to support awslogs (logging to
227
+ # CloudWatch).
228
+ execution_role_arn: execution_role.arn,
229
+ ).task_definition
230
+ task_definition_ref = AWS_AUTH_ECS_TASK_FAMILY
231
+ end
232
+
233
+ service = ecs_client.describe_services(
234
+ cluster: cluster_name,
235
+ services: [service_name],
236
+ ).services.first
237
+
238
+ if service && service.status.downcase == 'draining'
239
+ puts "Waiting for #{service_name} to drain"
240
+ ecs_client.wait_until(
241
+ :services_inactive, {
242
+ cluster: cluster.cluster_name,
243
+ services: [service_name],
244
+ },
245
+ delay: 5,
246
+ max_attempts: 36,
247
+ )
248
+ puts "... done."
249
+ service = nil
250
+ end
251
+ if service && service.status.downcase == 'inactive'
252
+ service = nil
253
+ end
254
+ if service
255
+ puts "Updating service with status #{service.status}"
256
+ service = ecs_client.update_service(
257
+ cluster: cluster_name,
258
+ service: service_name,
259
+ task_definition: task_definition_ref,
260
+ ).service
261
+ else
262
+ puts "Creating a new service"
263
+ vpc_config = {}
264
+ unless options[:ec2]
265
+ vpc_config[:assign_public_ip] = 'ENABLED'
266
+ end
267
+ service = ecs_client.create_service(
268
+ desired_count: 1,
269
+ service_name: service_name,
270
+ task_definition: task_definition_ref,
271
+ cluster: cluster_name,
272
+ launch_type: launch_type,
273
+ network_configuration: {
274
+ awsvpc_configuration: vpc_config.merge(
275
+ subnets: subnet_ids,
276
+ security_groups: [security_group_id],
277
+ ),
278
+ },
279
+ ).service
280
+ end
281
+ end
282
+
283
+ def wait_for_ecs_ready(
284
+ cluster_name: AWS_AUTH_ECS_CLUSTER_NAME,
285
+ service_name: AWS_AUTH_ECS_SERVICE_NAME,
286
+ timeout: 20
287
+ )
288
+ deadline = Time.now + timeout
289
+
290
+ # The AWS SDK waiter seems to immediately fail sometimes right after
291
+ # the service is created, so wait for the service to become active
292
+ # manually and then use the waiter to wait for the service to become
293
+ # stable.
294
+ #
295
+ # The failure may be due to the fact that apparently, it is possible for
296
+ # describe_services to not return an existing service for some time.
297
+ # Therefore, allow the lack of service to be a transient error.
298
+ loop do
299
+ service = ecs_client.describe_services(
300
+ cluster: cluster_name,
301
+ services: [service_name],
302
+ ).services.first
303
+
304
+ if service.nil?
305
+ puts "Service #{service_name} in cluster #{cluster_name} does not exist (yet?)"
306
+ status = 'MISSING'
307
+ elsif service.status.downcase == 'active'
308
+ break
309
+ else
310
+ status = service.status
311
+ end
312
+
313
+ now = Time.now
314
+ if now >= deadline
315
+ raise "Service #{service_name} in cluster #{cluster_name} did not become ready in #{timeout} seconds (current status: #{status})"
316
+ end
317
+
318
+ puts "Wating for service #{service_name} in cluster #{cluster_name} to become ready (#{'%2.1f' % (deadline - now)} seconds remaining, current status: #{status})"
319
+ sleep 5
320
+ end
321
+
322
+ puts "Wating for service #{service_name} in cluster #{cluster_name} to become stable"
323
+ ecs_client.wait_until(
324
+ :services_stable, {
325
+ cluster: cluster_name,
326
+ services: [service_name],
327
+ },
328
+ delay: 5,
329
+ max_attempts: 36,
330
+ )
331
+ end
332
+
333
+ def terminate_auth_ecs_task
334
+ ecs_client.describe_services(
335
+ cluster: AWS_AUTH_ECS_CLUSTER_NAME,
336
+ services: [AWS_AUTH_ECS_SERVICE_NAME],
337
+ ).each do |resp|
338
+ resp.services.each do |service|
339
+ puts "Terminating #{service.service_name}"
340
+ begin
341
+ ecs_client.update_service(
342
+ cluster: AWS_AUTH_ECS_CLUSTER_NAME,
343
+ service: service.service_name,
344
+ desired_count: 0,
345
+ )
346
+ rescue Aws::ECS::Errors::ServiceNotActiveException
347
+ # No action needed
348
+ end
349
+ ecs_client.delete_service(
350
+ cluster: AWS_AUTH_ECS_CLUSTER_NAME,
351
+ service: service.service_name,
352
+ )
353
+ end
354
+ end
355
+ end
356
+
357
+ private
358
+
359
+ def find_running_instance(reservations)
360
+ instance = nil
361
+ reservations.each do |reservation|
362
+ instance = reservation.instances.detect do |instance|
363
+ %w(pending running stopped).include?(instance.state.name)
364
+ end
365
+ break if instance
366
+ end
367
+ instance
368
+ end
369
+ end
370
+ end