rln-contract/test/RLNApp.t.sol

// SPDX-License-Identifier: Unlicense
pragma solidity ^0.8.15;

import "../contracts/RlnBase.sol";
import "./Verifier.sol";
import "forge-std/Test.sol";
import "forge-std/StdCheats.sol";
import "forge-std/console.sol";

contract RlnApp is RlnBase {
    uint256 public constant allowedIdCommitment =
        19014214495641488759237505126948346942972912379615652741039992445865937985820;

    uint256 private membershipDeposit = 1000000000000000;
    uint256 private depth = 20;

    constructor(address _verifier) RlnBase(membershipDeposit, depth, _verifier) {}

    function _validateRegistration(uint256 idCommitment) internal pure override {
        if (idCommitment != allowedIdCommitment) revert FailedValidation();
    }

    function _validateSlash(uint256 idCommitment, address payable receiver, uint256[8] calldata proof)
        internal
        pure
        override
    {
        if (idCommitment == allowedIdCommitment) revert FailedValidation();
    }
}

contract RLNAppTest is Test {
    RlnApp public rlnApp;
    TrueVerifier public trueVerifier;

    uint256 public constant MEMBERSHIP_DEPOSIT = 1000000000000000;
    uint256 public constant DEPTH = 20;
    uint256 public constant SET_SIZE = 1048576;
    uint256[8] public zeroedProof = [0, 0, 0, 0, 0, 0, 0, 0];

    function setUp() public {
        trueVerifier = new TrueVerifier();
        rlnApp = new RlnApp(address(trueVerifier));
    }

    function test__Constants() public {
        // sanity checking
        assertEq(rlnApp.MEMBERSHIP_DEPOSIT(), MEMBERSHIP_DEPOSIT);
        assertEq(rlnApp.DEPTH(), DEPTH);
        assertEq(rlnApp.SET_SIZE(), SET_SIZE);
    }

    function test__InvalidRegistration(uint256 idCommitment) public {
        vm.assume(idCommitment != rlnApp.allowedIdCommitment());
        vm.assume(rlnApp.isValidCommitment(idCommitment));
        vm.expectRevert(FailedValidation.selector);
        rlnApp.register{value: MEMBERSHIP_DEPOSIT}(idCommitment);
    }

    function test__ValidRegistration() public {
        rlnApp.register{value: MEMBERSHIP_DEPOSIT}(rlnApp.allowedIdCommitment());
    }

    function test__InvalidSlash() public {
        uint256 allowedIdCommitment = rlnApp.allowedIdCommitment();
        rlnApp.register{value: MEMBERSHIP_DEPOSIT}(allowedIdCommitment);
        vm.expectRevert(FailedValidation.selector);
        rlnApp.slash(allowedIdCommitment, payable(address(this)), zeroedProof);
    }
}