onchain-id/test/identities/VersionUpgrade.t.sol

// SPDX-License-Identifier: GPL-3.0
pragma solidity ^0.8.27;

import { ClaimSignerHelper } from "../helpers/ClaimSignerHelper.sol";
import { OnchainIDSetup } from "../helpers/OnchainIDSetup.sol";
import { KeyPurposes } from "contracts/libraries/KeyPurposes.sol";
import { KeyTypes } from "contracts/libraries/KeyTypes.sol";

contract VersionUpgradeTest is OnchainIDSetup {

    function test_returnInitialVersion() public view {
        assertEq(aliceIdentity.version(), "3.0.0");
    }

    function test_maintainVersionConstant() public view {
        assertEq(aliceIdentity.version(), "3.0.0");
    }

    function test_demonstrateUpgradePattern() public {
        assertEq(aliceIdentity.version(), "3.0.0");

        uint256 claimTopic = uint256(keccak256(bytes("test")));
        bytes memory claimData = bytes("test data");
        string memory claimUri = "https://example.com";

        // Add CLAIM_SIGNER key for alice on her identity
        vm.prank(alice);
        aliceIdentity.addKey(ClaimSignerHelper.addressToKey(alice), KeyPurposes.CLAIM_SIGNER, KeyTypes.ECDSA);

        // Sign claim properly
        bytes memory signature = ClaimSignerHelper.signClaim(alicePk, address(aliceIdentity), claimTopic, claimData);

        // Add self-issued claim with valid signature
        vm.prank(alice);
        aliceIdentity.addClaim(claimTopic, 1, address(aliceIdentity), signature, claimData, claimUri);

        // Verify claim
        bytes32 claimId = keccak256(abi.encode(address(aliceIdentity), claimTopic));
        (uint256 topic,, address returnedIssuer,, bytes memory data, string memory uri) =
            aliceIdentity.getClaim(claimId);

        assertEq(topic, claimTopic);
        assertEq(returnedIssuer, address(aliceIdentity));
        assertEq(data, claimData);
        assertEq(uri, claimUri);
    }

}