Skip to content
This repository has been archived by the owner on Jan 11, 2024. It is now read-only.

Add a test for new registration policy #404

Merged
merged 2 commits into from
Dec 19, 2023
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
37 changes: 34 additions & 3 deletions test/integration/SubnetActorDiamond.t.sol
Original file line number Diff line number Diff line change
Expand Up @@ -1267,14 +1267,13 @@ contract SubnetActorDiamondTest is Test, IntegrationTestBase {

(address validator1, bytes memory publicKey1) = TestUtils.deriveValidatorAddress(100);
vm.deal(validator1, DEFAULT_MIN_VALIDATOR_STAKE * 2);
vm.startPrank(validator1);
vm.prank(validator1);
saManager.join{value: DEFAULT_MIN_VALIDATOR_STAKE / 2}(publicKey1);

(address validator2, bytes memory publicKey2) = TestUtils.deriveValidatorAddress(101);
vm.deal(validator2, DEFAULT_MIN_VALIDATOR_STAKE * 2);
vm.startPrank(validator2);
vm.prank(validator2);
saManager.join{value: DEFAULT_MIN_VALIDATOR_STAKE / 2}(publicKey2);
vm.stopPrank();

require(saGetter.isActiveValidator(validator1), "not active validator 1");
require(saGetter.isActiveValidator(validator2), "not active validator 2");
Expand All @@ -1294,6 +1293,38 @@ contract SubnetActorDiamondTest is Test, IntegrationTestBase {
saManager.setFederatedPower(validators, publicKeys, powers);
}

function testSubnetActorDiamond_registration_policy() public {
(address validator1, bytes memory publicKey1) = TestUtils.deriveValidatorAddress(100);
vm.deal(validator1, DEFAULT_MIN_VALIDATOR_STAKE);

vm.expectRevert(abi.encodeWithSelector(MethodNotAllowed.selector, ERR_VALIDATOR_NOT_JOINED));
vm.prank(validator1);
saManager.stake{value: DEFAULT_MIN_VALIDATOR_STAKE / 2}();

vm.prank(validator1);
saManager.join{value: DEFAULT_MIN_VALIDATOR_STAKE / 2}(publicKey1);

require(saGetter.isActiveValidator(validator1), "active validator 1");
require(!saGetter.bootstrapped(), "subnet bootstrapped");

vm.expectRevert(abi.encodeWithSelector(MethodNotAllowed.selector, ERR_VALIDATOR_JOINED));
vm.prank(validator1);
saManager.join{value: DEFAULT_MIN_VALIDATOR_STAKE / 2}(publicKey1);

vm.prank(validator1);
saManager.stake{value: DEFAULT_MIN_VALIDATOR_STAKE / 2}();

require(saGetter.isActiveValidator(validator1), "active validator 1");

(address validator2, bytes memory publicKey2) = TestUtils.deriveValidatorAddress(101);
vm.deal(validator2, DEFAULT_MIN_VALIDATOR_STAKE * 2);
vm.prank(validator2);
saManager.join{value: DEFAULT_MIN_VALIDATOR_STAKE / 2}(publicKey2);

require(saGetter.isActiveValidator(validator1), "not active validator 1");
require(saGetter.bootstrapped(), "subnet not bootstrapped");
}

function testSubnetActorDiamond_FederatedValidation_bootstrapWorks() public {
gatewayAddress = address(gatewayDiamond);

Expand Down
Loading