[ML] making test more determinate (#40374) (#40381)

* [ML] making test more determinate

* unmuting test
This commit is contained in:
Benjamin Trent 2019-03-23 12:15:37 -05:00 committed by GitHub
parent 03839ba1a2
commit 88f510ffc2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 7 deletions

View File

@ -82,7 +82,6 @@ public class DataFrameTransformsCheckpointServiceTests extends ESTestCase {
}
}
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/40368")
public void testExtractIndexCheckpointsInconsistentGlobalCheckpoints() {
Map<String, long[]> expectedCheckpoints = new HashMap<>();
Set<String> indices = randomUserIndices();
@ -141,9 +140,19 @@ public class DataFrameTransformsCheckpointServiceTests extends ESTestCase {
// we need at least one replica for testing
int numShardCopies = randomIntBetween(2, 4);
int primaryShard = 0;
if (skipPrimaries) {
primaryShard = randomInt(numShardCopies - 1);
}
int inconsistentReplica = -1;
if (inconsistentGlobalCheckpoints) {
inconsistentReplica = randomIntBetween(0, numShardCopies - 1);
List<Integer> replicas = new ArrayList<>(numShardCopies - 1);
for (int i = 0; i < numShardCopies; i++) {
if (primaryShard != i) {
replicas.add(i);
}
}
inconsistentReplica = randomFrom(replicas);
}
// SeqNoStats asserts that checkpoints are logical
@ -156,11 +165,7 @@ public class DataFrameTransformsCheckpointServiceTests extends ESTestCase {
for (int replica = 0; replica < numShardCopies; replica++) {
ShardId shardId = new ShardId(index, shardIndex);
boolean primary = (replica == 0);
if (skipPrimaries) {
primary = randomBoolean();
}
boolean primary = (replica == primaryShard);
Path path = createTempDir().resolve("indices").resolve(index.getUUID()).resolve(String.valueOf(shardIndex));
ShardRouting shardRouting = ShardRouting.newUnassigned(shardId, primary,