[Segment Replication] Fix testAllConstraints test by using similar ca…

…lculations as used in code (opensearch-project#6817) Signed-off-by: Suraj Singh <[email protected]> Signed-off-by: Valentin Mitrofanov <[email protected]>
mitrofmep · Apr 5, 2023 · 3cae35e · 3cae35e
1 parent 408d6ed
commit 3cae35e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...r/src/test/java/org/opensearch/cluster/routing/allocation/AllocationConstraintsTests.java b/...r/src/test/java/org/opensearch/cluster/routing/allocation/AllocationConstraintsTests.java
@@ -200,9 +200,9 @@ public void testAllConstraints() {
         when(node.numShards(indexName)).thenReturn(shardCount);
         when(node.getNodeId()).thenReturn("test-node");
 
-        long expectedWeight = (shardCount >= avgPerIndexShardsPerNode) ? CONSTRAINT_WEIGHT : 0;
-        expectedWeight += perIndexPrimaryShardCount > avgPerIndexPrimaryShardsPerNode ? CONSTRAINT_WEIGHT : 0;
-        expectedWeight += primaryShardsPerNode >= avgPrimaryShardsPerNode ? CONSTRAINT_WEIGHT : 0;
+        long expectedWeight = (shardCount >= (int) Math.ceil(avgPerIndexShardsPerNode)) ? CONSTRAINT_WEIGHT : 0;
+        expectedWeight += perIndexPrimaryShardCount > (int) Math.ceil(avgPerIndexPrimaryShardsPerNode) ? CONSTRAINT_WEIGHT : 0;
+        expectedWeight += primaryShardsPerNode >= (int) Math.ceil(avgPrimaryShardsPerNode) ? CONSTRAINT_WEIGHT : 0;
         assertEquals(expectedWeight, constraints.weight(balancer, node, indexName));
     }