Add tests

ArkEcosystem · Jan 9, 2025 · 30f247e · 30f247e
1 parent efbe048
commit 30f247e
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/contracts/test/consensus/Consensus-CalculateTop.sol b/contracts/test/consensus/Consensus-CalculateTop.sol
@@ -152,7 +152,7 @@ contract ConsensusTest is Base {
             vm.deal(addr, balance);
 
             if (balance == highestBalance) {
-                if (addr > highest) {
+                if (addr < highest || highest == address(0)) {
                     highest = addr;
                 }
             }
@@ -179,15 +179,17 @@ contract ConsensusTest is Base {
 
         assertEq(validators[activeValidators - 1].addr, address(0x1B)); // Shuffled address
         validators = sortValidators(validators);
-        assertEq(validators[activeValidators - 1].addr, address(0x35));
+        assertEq(validators[0].addr, highest);
+        assertEq(validators[activeValidators - 1].addr, address(53));
 
-        // Seccond attempt should return the same result
+        // Second attempt should return the same result
         consensus.calculateActiveValidators(uint8(activeValidators));
 
         validators = consensus.getActiveValidators();
         assertEq(validators[activeValidators - 1].addr, address(0x1B)); // Shuffled address
         validators = sortValidators(validators);
         assertEq(validators.length, activeValidators);
-        assertEq(validators[activeValidators - 1].addr, address(0x35));
+        assertEq(validators[0].addr, highest);
+        assertEq(validators[activeValidators - 1].addr, address(53));
     }
 }