AttestationStation.t.sol 4.29 KB
Newer Older
Will Cory's avatar
Will Cory committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
//SPDX-License-Identifier: MIT
pragma solidity 0.8.15;

/* Testing utilities */
import { Test } from "forge-std/Test.sol";
import { AssetReceiver } from "../universal/AssetReceiver.sol";
import { AttestationStation } from "../universal/op-nft/AttestationStation.sol";

contract AssetReceiver_Initializer is Test {
    address alice_attestor = address(128);
    address bob = address(256);
    address sally = address(512);

    function _setUp() public {
        // Give alice and bob some ETH
        vm.deal(alice_attestor, 1 ether);

        vm.label(alice_attestor, "alice_attestor");
        vm.label(bob, "bob");
        vm.label(sally, "sally");
    }
}

contract AssetReceiverTest is AssetReceiver_Initializer {
    function setUp() public {
        super._setUp();
    }

    function test_attest_single() external {
        AttestationStation attestationStation = new AttestationStation();

Will Cory's avatar
Will Cory committed
32 33 34
        AttestationStation.AttestationData[]
            memory attestationDataArr = new AttestationStation.AttestationData[](1);

Will Cory's avatar
Will Cory committed
35 36 37 38 39 40 41 42 43
        // alice is going to attest about bob
        AttestationStation.AttestationData memory attestationData = AttestationStation
            .AttestationData({
                about: bob,
                key: bytes32("test-key:string"),
                val: bytes("test-value")
            });

        // assert the attestation starts empty
Will Cory's avatar
Will Cory committed
44 45
        assertEq(
            attestationStation.attestations(
46
                alice_attestor,
Will Cory's avatar
Will Cory committed
47
                attestationData.about,
48
                attestationData.key
Will Cory's avatar
Will Cory committed
49 50 51
            ),
            ""
        );
Will Cory's avatar
Will Cory committed
52 53 54

        // make attestation
        vm.prank(alice_attestor);
Will Cory's avatar
Will Cory committed
55 56
        attestationDataArr[0] = attestationData;
        attestationStation.attest(attestationDataArr);
Will Cory's avatar
Will Cory committed
57 58 59

        // assert the attestation is there
        assertEq(
60
            attestationStation.attestations(
Will Cory's avatar
Will Cory committed
61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
                alice_attestor,
                attestationData.about,
                attestationData.key
            ),
            attestationData.val
        );

        bytes memory new_val = bytes("new updated value");
        // make a new attestations to same about and key
        attestationData = AttestationStation.AttestationData({
            about: attestationData.about,
            key: attestationData.key,
            val: new_val
        });

        vm.prank(alice_attestor);
Will Cory's avatar
Will Cory committed
77 78
        attestationDataArr[0] = attestationData;
        attestationStation.attest(attestationDataArr);
Will Cory's avatar
Will Cory committed
79 80 81

        // assert the attestation is updated
        assertEq(
82
            attestationStation.attestations(
Will Cory's avatar
Will Cory committed
83 84 85 86 87 88 89
                alice_attestor,
                attestationData.about,
                attestationData.key
            ),
            attestationData.val
        );
    }
Will Cory's avatar
Will Cory committed
90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143

    function test_attest_bulk() external {
        AttestationStation attestationStation = new AttestationStation();

        vm.prank(alice_attestor);

        AttestationStation.AttestationData[]
            memory attestationData = new AttestationStation.AttestationData[](3);
        attestationData[0] = AttestationStation.AttestationData({
            about: bob,
            key: bytes32("test-key:string"),
            val: bytes("test-value")
        });

        attestationData[1] = AttestationStation.AttestationData({
            about: bob,
            key: bytes32("test-key2"),
            val: bytes("test-value2")
        });

        attestationData[2] = AttestationStation.AttestationData({
            about: sally,
            key: bytes32("test-key:string"),
            val: bytes("test-value3")
        });

        attestationStation.attest(attestationData);

        // assert the attestations are there
        assertEq(
            attestationStation.attestations(
                alice_attestor,
                attestationData[0].about,
                attestationData[0].key
            ),
            attestationData[0].val
        );
        assertEq(
            attestationStation.attestations(
                alice_attestor,
                attestationData[1].about,
                attestationData[1].key
            ),
            attestationData[1].val
        );
        assertEq(
            attestationStation.attestations(
                alice_attestor,
                attestationData[2].about,
                attestationData[2].key
            ),
            attestationData[2].val
        );
    }
Will Cory's avatar
Will Cory committed
144
}