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
|
/**
* Tests that tenant migration recipient's in memory state is recovered correctly on startup. This
* test randomly selects a point during the migration to shutdown the recipient.
*
* Tenant migrations are not expected to be run on servers with ephemeralForTest.
*
* @tags: [
* incompatible_with_eft,
* incompatible_with_macos,
* incompatible_with_windows_tls,
* requires_majority_read_concern,
* requires_persistence,
* ]
*/
(function() {
"use strict";
load("jstests/libs/fail_point_util.js");
load("jstests/libs/uuid_util.js");
load("jstests/replsets/libs/tenant_migration_test.js");
const recipientRst = new ReplSetTest({
nodes: 1,
name: 'recipient',
nodeOptions: Object.assign(TenantMigrationUtil.makeX509OptionsForTest().recipient, {
setParameter:
{"failpoint.PrimaryOnlyServiceSkipRebuildingInstances": tojson({mode: "alwaysOn"})}
})
});
recipientRst.startSet();
recipientRst.initiate();
const tenantMigrationTest = new TenantMigrationTest({name: jsTestName(), recipientRst});
if (!tenantMigrationTest.isFeatureFlagEnabled()) {
jsTestLog("Skipping test because the tenant migrations feature flag is disabled");
recipientRst.stopSet();
return;
}
const kMaxSleepTimeMS = 7500;
const kTenantId = 'testTenantId';
let recipientPrimary = tenantMigrationTest.getRecipientPrimary();
// Force the migration to pause after entering a randomly selected state.
Random.setRandomSeed();
const kMigrationFpNames = [
"fpAfterCollectionClonerDone",
"fpAfterWaitForRejectReadsBeforeTimestamp",
];
const index = Random.randInt(kMigrationFpNames.length + 1);
if (index < kMigrationFpNames.length) {
configureFailPoint(recipientPrimary, kMigrationFpNames[index], {action: "hang"});
}
const migrationOpts = {
migrationIdString: extractUUIDFromObject(UUID()),
tenantId: kTenantId,
};
assert.commandWorked(tenantMigrationTest.startMigration(migrationOpts));
sleep(Math.random() * kMaxSleepTimeMS);
recipientRst.stopSet(null /* signal */, true /*forRestart */);
recipientRst.startSet({
restart: true,
setParameter: {"failpoint.PrimaryOnlyServiceSkipRebuildingInstances": "{'mode':'alwaysOn'}"}
});
recipientPrimary = recipientRst.getPrimary();
const configRecipientsColl =
recipientPrimary.getCollection(TenantMigrationTest.kConfigRecipientsNS);
const recipientDoc = configRecipientsColl.findOne({tenantId: kTenantId});
if (recipientDoc) {
switch (recipientDoc.state) {
case TenantMigrationTest.RecipientState.kStarted:
if (recipientDoc.dataConsistentStopDonorOpTime) {
assert.soon(() => tenantMigrationTest
.getTenantMigrationAccessBlocker(recipientPrimary, kTenantId)
.recipient.state ==
TenantMigrationTest.RecipientAccessState.kReject);
}
break;
case TenantMigrationTest.RecipientState.kConsistent:
if (recipientDoc.rejectReadsBeforeTimestamp) {
assert.soon(() => tenantMigrationTest
.getTenantMigrationAccessBlocker(recipientPrimary, kTenantId)
.recipient.state ==
TenantMigrationTest.RecipientAccessState.kRejectBefore);
assert.soon(() => bsonWoCompare(tenantMigrationTest
.getTenantMigrationAccessBlocker(
recipientPrimary, kTenantId)
.recipient.rejectBeforeTimestamp,
recipientDoc.rejectReadsBeforeTimestamp) == 0);
} else {
assert.soon(() => tenantMigrationTest
.getTenantMigrationAccessBlocker(recipientPrimary, kTenantId)
.recipient.state ==
TenantMigrationTest.RecipientAccessState.kReject);
}
break;
default:
throw new Error(`Invalid state "${state}" from recipient doc.`);
}
}
tenantMigrationTest.stop();
recipientRst.stopSet();
})();
|
