1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
# frozen_string_literal: true
# rubocop:todo all
module Mongo
module CRUD
# Represents a single CRUD test.
#
# @since 2.0.0
class CRUDTest < CRUDTestBase
# Spec tests have configureFailPoint as a string, make it a string here too
FAIL_POINT_BASE_COMMAND = BSON::Document.new(
'configureFailPoint' => "onPrimaryTransactionalWrite",
).freeze
# Instantiate the new CRUDTest.
#
# data can be an array of hashes, with each hash corresponding to a
# document to be inserted into the collection whose name is given in
# collection_name as configured in the YAML file. Alternatively data
# can be a map of collection names to arrays of hashes.
#
# @param [ Crud::Spec ] crud_spec The top level YAML specification object.
# @param [ Hash | Array<Hash> ] data The documents the collection
# must have before the test runs.
# @param [ Hash ] test The test specification.
#
# @since 2.0.0
def initialize(crud_spec, data, test)
@spec = crud_spec
@data = data
@description = test['description']
@client_options = ::Utils.convert_client_options(test['clientOptions'] || {})
if test['failPoint']
@fail_point_command = FAIL_POINT_BASE_COMMAND.merge(test['failPoint'])
end
if test['operations']
@operations = test['operations'].map do |op_spec|
Operation.new(self, op_spec)
end
else
@operations = [Operation.new(self, test['operation'], test['outcome'])]
end
@expectations = BSON::ExtJSON.parse_obj(test['expectations'], mode: :bson)
if test['outcome']
@outcome = Mongo::CRUD::Outcome.new(BSON::ExtJSON.parse_obj(test['outcome'], mode: :bson))
end
end
attr_reader :client_options
# Operations to be performed by the test.
#
# For CRUD tests, there is one operation for test. For retryable writes,
# there are multiple operations for each test. In either case we build
# an array of operations.
attr_reader :operations
attr_reader :outcome
# Run the test.
#
# The specified number of operations are executed, so that the
# test can assert on the outcome of each specified operation in turn.
#
# @param [ Client ] client The client the test
# should be run with.
# @param [ Integer ] num_ops Number of operations to run.
#
# @return [ Result, Array<Hash> ] The result(s) of running the test.
#
# @since 2.0.0
def run(client, num_ops)
result = nil
1.upto(num_ops) do |i|
operation = @operations[i-1]
target = resolve_target(client, operation)
result = operation.execute(target)
end
result
end
class DataConverter
include Mongo::GridFS::Convertible
end
def setup_test(spec, client)
clear_fail_point(client)
if @data.nil?
# nothing to do
elsif @data.is_a?(Array)
collection = client[spec.collection_name, write_concern: {w: :majority}]
collection.delete_many
collection.insert_many(@data) unless @data.empty?
elsif @data.is_a?(Hash)
converter = DataConverter.new
@data.each do |collection_name, data|
collection = client[collection_name]
collection.delete_many
data = converter.transform_docs(data)
collection.insert_many(data)
end
else
raise "Unknown type of data: #{@data}"
end
setup_fail_point(client)
end
end
end
end
|
