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
|
/**
* Tests that tenant migration recipient's in memory state is initialized correctly on initial sync.
* This test randomly selects a point during the migration to add a node to the recipient.
*
* @tags: [
* incompatible_with_macos,
* incompatible_with_shard_merge,
* incompatible_with_windows_tls,
* requires_fcv_62,
* requires_majority_read_concern,
* requires_persistence,
* serverless,
* ]
*/
(function() {
"use strict";
load("jstests/libs/fail_point_util.js");
load("jstests/libs/uuid_util.js");
load("jstests/libs/write_concern_util.js");
load("jstests/replsets/libs/tenant_migration_test.js");
const {ServerlessLockType, getServerlessOperationLock} = TenantMigrationUtil;
const tenantMigrationTest = new TenantMigrationTest({name: jsTestName()});
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 = [
"fpBeforeFetchingCommittedTransactions",
"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);
// Add the initial sync node and make sure that it does not step up.
const recipientRst = tenantMigrationTest.getRecipientRst();
const initialSyncNode = recipientRst.add({rsConfig: {priority: 0, votes: 0}});
recipientRst.reInitiate();
jsTestLog("Waiting for initial sync to finish.");
recipientRst.awaitSecondaryNodes();
recipientRst.awaitReplication();
// Stop replication on the node so that the TenantMigrationAccessBlocker cannot transition its state
// past what is reflected in the state doc read below.
stopServerReplication(initialSyncNode);
const configRecipientsColl = initialSyncNode.getCollection(TenantMigrationTest.kConfigRecipientsNS);
assert.lte(configRecipientsColl.count(), 1);
const recipientDoc = configRecipientsColl.findOne();
if (recipientDoc) {
switch (recipientDoc.state) {
case TenantMigrationTest.RecipientState.kStarted:
if (recipientDoc.dataConsistentStopDonorOpTime) {
assert.soon(() => tenantMigrationTest
.getTenantMigrationAccessBlocker(
{recipientNode: initialSyncNode, tenantId: kTenantId})
.recipient.state ==
TenantMigrationTest.RecipientAccessState.kReject);
}
break;
case TenantMigrationTest.RecipientState.kConsistent:
if (recipientDoc.rejectReadsBeforeTimestamp) {
assert.soon(() => tenantMigrationTest
.getTenantMigrationAccessBlocker(
{recipientNode: initialSyncNode, tenantId: kTenantId})
.recipient.state ==
TenantMigrationTest.RecipientAccessState.kRejectBefore);
assert.soon(() => bsonWoCompare(
tenantMigrationTest
.getTenantMigrationAccessBlocker(
{recipientNode: initialSyncNode, tenantId: kTenantId})
.recipient.rejectBeforeTimestamp,
recipientDoc.rejectReadsBeforeTimestamp) == 0);
} else {
assert.soon(() => tenantMigrationTest
.getTenantMigrationAccessBlocker(
{recipientNode: initialSyncNode, tenantId: kTenantId})
.recipient.state ==
TenantMigrationTest.RecipientAccessState.kReject);
}
break;
default:
throw new Error(`Invalid state "${state}" from recipient doc.`);
}
}
const activeServerlessLock = getServerlessOperationLock(initialSyncNode);
if (recipientDoc && !recipientDoc.expireAt) {
assert.eq(activeServerlessLock, ServerlessLockType.TenantMigrationRecipient);
} else {
assert.eq(activeServerlessLock, ServerlessLockType.None);
}
restartServerReplication(initialSyncNode);
tenantMigrationTest.stop();
})();
|
