synapsecns/sanguine

// SPDX-License-Identifier: MIT
pragma solidity 0.8.17;

import {MerkleMath} from "../../../contracts/libs/merkle/MerkleMath.sol";
import {Snapshot, SNAPSHOT_MAX_STATES, SNAPSHOT_TREE_HEIGHT, State} from "../../../contracts/libs/memory/Snapshot.sol";
import {fakeSnapshot, RawState, RawStateIndex, RawSnapshot} from "../libs/FakeIt.t.sol";
import {AttestationProofGenerator} from "./AttestationProofGenerator.t.sol";

import {Test} from "forge-std/Test.sol";

// solhint-disable func-name-mixedcase
contract AttestationProofGeneratorTest is Test {
    AttestationProofGenerator internal proofGen;

    function setUp() public {
        proofGen = new AttestationProofGenerator();
    }

    function test_attestationProof(RawState memory rawState, RawStateIndex memory rsi) public {
        rsi.boundStateIndex();
        RawSnapshot memory rawSnap = fakeSnapshot(rawState, rsi);
        Snapshot snapshot = rawSnap.castToSnapshot();
        State state = rawState.castToState();
        bytes[] memory states = rawSnap.formatStates();
        proofGen.acceptSnapshot(states);
        bytes32 snapshotRoot = proofGen.root();
        assertEq(snapshotRoot, snapshot.calculateRoot(), "!snapshotRoot");

        (bytes32 item,) = state.subLeafs();
        uint256 itemIndex = rsi.stateIndex << 1;
        bytes32[] memory proof = proofGen.generateProof(rsi.stateIndex);
        assertEq(MerkleMath.proofRoot(itemIndex, item, proof, SNAPSHOT_TREE_HEIGHT), snapshotRoot, "!proof");
    }
}