- adding test, too little SD Collateral based on validator count

contracts/PermissionedNodeRegistry.sol

@@ -18,7 +18,7 @@
 import { ISDCollateral } from "./interfaces/SDCollateral/ISDCollateral.sol";
 import { IPermissionedNodeRegistry } from "./interfaces/IPermissionedNodeRegistry.sol";
 
 contract PermissionedNodeRegistry is
     INodeRegistry,
     IPermissionedNodeRegistry,
     AccessControlUpgradeable,
@@ -624,7 +624,7 @@
             }
         }
         // If the result array isn't full, resize it to remove the unused elements
         assembly {
             mstore(validators, validatorCount)
         }
 
@@ -722,7 +722,7 @@
         totalKeys = getOperatorTotalKeys(_operatorId);
         uint256 totalNonTerminalKeys = getOperatorTotalNonTerminalKeys(msg.sender, 0, totalKeys);
         if ((totalNonTerminalKeys + keyCount) > maxNonTerminalKeyPerOperator) {
-            revert maxKeyLimitReached();
+            revert MaxKeyLimitReached();
         }
 
         //checks if operator has enough SD collateral for adding `keyCount` keys

contracts/PermissionlessNodeRegistry.sol

@@ -21,7 +21,7 @@
 import { IPermissionlessNodeRegistry } from "./interfaces/IPermissionlessNodeRegistry.sol";
 import { IOperatorRewardsCollector } from "./interfaces/IOperatorRewardsCollector.sol";
 
 contract PermissionlessNodeRegistry is
     INodeRegistry,
     IPermissionlessNodeRegistry,
     AccessControlUpgradeable,
@@ -555,7 +555,7 @@
             }
         }
         // If the result array isn't full, resize it to remove the unused elements
         assembly {
             mstore(validators, validatorCount)
         }
 
@@ -687,7 +687,7 @@
         totalKeys = getOperatorTotalKeys(_operatorId);
         uint256 totalNonTerminalKeys = getOperatorTotalNonTerminalKeys(msg.sender, 0, totalKeys);
         if ((totalNonTerminalKeys + keyCount) > maxNonTerminalKeyPerOperator) {
-            revert maxKeyLimitReached();
+            revert MaxKeyLimitReached();
         }
 
         // check for collateral ETH for adding keys

contracts/interfaces/INodeRegistry.sol

@@ -27,7 +27,7 @@ interface INodeRegistry {
     // Errors
     error DuplicatePoolIDOrPoolNotAdded();
     error OperatorAlreadyOnBoardedInProtocol();
-    error maxKeyLimitReached();
+    error MaxKeyLimitReached();
     error OperatorNotOnBoarded();
     error InvalidKeyCount();
     error InvalidStartAndEndIndex();

test/foundry_tests/PermissionedNodeRegistry.t.sol

@@ -227,6 +227,24 @@ contract PermissionedNodeRegistryTest is Test {
         assertEq(nodeRegistry.isExistingPubkey(pubkeys[0]), true);
     }
 
+    function testAddValidatorKeysNotEnoughSDCollateral() public {
+        (
+            bytes[] memory pubkeys,
+            bytes[] memory preDepositSignature,
+            bytes[] memory depositSignature
+        ) = getValidatorKeys();
+        vm.startPrank(permissionedNO);
+        nodeRegistry.onboardNodeOperator("testOP", payable(address(this)));
+        vm.mockCall(
+            address(sdCollateral),
+            abi.encodeWithSelector(ISDCollateral.hasEnoughSDCollateral.selector),
+            abi.encode(false)
+        );
+        vm.expectRevert(INodeRegistry.NotEnoughSDCollateral.selector);
+        nodeRegistry.addValidatorKeys(pubkeys, preDepositSignature, depositSignature);
+        vm.stopPrank();
+    }
+
     function test_addValidatorKeysWithMisMatchingInputs() public {
         bytes[] memory pubkeys = new bytes[](1);
         bytes[] memory preDepositSignature = new bytes[](1);
@@ -263,7 +281,7 @@ contract PermissionedNodeRegistryTest is Test {
         nodeRegistry.updateMaxNonTerminalKeyPerOperator(2);
         vm.startPrank(permissionedNO);
         nodeRegistry.onboardNodeOperator("testOP", payable(address(this)));
-        vm.expectRevert(INodeRegistry.maxKeyLimitReached.selector);
+        vm.expectRevert(INodeRegistry.MaxKeyLimitReached.selector);
         nodeRegistry.addValidatorKeys(pubkeys, preDepositSignature, depositSignature);
         vm.stopPrank();
     }

test/foundry_tests/PermissionlessNodeRegistry.t.sol

@@ -254,6 +254,24 @@ contract PermissionlessNodeRegistryTest is Test {
         assertEq(nodeRegistry.isExistingPubkey(pubkeys[0]), true);
     }
 
+    function testAddValidatorKeysNotEnoughSDCollateral() public {
+        (
+            bytes[] memory pubkeys,
+            bytes[] memory preDepositSignature,
+            bytes[] memory depositSignature
+        ) = getValidatorKeys();
+        startHoax(address(this));
+        nodeRegistry.onboardNodeOperator(true, "testOP", payable(address(this)));
+        vm.mockCall(
+            address(sdCollateral),
+            abi.encodeWithSelector(ISDCollateral.hasEnoughSDCollateral.selector),
+            abi.encode(false)
+        );
+        vm.expectRevert(INodeRegistry.NotEnoughSDCollateral.selector);
+        nodeRegistry.addValidatorKeys{ value: 12 ether }(pubkeys, preDepositSignature, depositSignature);
+        vm.stopPrank();
+    }
+
     function test_addValidatorKeysWithMisMatchingInputs() public {
         bytes[] memory pubkeys = new bytes[](1);
         bytes[] memory preDepositSignature = new bytes[](1);
@@ -290,7 +308,7 @@ contract PermissionlessNodeRegistryTest is Test {
         nodeRegistry.updateMaxNonTerminalKeyPerOperator(2);
         startHoax(address(this));
         nodeRegistry.onboardNodeOperator(true, "testOP", payable(address(this)));
-        vm.expectRevert(INodeRegistry.maxKeyLimitReached.selector);
+        vm.expectRevert(INodeRegistry.MaxKeyLimitReached.selector);
         nodeRegistry.addValidatorKeys{ value: 12 ether }(pubkeys, preDepositSignature, depositSignature);
         vm.stopPrank();
     }