better amount distribution verification

mekanism · Apr 20, 2024 · 9d8647d · 9d8647d
1 parent 05ca049
commit 9d8647d
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/src/test/java/mekanism/common/lib/distribution/DistributionTest.java b/src/test/java/mekanism/common/lib/distribution/DistributionTest.java
@@ -36,6 +36,20 @@ void testEvenDistribution() {
         int toSend = 10;
         IntegerTarget availableAcceptors = getTargets(toSend, 0, 0);
         Assertions.assertEquals(toSend, EmitUtils.sendToAcceptors(availableAcceptors, toSend, toSend));
+        for (IntegerHandler handler : availableAcceptors.handlers) {
+            Assertions.assertEquals(1, handler.getAccepted());
+        }
+    }
+
+    @Test
+    @DisplayName("Test sending to targets where the amounts divide evenly with more than 1 each")
+    void testEvenDistribution2() {
+        int toSend = 40;
+        IntegerTarget availableAcceptors = getTargets(toSend / 4, 0, 0);
+        Assertions.assertEquals(toSend, EmitUtils.sendToAcceptors(availableAcceptors, toSend, toSend));
+        for (IntegerHandler handler : availableAcceptors.handlers) {
+            Assertions.assertEquals(4, handler.getAccepted());
+        }
     }
 
     @Test
@@ -44,6 +58,17 @@ void testRemainderDistribution() {
         int toSend = 10;
         IntegerTarget availableAcceptors = getTargets(7, 0, 0);
         Assertions.assertEquals(toSend, EmitUtils.sendToAcceptors(availableAcceptors, toSend, toSend));
+        int singleAccepted = 0, twoAccepted = 0;
+        for (IntegerHandler handler : availableAcceptors.handlers) {
+            Assertions.assertTrue(handler.getAccepted() == 1 || handler.getAccepted() == 2);
+            if (handler.getAccepted() == 1) {
+                singleAccepted++;
+            } else {
+                twoAccepted++;
+            }
+        }
+        Assertions.assertEquals(4, singleAccepted);
+        Assertions.assertEquals(3, twoAccepted);
     }
 
     @Test
@@ -57,6 +82,7 @@ void testAllRemainder() {
         for (IntegerHandler availableAcceptor : availableAcceptors.handlers) {
             if (availableAcceptor.getAccepted() > 0) {
                 destinationsAccepted++;
+                Assertions.assertEquals(1, availableAcceptor.getAccepted());
             } else {
                 destinationsNotAccepted++;
             }