mongo 2.20.0 → 2.20.1

Sign up to get free protection for your applications and to get access to all the features.
Files changed (109) hide show
  1. checksums.yaml +4 -4
  2. data/README.md +37 -1
  3. data/Rakefile +57 -21
  4. data/lib/mongo/config.rb +2 -2
  5. data/lib/mongo/retryable/base_worker.rb +28 -3
  6. data/lib/mongo/retryable/read_worker.rb +9 -8
  7. data/lib/mongo/retryable/write_worker.rb +4 -4
  8. data/lib/mongo/server/pending_connection.rb +19 -6
  9. data/lib/mongo/socket/ssl.rb +22 -1
  10. data/lib/mongo/version.rb +1 -5
  11. data/mongo.gemspec +8 -11
  12. data/spec/integration/client_side_encryption/range_explicit_encryption_prose_spec.rb +3 -0
  13. data/spec/integration/retryable_reads_errors_spec.rb +35 -23
  14. data/spec/integration/sdam_error_handling_spec.rb +2 -0
  15. data/spec/lite_spec_helper.rb +0 -10
  16. data/spec/mongo/collection_crud_spec.rb +1 -0
  17. data/spec/mongo/server/connection_spec.rb +22 -0
  18. data/spec/spec_tests/data/client_side_encryption/fle2v2-BypassQueryAnalysis.yml +1 -0
  19. data/spec/spec_tests/data/client_side_encryption/fle2v2-Compact.yml +1 -0
  20. data/spec/spec_tests/data/client_side_encryption/fle2v2-CreateCollection.yml +1 -0
  21. data/spec/spec_tests/data/client_side_encryption/fle2v2-DecryptExistingData.yml +1 -0
  22. data/spec/spec_tests/data/client_side_encryption/fle2v2-Delete.yml +1 -0
  23. data/spec/spec_tests/data/client_side_encryption/fle2v2-EncryptedFields-vs-EncryptedFieldsMap.yml +1 -0
  24. data/spec/spec_tests/data/client_side_encryption/fle2v2-EncryptedFields-vs-jsonSchema.yml +1 -0
  25. data/spec/spec_tests/data/client_side_encryption/fle2v2-EncryptedFieldsMap-defaults.yml +1 -0
  26. data/spec/spec_tests/data/client_side_encryption/fle2v2-FindOneAndUpdate.yml +1 -0
  27. data/spec/spec_tests/data/client_side_encryption/fle2v2-InsertFind-Indexed.yml +1 -0
  28. data/spec/spec_tests/data/client_side_encryption/fle2v2-InsertFind-Unindexed.yml +1 -0
  29. data/spec/spec_tests/data/client_side_encryption/fle2v2-MissingKey.yml +1 -0
  30. data/spec/spec_tests/data/client_side_encryption/fle2v2-NoEncryption.yml +1 -0
  31. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Date-Aggregate.yml +1 -0
  32. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Date-Correctness.yml +1 -0
  33. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Date-Delete.yml +1 -0
  34. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Date-FindOneAndUpdate.yml +1 -0
  35. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Date-InsertFind.yml +1 -0
  36. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Date-Update.yml +1 -0
  37. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Decimal-Aggregate.yml +1 -0
  38. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Decimal-Correctness.yml +1 -0
  39. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Decimal-Delete.yml +1 -0
  40. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Decimal-FindOneAndUpdate.yml +1 -0
  41. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Decimal-InsertFind.yml +1 -0
  42. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Decimal-Update.yml +1 -0
  43. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DecimalPrecision-Aggregate.yml +1 -0
  44. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DecimalPrecision-Correctness.yml +1 -0
  45. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DecimalPrecision-Delete.yml +1 -0
  46. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DecimalPrecision-FindOneAndUpdate.yml +1 -0
  47. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DecimalPrecision-InsertFind.yml +1 -0
  48. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DecimalPrecision-Update.yml +1 -0
  49. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Double-Aggregate.yml +1 -0
  50. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Double-Correctness.yml +1 -0
  51. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Double-Delete.yml +1 -0
  52. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Double-FindOneAndUpdate.yml +1 -0
  53. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Double-InsertFind.yml +1 -0
  54. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Double-Update.yml +1 -0
  55. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DoublePrecision-Aggregate.yml +1 -0
  56. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DoublePrecision-Correctness.yml +1 -0
  57. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DoublePrecision-Delete.yml +1 -0
  58. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DoublePrecision-FindOneAndUpdate.yml +1 -0
  59. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DoublePrecision-InsertFind.yml +1 -0
  60. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-DoublePrecision-Update.yml +1 -0
  61. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Int-Aggregate.yml +1 -0
  62. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Int-Correctness.yml +1 -0
  63. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Int-Delete.yml +1 -0
  64. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Int-FindOneAndUpdate.yml +1 -0
  65. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Int-InsertFind.yml +1 -0
  66. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Int-Update.yml +1 -0
  67. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Long-Aggregate.yml +1 -0
  68. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Long-Correctness.yml +1 -0
  69. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Long-Delete.yml +1 -0
  70. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Long-FindOneAndUpdate.yml +1 -0
  71. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Long-InsertFind.yml +1 -0
  72. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-Long-Update.yml +1 -0
  73. data/spec/spec_tests/data/client_side_encryption/fle2v2-Range-WrongType.yml +1 -0
  74. data/spec/spec_tests/data/client_side_encryption/fle2v2-Update.yml +1 -0
  75. data/spec/spec_tests/data/client_side_encryption/fle2v2-validatorAndPartialFieldExpression.yml +2 -1
  76. data/spec/spec_tests/data/crud_unified/aggregate-write-readPreference.yml +2 -0
  77. data/spec/spec_tests/data/crud_unified/db-aggregate-write-readPreference.yml +2 -0
  78. data/spec/spec_tests/transactions_unified_spec.rb +2 -1
  79. data/spec/support/certificates/atlas-ocsp-ca.crt +89 -77
  80. data/spec/support/certificates/atlas-ocsp.crt +117 -122
  81. data/spec/support/certificates/retrieve-atlas-cert +1 -1
  82. metadata +12 -84
  83. checksums.yaml.gz.sig +0 -0
  84. data/spec/shared/LICENSE +0 -20
  85. data/spec/shared/bin/get-mongodb-download-url +0 -17
  86. data/spec/shared/bin/s3-copy +0 -45
  87. data/spec/shared/bin/s3-upload +0 -69
  88. data/spec/shared/lib/mrss/child_process_helper.rb +0 -80
  89. data/spec/shared/lib/mrss/cluster_config.rb +0 -231
  90. data/spec/shared/lib/mrss/constraints.rb +0 -378
  91. data/spec/shared/lib/mrss/docker_runner.rb +0 -298
  92. data/spec/shared/lib/mrss/eg_config_utils.rb +0 -51
  93. data/spec/shared/lib/mrss/event_subscriber.rb +0 -210
  94. data/spec/shared/lib/mrss/lite_constraints.rb +0 -238
  95. data/spec/shared/lib/mrss/server_version_registry.rb +0 -113
  96. data/spec/shared/lib/mrss/session_registry.rb +0 -69
  97. data/spec/shared/lib/mrss/session_registry_legacy.rb +0 -60
  98. data/spec/shared/lib/mrss/spec_organizer.rb +0 -179
  99. data/spec/shared/lib/mrss/utils.rb +0 -37
  100. data/spec/shared/share/Dockerfile.erb +0 -281
  101. data/spec/shared/share/haproxy-1.conf +0 -16
  102. data/spec/shared/share/haproxy-2.conf +0 -17
  103. data/spec/shared/shlib/config.sh +0 -27
  104. data/spec/shared/shlib/distro.sh +0 -74
  105. data/spec/shared/shlib/server.sh +0 -417
  106. data/spec/shared/shlib/set_env.sh +0 -146
  107. data/spec/support/faas/app/aws_lambda/mongodb/Gemfile.lock +0 -19
  108. data.tar.gz.sig +0 -0
  109. metadata.gz.sig +0 -3
@@ -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,298 +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
- 'debian11' => 'debian:bullseye',
199
- 'ubuntu1404' => 'ubuntu:trusty',
200
- 'ubuntu1604' => 'ubuntu:xenial',
201
- 'ubuntu1804' => 'ubuntu:bionic',
202
- 'ubuntu2004' => 'ubuntu:focal',
203
- 'ubuntu2204' => 'ubuntu:jammy',
204
- 'rhel62' => 'centos:6',
205
- 'rhel70' => 'centos:7',
206
- 'rhel80' => 'rockylinux:8',
207
- }.freeze
208
-
209
- def base_image
210
- BASE_IMAGES[distro] or raise "Unknown distro: #{distro}"
211
- end
212
-
213
- def ruby
214
- @env['RVM_RUBY']
215
- end
216
-
217
- def ruby_head?
218
- ruby == 'ruby-head'
219
- end
220
-
221
- def system_ruby?
222
- %w(1 true yes).include?(@env['SYSTEM_RUBY']&.downcase)
223
- end
224
-
225
- def server_version
226
- @env['MONGODB_VERSION']
227
- end
228
-
229
- def script
230
- @options[:script] || options.fetch(:default_script)
231
- end
232
-
233
- def debian?
234
- distro =~ /debian|ubuntu/
235
- end
236
-
237
- def ubuntu?
238
- distro=~ /ubuntu/
239
- end
240
-
241
- def preload?
242
- !!@options[:preload]
243
- end
244
-
245
- def interactive?
246
- !!@options[:interactive]
247
- end
248
-
249
- def project_lib_subdir
250
- options.fetch(:project_lib_subdir)
251
- end
252
-
253
- def server_download_url
254
- @server_download_url ||= ServerVersionRegistry.new(server_version, distro).download_url
255
- end
256
-
257
- def libmongocrypt_path
258
- case distro
259
- when /ubuntu1604/
260
- "./ubuntu1604/nocrypto/lib64/libmongocrypt.so"
261
- when /ubuntu1804/
262
- "./ubuntu1804-64/nocrypto/lib64/libmongocrypt.so"
263
- when /debian92/
264
- "./debian92/nocrypto/lib64/libmongocrypt.so"
265
- else
266
- raise "This script does not support running FLE tests on #{distro}. Use ubuntu1604, ubuntu1804 or debian92 instead"
267
- end
268
- end
269
-
270
- def expose?
271
- !!@options[:mongo_only]
272
- end
273
-
274
- def fle?
275
- %w(1 true yes).include?(@env['FLE']&.downcase)
276
- end
277
-
278
- # Mongoid
279
- def app_tests?
280
- %w(1 true yes).include?(@env['APP_TESTS']&.downcase)
281
- end
282
-
283
- def num_exposed_ports
284
- case @env['TOPOLOGY'] || 'standalone'
285
- when 'standalone'
286
- 1
287
- when 'replica-set'
288
- 3
289
- when 'sharded-cluster'
290
- if @env['SINGLE_MONGOS']
291
- 1
292
- else
293
- 2
294
- end
295
- end
296
- end
297
- end
298
- end