mongo 2.19.3 → 2.20.1

Sign up to get free protection for your applications and to get access to all the features.
Files changed (178) hide show
  1. checksums.yaml +4 -4
  2. data/README.md +37 -1
  3. data/Rakefile +81 -172
  4. data/lib/mongo/cluster/topology/base.rb +16 -0
  5. data/lib/mongo/cluster.rb +27 -1
  6. data/lib/mongo/collection/view/iterable.rb +1 -0
  7. data/lib/mongo/collection.rb +4 -2
  8. data/lib/mongo/config.rb +2 -2
  9. data/lib/mongo/error/transactions_not_supported.rb +34 -0
  10. data/lib/mongo/error.rb +1 -0
  11. data/lib/mongo/grid/fs_bucket.rb +6 -0
  12. data/lib/mongo/monitoring/event/secure.rb +1 -1
  13. data/lib/mongo/operation/shared/executable.rb +43 -27
  14. data/lib/mongo/operation/shared/response_handling.rb +23 -25
  15. data/lib/mongo/retryable/base_worker.rb +28 -3
  16. data/lib/mongo/retryable/read_worker.rb +16 -14
  17. data/lib/mongo/retryable/write_worker.rb +11 -8
  18. data/lib/mongo/retryable.rb +2 -2
  19. data/lib/mongo/server/app_metadata/environment.rb +64 -9
  20. data/lib/mongo/server/app_metadata.rb +5 -4
  21. data/lib/mongo/server/description/features.rb +1 -0
  22. data/lib/mongo/server/pending_connection.rb +19 -6
  23. data/lib/mongo/server_selector/base.rb +32 -6
  24. data/lib/mongo/session/server_session/dirtyable.rb +52 -0
  25. data/lib/mongo/session/server_session.rb +3 -0
  26. data/lib/mongo/session/session_pool.rb +12 -18
  27. data/lib/mongo/session.rb +32 -0
  28. data/lib/mongo/socket/ssl.rb +22 -1
  29. data/lib/mongo/uri.rb +0 -4
  30. data/lib/mongo/version.rb +1 -5
  31. data/mongo.gemspec +9 -18
  32. data/spec/atlas/atlas_connectivity_spec.rb +4 -4
  33. data/spec/faas/ruby-sam-app/Gemfile +9 -0
  34. data/spec/faas/ruby-sam-app/mongodb/Gemfile +4 -0
  35. data/spec/faas/ruby-sam-app/mongodb/app.rb +149 -0
  36. data/spec/faas/ruby-sam-app/template.yaml +48 -0
  37. data/spec/integration/client_side_encryption/corpus_spec.rb +10 -2
  38. data/spec/integration/client_side_encryption/range_explicit_encryption_prose_spec.rb +3 -0
  39. data/spec/integration/retryable_reads_errors_spec.rb +196 -31
  40. data/spec/integration/retryable_writes_errors_spec.rb +156 -0
  41. data/spec/integration/sdam_error_handling_spec.rb +2 -0
  42. data/spec/lite_spec_helper.rb +0 -10
  43. data/spec/mongo/cluster_spec.rb +36 -0
  44. data/spec/mongo/collection/view/aggregation_spec.rb +6 -1
  45. data/spec/mongo/collection/view/explainable_spec.rb +2 -0
  46. data/spec/mongo/collection_crud_spec.rb +2 -1
  47. data/spec/mongo/operation/insert_spec.rb +1 -1
  48. data/spec/mongo/retryable/write_worker_spec.rb +39 -0
  49. data/spec/mongo/server/app_metadata/environment_spec.rb +135 -0
  50. data/spec/mongo/server/app_metadata_spec.rb +12 -2
  51. data/spec/mongo/server/connection_spec.rb +26 -0
  52. data/spec/mongo/session/session_pool_spec.rb +1 -16
  53. data/spec/mongo/session_transaction_spec.rb +15 -0
  54. data/spec/mongo/uri_spec.rb +0 -9
  55. data/spec/runners/crud/test.rb +0 -8
  56. data/spec/runners/crud.rb +1 -1
  57. data/spec/runners/transactions/test.rb +12 -3
  58. data/spec/runners/unified/assertions.rb +16 -3
  59. data/spec/runners/unified/crud_operations.rb +12 -0
  60. data/spec/runners/unified/support_operations.rb +3 -5
  61. data/spec/runners/unified/test.rb +8 -1
  62. data/spec/spec_tests/data/client_side_encryption/explain.yml +2 -2
  63. data/spec/spec_tests/data/client_side_encryption/fle2v2-BypassQueryAnalysis.yml +1 -0
  64. data/spec/spec_tests/data/client_side_encryption/fle2v2-Compact.yml +1 -0
  65. data/spec/spec_tests/data/client_side_encryption/fle2v2-CreateCollection.yml +1 -0
  66. data/spec/spec_tests/data/client_side_encryption/fle2v2-DecryptExistingData.yml +1 -0
  67. data/spec/spec_tests/data/client_side_encryption/fle2v2-Delete.yml +1 -0
  68. data/spec/spec_tests/data/client_side_encryption/fle2v2-EncryptedFields-vs-EncryptedFieldsMap.yml +1 -0
  69. data/spec/spec_tests/data/client_side_encryption/fle2v2-EncryptedFields-vs-jsonSchema.yml +1 -0
  70. data/spec/spec_tests/data/client_side_encryption/fle2v2-EncryptedFieldsMap-defaults.yml +1 -0
  71. data/spec/spec_tests/data/client_side_encryption/fle2v2-FindOneAndUpdate.yml +1 -0
  72. data/spec/spec_tests/data/client_side_encryption/fle2v2-InsertFind-Indexed.yml +1 -0
  73. data/spec/spec_tests/data/client_side_encryption/fle2v2-InsertFind-Unindexed.yml +1 -0
  74. data/spec/spec_tests/data/client_side_encryption/fle2v2-MissingKey.yml +1 -0
  75. data/spec/spec_tests/data/client_side_encryption/fle2v2-NoEncryption.yml +1 -0
  76. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Date-Aggregate.yml +1 -0
  77. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Date-Correctness.yml +1 -0
  78. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Date-Delete.yml +1 -0
  79. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Date-FindOneAndUpdate.yml +1 -0
  80. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Date-InsertFind.yml +1 -0
  81. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Date-Update.yml +1 -0
  82. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Decimal-Aggregate.yml +1 -0
  83. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Decimal-Correctness.yml +1 -0
  84. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Decimal-Delete.yml +1 -0
  85. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Decimal-FindOneAndUpdate.yml +1 -0
  86. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Decimal-InsertFind.yml +1 -0
  87. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Decimal-Update.yml +1 -0
  88. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DecimalPrecision-Aggregate.yml +1 -0
  89. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DecimalPrecision-Correctness.yml +1 -0
  90. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DecimalPrecision-Delete.yml +1 -0
  91. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DecimalPrecision-FindOneAndUpdate.yml +1 -0
  92. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DecimalPrecision-InsertFind.yml +1 -0
  93. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DecimalPrecision-Update.yml +1 -0
  94. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Double-Aggregate.yml +1 -0
  95. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Double-Correctness.yml +1 -0
  96. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Double-Delete.yml +1 -0
  97. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Double-FindOneAndUpdate.yml +1 -0
  98. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Double-InsertFind.yml +1 -0
  99. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Double-Update.yml +1 -0
  100. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DoublePrecision-Aggregate.yml +1 -0
  101. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DoublePrecision-Correctness.yml +1 -0
  102. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DoublePrecision-Delete.yml +1 -0
  103. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DoublePrecision-FindOneAndUpdate.yml +1 -0
  104. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DoublePrecision-InsertFind.yml +1 -0
  105. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DoublePrecision-Update.yml +1 -0
  106. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Int-Aggregate.yml +1 -0
  107. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Int-Correctness.yml +1 -0
  108. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Int-Delete.yml +1 -0
  109. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Int-FindOneAndUpdate.yml +1 -0
  110. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Int-InsertFind.yml +1 -0
  111. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Int-Update.yml +1 -0
  112. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Long-Aggregate.yml +1 -0
  113. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Long-Correctness.yml +1 -0
  114. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Long-Delete.yml +1 -0
  115. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Long-FindOneAndUpdate.yml +1 -0
  116. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Long-InsertFind.yml +1 -0
  117. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Long-Update.yml +1 -0
  118. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-WrongType.yml +1 -0
  119. data/spec/spec_tests/data/client_side_encryption/fle2v2-Update.yml +1 -0
  120. data/spec/spec_tests/data/client_side_encryption/fle2v2-validatorAndPartialFieldExpression.yml +2 -1
  121. data/spec/spec_tests/data/connection_string/invalid-uris.yml +0 -10
  122. data/spec/spec_tests/data/connection_string/valid-options.yml +13 -0
  123. data/spec/spec_tests/data/crud_unified/aggregate-write-readPreference.yml +2 -0
  124. data/spec/spec_tests/data/crud_unified/db-aggregate-write-readPreference.yml +2 -0
  125. data/spec/spec_tests/data/crud_unified/find-test-all-options.yml +348 -0
  126. data/spec/spec_tests/data/index_management/createSearchIndex.yml +5 -3
  127. data/spec/spec_tests/data/index_management/createSearchIndexes.yml +7 -4
  128. data/spec/spec_tests/data/index_management/dropSearchIndex.yml +2 -1
  129. data/spec/spec_tests/data/index_management/listSearchIndexes.yml +13 -7
  130. data/spec/spec_tests/data/index_management/updateSearchIndex.yml +2 -1
  131. data/spec/spec_tests/data/retryable_writes/unified/bulkWrite-serverErrors.yml +3 -6
  132. data/spec/spec_tests/data/retryable_writes/unified/insertOne-serverErrors.yml +3 -6
  133. data/spec/spec_tests/data/run_command_unified/runCommand.yml +319 -0
  134. data/spec/spec_tests/data/sessions_unified/driver-sessions-dirty-session-errors.yml +351 -0
  135. data/spec/spec_tests/data/unified/valid-pass/poc-crud.yml +1 -1
  136. data/spec/spec_tests/data/unified/valid-pass/poc-retryable-writes.yml +7 -7
  137. data/spec/spec_tests/data/unified/valid-pass/poc-sessions.yml +3 -4
  138. data/spec/spec_tests/data/unified/valid-pass/poc-transactions-convenient-api.yml +1 -1
  139. data/spec/spec_tests/data/unified/valid-pass/poc-transactions-mongos-pin-auto.yml +1 -1
  140. data/spec/spec_tests/data/unified/valid-pass/poc-transactions.yml +3 -3
  141. data/spec/spec_tests/run_command_unified_spec.rb +13 -0
  142. data/spec/spec_tests/sdam_unified_spec.rb +2 -0
  143. data/spec/spec_tests/transactions_unified_spec.rb +2 -1
  144. data/spec/support/certificates/atlas-ocsp-ca.crt +89 -77
  145. data/spec/support/certificates/atlas-ocsp.crt +117 -122
  146. data/spec/support/certificates/retrieve-atlas-cert +1 -1
  147. data/spec/support/constraints.rb +6 -0
  148. data/spec/support/ocsp +1 -1
  149. data/spec/support/recording_logger.rb +27 -0
  150. metadata +1245 -1298
  151. checksums.yaml.gz.sig +0 -0
  152. data/spec/shared/LICENSE +0 -20
  153. data/spec/shared/bin/get-mongodb-download-url +0 -17
  154. data/spec/shared/bin/s3-copy +0 -45
  155. data/spec/shared/bin/s3-upload +0 -69
  156. data/spec/shared/lib/mrss/child_process_helper.rb +0 -80
  157. data/spec/shared/lib/mrss/cluster_config.rb +0 -231
  158. data/spec/shared/lib/mrss/constraints.rb +0 -378
  159. data/spec/shared/lib/mrss/docker_runner.rb +0 -295
  160. data/spec/shared/lib/mrss/eg_config_utils.rb +0 -51
  161. data/spec/shared/lib/mrss/event_subscriber.rb +0 -210
  162. data/spec/shared/lib/mrss/lite_constraints.rb +0 -238
  163. data/spec/shared/lib/mrss/server_version_registry.rb +0 -113
  164. data/spec/shared/lib/mrss/session_registry.rb +0 -69
  165. data/spec/shared/lib/mrss/session_registry_legacy.rb +0 -60
  166. data/spec/shared/lib/mrss/spec_organizer.rb +0 -179
  167. data/spec/shared/lib/mrss/utils.rb +0 -37
  168. data/spec/shared/share/Dockerfile.erb +0 -330
  169. data/spec/shared/share/haproxy-1.conf +0 -16
  170. data/spec/shared/share/haproxy-2.conf +0 -17
  171. data/spec/shared/shlib/config.sh +0 -27
  172. data/spec/shared/shlib/distro.sh +0 -74
  173. data/spec/shared/shlib/server.sh +0 -416
  174. data/spec/shared/shlib/set_env.sh +0 -169
  175. data/spec/spec_tests/data/cmap/pool-clear-interrupt-immediately.yml +0 -49
  176. data/spec/support/faas/app/aws_lambda/mongodb/Gemfile.lock +0 -19
  177. data.tar.gz.sig +0 -3
  178. metadata.gz.sig +0 -0
@@ -1,378 +0,0 @@
1
- # frozen_string_literal: true
2
- # encoding: utf-8
3
-
4
- module Mrss
5
- module Constraints
6
- def min_server_version(version)
7
- parsed_version = Gem::Version.new(version)
8
-
9
- before(:all) do
10
- if parsed_version > Gem::Version.new(ClusterConfig.instance.server_version)
11
- skip "Server version #{version} or higher required, we have #{ClusterConfig.instance.server_version}"
12
- end
13
- end
14
- end
15
-
16
- def max_server_version(version)
17
- parsed_version = Gem::Version.new(version)
18
-
19
- before(:all) do
20
- if parsed_version < Gem::Version.new(ClusterConfig.instance.server_version)
21
- skip "Server version #{version} or lower required, we have #{ClusterConfig.instance.server_version}"
22
- end
23
- end
24
- end
25
-
26
- def min_server_fcv(version)
27
- parsed_version = Gem::Version.new(version)
28
-
29
- before(:all) do
30
- unless Gem::Version.new(ClusterConfig.instance.fcv_ish) >= parsed_version
31
- skip "FCV #{version} or higher required, we have #{ClusterConfig.instance.fcv_ish} (server #{ClusterConfig.instance.server_version})"
32
- end
33
- end
34
- end
35
-
36
- def max_server_fcv(version)
37
- parsed_version = Gem::Version.new(version)
38
-
39
- before(:all) do
40
- if parsed_version < Gem::Version.new(ClusterConfig.instance.fcv_ish)
41
- skip "FCV #{version} or lower required, we have #{ClusterConfig.instance.fcv_ish} (server #{ClusterConfig.instance.server_version})"
42
- end
43
- end
44
- end
45
-
46
- def require_topology(*topologies)
47
- invalid_topologies = topologies - [:single, :replica_set, :sharded, :load_balanced]
48
-
49
- unless invalid_topologies.empty?
50
- raise ArgumentError, "Invalid topologies requested: #{invalid_topologies.join(', ')}"
51
- end
52
-
53
- before(:all) do
54
- unless topologies.include?(topology = ClusterConfig.instance.topology)
55
- skip "Topology #{topologies.join(' or ')} required, we have #{topology}"
56
- end
57
- end
58
- end
59
-
60
- def max_example_run_time(timeout)
61
- around do |example|
62
- TimeoutInterrupt.timeout(timeout, TimeoutInterrupt::Error.new("Test execution terminated after #{timeout} seconds")) do
63
- example.run
64
- end
65
- end
66
- end
67
-
68
- def require_transaction_support
69
- before(:all) do
70
- case ClusterConfig.instance.topology
71
- when :single
72
- skip 'Transactions tests require a replica set (4.0+) or a sharded cluster (4.2+)'
73
- when :replica_set
74
- unless ClusterConfig.instance.server_version >= '4.0'
75
- skip 'Transactions tests in a replica set topology require server 4.0+'
76
- end
77
- when :sharded, :load_balanced
78
- unless ClusterConfig.instance.server_version >= '4.2'
79
- skip 'Transactions tests in a sharded cluster topology require server 4.2+'
80
- end
81
- else
82
- raise NotImplementedError
83
- end
84
- end
85
- end
86
-
87
- # Fail command fail point was added to mongod in 4.0 and to mongos in 4.2.
88
- def require_fail_command
89
- require_transaction_support
90
- end
91
-
92
- def require_tls
93
- before(:all) do
94
- unless SpecConfig.instance.ssl?
95
- skip "SSL not enabled"
96
- end
97
- end
98
- end
99
-
100
- def require_no_tls
101
- before(:all) do
102
- if SpecConfig.instance.ssl?
103
- skip "SSL enabled"
104
- end
105
- end
106
- end
107
-
108
- def require_retry_writes
109
- before(:all) do
110
- unless SpecConfig.instance.retry_writes?
111
- skip "Retry writes is disabled"
112
- end
113
- end
114
- end
115
-
116
- def require_no_retry_writes
117
- before(:all) do
118
- if SpecConfig.instance.retry_writes?
119
- skip "Retry writes is enabled"
120
- end
121
- end
122
- end
123
-
124
- def require_compression
125
- before(:all) do
126
- if SpecConfig.instance.compressors.nil?
127
- skip "Compression is not enabled"
128
- end
129
- end
130
- end
131
-
132
- def require_zlib_compression
133
- before(:all) do
134
- compressors = SpecConfig.instance.compressors
135
- unless compressors && compressors.include?('zlib')
136
- skip "Zlib compression is not enabled"
137
- end
138
- end
139
- end
140
-
141
- def require_snappy_compression
142
- before(:all) do
143
- compressors = SpecConfig.instance.compressors
144
- unless compressors && compressors.include?('snappy')
145
- skip "Snappy compression is not enabled"
146
- end
147
- end
148
- end
149
-
150
- def require_no_snappy_compression
151
- before(:all) do
152
- compressors = SpecConfig.instance.compressors
153
- if compressors && compressors.include?('snappy')
154
- skip "Snappy compression is enabled"
155
- end
156
- end
157
- end
158
-
159
- def require_zstd_compression
160
- before(:all) do
161
- compressors = SpecConfig.instance.compressors
162
- unless compressors && compressors.include?('zstd')
163
- skip "Zstd compression is not enabled"
164
- end
165
- end
166
- end
167
-
168
- def require_no_zstd_compression
169
- before(:all) do
170
- compressors = SpecConfig.instance.compressors
171
- if compressors && compressors.include?('zstd')
172
- skip "Zstd compression is enabled"
173
- end
174
- end
175
- end
176
-
177
- def require_no_compression
178
- before(:all) do
179
- if SpecConfig.instance.compressors
180
- skip "Compression is enabled"
181
- end
182
- end
183
- end
184
-
185
- def ruby_version_gte(version)
186
- before(:all) do
187
- if RUBY_VERSION < version
188
- skip "Ruby version #{version} or higher required"
189
- end
190
- end
191
- end
192
-
193
- def ruby_version_lt(version)
194
- before(:all) do
195
- if RUBY_VERSION >= version
196
- skip "Ruby version less than #{version} required"
197
- end
198
- end
199
- end
200
-
201
- def require_auth(*values)
202
- before(:all) do
203
- if values.any?
204
- unless values.include?(ENV['AUTH'])
205
- msg = values.map { |v| "AUTH=#{v}" }.join(' or ')
206
- skip "This test requires #{msg}"
207
- end
208
- else
209
- unless ENV['AUTH'] == 'auth' || SpecConfig.instance.user || ClusterConfig.instance.auth_enabled?
210
- skip "Auth required"
211
- end
212
- end
213
- end
214
- end
215
-
216
- def require_no_auth
217
- before(:all) do
218
- auth = ENV.fetch('AUTH', '')
219
- if (!auth.empty? && auth != 'noauth') || SpecConfig.instance.user || ClusterConfig.instance.auth_enabled?
220
- skip "Auth not allowed"
221
- end
222
- end
223
- end
224
-
225
- def require_x509_auth
226
- before(:all) do
227
- unless SpecConfig.instance.x509_auth?
228
- skip "X.509 auth required"
229
- end
230
- end
231
- end
232
-
233
- def require_no_external_user
234
- before(:all) do
235
- if SpecConfig.instance.external_user?
236
- skip "External user configurations are not compatible with this test"
237
- end
238
- end
239
- end
240
-
241
- # Can the driver specify a write concern that won't be overridden?
242
- # (mongos 4.0+ overrides the write concern)
243
- def require_set_write_concern
244
- before(:all) do
245
- if %i(sharded load_balanced).include?(ClusterConfig.instance.topology) &&
246
- ClusterConfig.instance.short_server_version >= '4.0'
247
- then
248
- skip "mongos 4.0+ overrides write concern"
249
- end
250
- end
251
- end
252
-
253
- def require_multi_mongos
254
- before(:all) do
255
- if ClusterConfig.instance.topology == :sharded && SpecConfig.instance.addresses.length == 1
256
- skip 'Test requires a minimum of two mongoses if run in sharded topology'
257
- end
258
-
259
- if ClusterConfig.instance.topology == :load_balanced && SpecConfig.instance.single_mongos?
260
- skip 'Test requires a minimum of two mongoses if run in load-balanced topology'
261
- end
262
- end
263
- end
264
-
265
- # In sharded topology operations are distributed to the mongoses.
266
- # When we set fail points, the fail point may be set on one mongos and
267
- # operation may be executed on another mongos, causing failures.
268
- # Tests that are not setting targeted fail points should utilize this
269
- # method to restrict themselves to single mongos.
270
- #
271
- # In load-balanced topology, the same problem can happen when there is
272
- # more than one mongos behind the load balancer.
273
- def require_no_multi_mongos
274
- before(:all) do
275
- if ClusterConfig.instance.topology == :sharded && SpecConfig.instance.addresses.length > 1
276
- skip 'Test requires a single mongos if run in sharded topology'
277
- end
278
- if ClusterConfig.instance.topology == :load_balanced && !SpecConfig.instance.single_mongos?
279
- skip 'Test requires a single mongos, as indicated by SINGLE_MONGOS=1 environment variable, if run in load-balanced topology'
280
- end
281
- end
282
- end
283
-
284
- alias :require_no_multi_shard :require_no_multi_mongos
285
-
286
- def require_wired_tiger
287
- before(:all) do
288
- # Storage detection fails for serverless instances. However, it is safe to
289
- # assume that a serverless instance uses WiredTiger Storage Engine.
290
- if !SpecConfig.instance.serverless? && ClusterConfig.instance.storage_engine != :wired_tiger
291
- skip 'Test requires WiredTiger storage engine'
292
- end
293
- end
294
- end
295
-
296
- def require_wired_tiger_on_36
297
- before(:all) do
298
- if ClusterConfig.instance.short_server_version >= '3.6'
299
- # Storage detection fails for serverless instances. However, it is safe to
300
- # assume that a serverless instance uses WiredTiger Storage Engine.
301
- if !SpecConfig.instance.serverless? && ClusterConfig.instance.storage_engine != :wired_tiger
302
- skip 'Test requires WiredTiger storage engine on 3.6+ servers'
303
- end
304
- end
305
- end
306
- end
307
-
308
- def require_mmapv1
309
- before(:all) do
310
- if SpecConfig.instance.serverless? || ClusterConfig.instance.storage_engine != :mmapv1
311
- skip 'Test requires MMAPv1 storage engine'
312
- end
313
- end
314
- end
315
-
316
- def require_enterprise
317
- before(:all) do
318
- unless ClusterConfig.instance.enterprise?
319
- skip 'Test requires enterprise build of MongoDB'
320
- end
321
- end
322
- end
323
-
324
- # Integration tests for SRV polling require internet connectivity to
325
- # look up SRV records and a sharded cluster configured on default port on
326
- # localhost (localhost:27017, localhost:27018).
327
- def require_default_port_deployment
328
- # Because the DNS records at test1.test.build.10gen.cc point to
329
- # localhost:27017 & localhost:27018, the test suite must have been
330
- # configured to use these addresses
331
- before(:all) do
332
- have_default_port = SpecConfig.instance.addresses.any? do |address|
333
- %w(127.0.0.1 127.0.0.1:27017 localhost localhost:27017).include?(address)
334
- end
335
- unless have_default_port
336
- skip 'This test requires the test suite to be configured for localhost:27017'
337
- end
338
- end
339
- end
340
-
341
- # Some tests perform assertions on what the driver is logging.
342
- # Some test configurations, for example OCSP with unknown response,
343
- # produce warnings due to optional checks failing.
344
- # This constraint skips tests that issue logging assertions on configurations
345
- # that may produce non-test-originated log entries.
346
- def require_warning_clean
347
- before(:all) do
348
- if ENV['OCSP_STATUS'] == 'unknown'
349
- skip 'Unknown OCSP status is not global warning-clean'
350
- end
351
- end
352
- end
353
-
354
- def require_required_api_version
355
- before(:all) do
356
- unless ENV['API_VERSION_REQUIRED'] == '1'
357
- skip 'Set API_VERSION_REQUIRED=1 to run this test'
358
- end
359
- end
360
- end
361
-
362
- def require_no_required_api_version
363
- before(:all) do
364
- if ENV['API_VERSION_REQUIRED'] == '1'
365
- skip 'Cannot have API_VERSION_REQUIRED=1 to run this test'
366
- end
367
- end
368
- end
369
-
370
- def require_unix_socket
371
- before(:all) do
372
- if ENV['TOPOLOGY'] == 'load-balanced'
373
- skip 'Load balancer does not listen on Unix sockets'
374
- end
375
- end
376
- end
377
- end
378
- end
@@ -1,295 +0,0 @@
1
- # frozen_string_literal: true
2
- # encoding: utf-8
3
-
4
- require 'optparse'
5
- require 'erb'
6
- autoload :Dotenv, 'dotenv'
7
-
8
- module Mrss
9
- autoload :ServerVersionRegistry, 'mrss/server_version_registry'
10
-
11
- class DockerRunner
12
- def initialize(**opts)
13
- # These options are required:
14
- opts.fetch(:image_tag)
15
- opts.fetch(:dockerfile_path)
16
- opts.fetch(:default_script)
17
- opts.fetch(:project_lib_subdir)
18
-
19
- @options = opts.merge(preload: true)
20
- end
21
-
22
- attr_reader :options
23
-
24
- def run
25
- process_arguments
26
- unless @options[:exec_only]
27
- create_dockerfile
28
- create_image
29
- end
30
- if @options[:mongo_only]
31
- run_deployment
32
- else
33
- run_tests
34
- end
35
- end
36
-
37
- private
38
-
39
- def process_arguments
40
- #@options = {}
41
- OptionParser.new do |opts|
42
- opts.banner = "Usage: test-on-docker [-d distro] [evergreen_key=value ...]"
43
-
44
- opts.on("-a", "--add-env=PATH", "Load environment variables from PATH in .env format") do |path|
45
- @options[:extra_env] ||= {}
46
- unless File.exist?(path)
47
- raise "-a option references nonexistent file #{path}"
48
- end
49
- Dotenv.parse(path).each do |k, v|
50
- @options[:extra_env][k] = v
51
- end
52
- end
53
-
54
- opts.on("-d", "--distro=DISTRO", "Distro to use") do |v|
55
- @options[:distro] = v
56
- end
57
-
58
- opts.on('-e', '--exec-only', 'Execute tests using existing Dockerfile (for offline user)') do |v|
59
- @options[:exec_only] = v
60
- end
61
-
62
- opts.on('-m', '--mongo-only=PORT', 'Start the MongoDB deployment and expose it to host on ports starting with PORT') do |v|
63
- @options[:mongo_only] = v.to_i
64
- end
65
-
66
- opts.on('-p', '--preload', 'Preload Ruby toolchain and server binaries in docker (default)') do |v|
67
- @options[:preload] = v
68
- end
69
-
70
- opts.on('-P', '--no-preload', 'Do not preload Ruby toolchain and server binaries in docker') do
71
- @options[:preload] = false
72
- end
73
-
74
- opts.on('-s', '--script=SCRIPT', 'Test script to invoke') do |v|
75
- @options[:script] = v
76
- end
77
-
78
- opts.on('-i', '--interactive', 'Interactive mode - disable per-test timeouts') do |v|
79
- @options[:interactive] = v
80
- end
81
- end.parse!
82
-
83
- @env = Hash[ARGV.map do |arg|
84
- arg.split('=', 2)
85
- end]
86
-
87
- @env['RVM_RUBY'] ||= 'ruby-2.7'
88
- unless ruby =~ /^j?ruby-/
89
- raise "RVM_RUBY option is not in expected format: #{ruby}"
90
- end
91
-
92
- @env['MONGODB_VERSION'] ||= '4.4'
93
- end
94
-
95
- def create_dockerfile
96
- template_path = File.join(File.dirname(__FILE__), '../../share/Dockerfile.erb')
97
- result = ERB.new(File.read(template_path)).result(binding)
98
- File.open(dockerfile_path, 'w') do |f|
99
- f << result
100
- end
101
- end
102
-
103
- def image_tag
104
- options.fetch(:image_tag)
105
- end
106
-
107
- def dockerfile_path
108
- options.fetch(:dockerfile_path)
109
- end
110
-
111
- def create_image
112
- run_command(['docker', 'build',
113
- '-t', image_tag,
114
- '-f', dockerfile_path,
115
- '.'])
116
- end
117
-
118
- BASE_TEST_COMMAND = %w(docker run --rm -i --tmpfs /tmpfs:exec).freeze
119
-
120
- def run_tests
121
- run_command(BASE_TEST_COMMAND + tty_arg + extra_env + [image_tag] +
122
- script.split(/\s+/))
123
- end
124
-
125
- def run_deployment
126
- run_command(BASE_TEST_COMMAND + tty_arg + extra_env + [
127
- '-e', %q`TEST_CMD=watch -x bash -c "ps awwxu |egrep 'mongo|ocsp'"`,
128
- '-e', 'BIND_ALL=true',
129
- ] + port_forwards + [image_tag] + script.split(/\s+/))
130
- end
131
-
132
- def tty_arg
133
- tty = File.open('/dev/stdin') do |f|
134
- f.isatty
135
- end
136
- if tty
137
- %w(-t --init)
138
- else
139
- []
140
- end
141
- end
142
-
143
- def extra_env
144
- if @options[:extra_env]
145
- @options[:extra_env].map do |k, v|
146
- # Here the value must not be escaped
147
- ['-e', "#{k}=#{v}"]
148
- end.flatten
149
- else
150
- []
151
- end
152
- end
153
-
154
- def port_forwards
155
- args = (0...num_exposed_ports).map do |i|
156
- host_port = @options[:mongo_only] + i
157
- container_port = 27017 + i
158
- ['-p', "#{host_port}:#{container_port}"]
159
- end.flatten
160
-
161
- if @env['OCSP_ALGORITHM'] && !@env['OCSP_VERIFIER']
162
- args += %w(-p 8100:8100)
163
- end
164
-
165
- args
166
- end
167
-
168
- def run_command(cmd)
169
- if pid = fork
170
- Process.wait(pid)
171
- unless $?.exitstatus == 0
172
- raise "Process exited with code #{$?.exitstatus}"
173
- end
174
- else
175
- exec(*cmd)
176
- end
177
- end
178
-
179
- def distro
180
- @options[:distro] || if app_tests?
181
- 'ubuntu2004'
182
- else
183
- case server_version
184
- when '3.6'
185
- 'debian9'
186
- when '4.0', '4.2'
187
- 'ubuntu1804'
188
- else
189
- 'ubuntu2004'
190
- end
191
- end
192
- end
193
-
194
- BASE_IMAGES = {
195
- 'debian81' => 'debian:jessie',
196
- 'debian92' => 'debian:stretch',
197
- 'debian10' => 'debian:buster',
198
- 'ubuntu1404' => 'ubuntu:trusty',
199
- 'ubuntu1604' => 'ubuntu:xenial',
200
- 'ubuntu1804' => 'ubuntu:bionic',
201
- 'ubuntu2004' => 'ubuntu:focal',
202
- 'rhel62' => 'centos:6',
203
- 'rhel70' => 'centos:7',
204
- }.freeze
205
-
206
- def base_image
207
- BASE_IMAGES[distro] or raise "Unknown distro: #{distro}"
208
- end
209
-
210
- def ruby
211
- @env['RVM_RUBY']
212
- end
213
-
214
- def ruby_head?
215
- ruby == 'ruby-head'
216
- end
217
-
218
- def system_ruby?
219
- %w(1 true yes).include?(@env['SYSTEM_RUBY']&.downcase)
220
- end
221
-
222
- def server_version
223
- @env['MONGODB_VERSION']
224
- end
225
-
226
- def script
227
- @options[:script] || options.fetch(:default_script)
228
- end
229
-
230
- def debian?
231
- distro =~ /debian|ubuntu/
232
- end
233
-
234
- def ubuntu?
235
- distro=~ /ubuntu/
236
- end
237
-
238
- def preload?
239
- !!@options[:preload]
240
- end
241
-
242
- def interactive?
243
- !!@options[:interactive]
244
- end
245
-
246
- def project_lib_subdir
247
- options.fetch(:project_lib_subdir)
248
- end
249
-
250
- def server_download_url
251
- @server_download_url ||= ServerVersionRegistry.new(server_version, distro).download_url
252
- end
253
-
254
- def libmongocrypt_path
255
- case distro
256
- when /ubuntu1604/
257
- "./ubuntu1604/nocrypto/lib64/libmongocrypt.so"
258
- when /ubuntu1804/
259
- "./ubuntu1804-64/nocrypto/lib64/libmongocrypt.so"
260
- when /debian92/
261
- "./debian92/nocrypto/lib64/libmongocrypt.so"
262
- else
263
- raise "This script does not support running FLE tests on #{distro}. Use ubuntu1604, ubuntu1804 or debian92 instead"
264
- end
265
- end
266
-
267
- def expose?
268
- !!@options[:mongo_only]
269
- end
270
-
271
- def fle?
272
- %w(1 true yes).include?(@env['FLE']&.downcase)
273
- end
274
-
275
- # Mongoid
276
- def app_tests?
277
- %w(1 true yes).include?(@env['APP_TESTS']&.downcase)
278
- end
279
-
280
- def num_exposed_ports
281
- case @env['TOPOLOGY'] || 'standalone'
282
- when 'standalone'
283
- 1
284
- when 'replica-set'
285
- 3
286
- when 'sharded-cluster'
287
- if @env['SINGLE_MONGOS']
288
- 1
289
- else
290
- 2
291
- end
292
- end
293
- end
294
- end
295
- end