SEARCH-1681 creating a loop to go through instances and shards

This commit is contained in:
Keerat
2020-02-13 14:36:37 +00:00
parent e22bb2966d
commit 6cb9b47fbb

View File

@@ -20,6 +20,7 @@ import static org.testng.Assert.assertNotNull;
import static org.testng.Assert.assertTrue; import static org.testng.Assert.assertTrue;
import java.util.Arrays; import java.util.Arrays;
import java.util.Iterator;
import java.util.List; import java.util.List;
import java.util.Set; import java.util.Set;
import java.util.stream.Collectors; import java.util.stream.Collectors;
@@ -155,42 +156,42 @@ public class ShardInfoTest extends AbstractE2EFunctionalTest
for (RestShardInfoModel shardInfoModel : entries) for (RestShardInfoModel shardInfoModel : entries)
{ {
RestShardInfoModel model = shardInfoModel.getModel(); RestShardInfoModel model = shardInfoModel.getModel();
assertEquals(model.getTemplate(), "rerank"); assertEquals(model.getTemplate(), "rerank", "Template is not rerank, template found: "+ model.getTemplate());
assertEquals(model.getMode(), "MASTER"); assertEquals(model.getMode(), "MASTER", "Mode is not MASTER, mode found: "+ model.getMode());
assertTrue(model.getHasContent()); assertTrue(model.getHasContent(), "There is no content on the shards");
assertTrue(model.getNumberOfShards()>=2); assertTrue(model.getNumberOfShards()>=2, "Number of shards is not equal to or greater than 2");
List<String> shardingMethods = Arrays.asList("DB_ID", "DB_ID_RANGE", "EXPLICIT_ID", "ACL_ID", "MOD_ACL_ID", "DATE", "PROPERTY"); List<String> shardingMethods = Arrays.asList("DB_ID", "DB_ID_RANGE", "EXPLICIT_ID", "ACL_ID", "MOD_ACL_ID", "DATE", "PROPERTY");
String shardingMethod = model.getShardMethod(); String shardingMethod = model.getShardMethod();
assertTrue(shardingMethods.contains(shardingMethod), "Unexpected Sharding Method Found: " + shardingMethod); assertTrue(shardingMethods.contains(shardingMethod), "Unexpected Sharding Method Found: " + shardingMethod);
List<RestShardModel> shards = model.getShards(); List<RestShardModel> shards = model.getShards();
assertNotNull(shards, "Unexpected number of shards found:" + shards); assertNotNull(shards);
RestShardModel shard = shards.iterator().next(); for (RestShardModel shardInstance : shards)
assertNotNull(shard); {
List<RestInstanceModel> instances = shard.getInstances(); List<RestInstanceModel> instanceList = shardInstance.getInstances();
assertNotNull(instances, "Unexpected number of instances found:" + instances); for (RestInstanceModel instanceX : instanceList)
RestInstanceModel instance = instances.iterator().next(); {
assertNotNull(instance); assertTrue(baseUrls.contains(instanceX.getBaseUrl()));
assertTrue(baseUrls.contains(instance.getBaseUrl())); assertEquals(instanceX.getState(), "ACTIVE", "Shard state is not ACTIVE, shard state is: " + instanceX.getState());
assertNotNull(instanceX.getPort(), "There is not port found for the instance");
assertEquals(instance.getState(), "ACTIVE"); assertEquals(instanceX.getMode(), "MASTER", "Mode is not MASTER, mode found: "+ instanceX.getMode());
assertTrue(instanceX.getTransactionsRemaining() >= 0, "Transactions remaining is not more than 0, transactions remaining: " + instanceX.getTransactionsRemaining());
// shardparams related checks String shardParams = (instanceX).getShardParams();
String shardParams = instance.getShardParams();
switch (shardingMethod) switch (shardingMethod)
{ {
case "MOD_ACL_ID": case "MOD_ACL_ID":
assertTrue(shardingMethod == "MOD_ACL_ID", "Sharding Method is not MOD_ACL_ID"); assertEquals(shardingMethod = "MOD_ACL_ID", "MOD_ACL_ID", "Sharding Method is not MOD_ACL_ID");
break; break;
case "ACL_ID": case "ACL_ID":
assertTrue(shardingMethod == "ACL_ID", "Sharding Method is not ACL_ID"); assertEquals(shardingMethod = "ACL_ID", "ACL_ID", "Sharding Method is not ACL_ID");
break; break;
case "DB_ID": case "DB_ID":
assertTrue(shardingMethod == "DB_ID", "Sharding Method is not DB_ID"); assertEquals(shardingMethod = "DB_ID", "DB_ID", "Sharding Method is not DB_ID");
break; break;
case "DB_ID_RANGE": case "DB_ID_RANGE":
assertTrue(shardingMethod == "DB_ID_RANGE", "Sharding Method is not DB_ID_RANGE"); assertEquals(shardingMethod = "DB_ID_RANGE", "DB_ID_RANGE", "Sharding Method is not DB_ID_RANGE");
//assertEquals(shardParams.contains("shard.range="), "Unexpected shard params defined for DB_ID_RANGE");
break; break;
case "DATE": case "DATE":
assertTrue(shardParams.contains("shard.key="), "Unexpected shard params defined for DATE"); assertTrue(shardParams.contains("shard.key="), "Unexpected shard params defined for DATE");
@@ -208,3 +209,5 @@ public class ShardInfoTest extends AbstractE2EFunctionalTest
} }
} }
} }
}
}