minime/certora/harness/MiniMeTokenHarness.sol

// SPDX-License-Identifier: GPL-3.0
pragma solidity 0.8.18;

import { MiniMeToken } from "../munged/MiniMeToken.sol";

contract MiniMeTokenHarness is MiniMeToken {

    error MinimeTokenHarness_InvalidCheckpointLength();

    constructor(
        MiniMeToken _parentToken,
        uint256 _parentSnapShotBlock,
        string memory _tokenName,
        uint8 _decimalUnits,
        string memory _tokenSymbol,
        bool _transfersEnabled
    ) public MiniMeToken(
      _parentToken, 
      _parentSnapShotBlock, 
      _tokenName, 
      _decimalUnits, 
      _tokenSymbol, 
      _transfersEnabled
    ) {}

    function getCheckpointsLengthByAddress(address user) public view returns (uint256) {
        return balances[user].length;
    }

    function getLatestBlockNumberByAddress(address user) public view returns (uint256) {
        uint256 checkpointsLength = getCheckpointsLengthByAddress(user);
        if (checkpointsLength == 0) revert MinimeTokenHarness_InvalidCheckpointLength();
        Checkpoint memory latestCheckPoint = getCheckpointByAddressAndIndex(user, checkpointsLength - 1);
        return latestCheckPoint.fromBlock;
    }

    function getCheckpointByAddressAndIndex(address user, uint256 index) public view returns (Checkpoint memory) {
        uint256 checkpointsLength = getCheckpointsLengthByAddress(user);
        if (checkpointsLength == 0 || index >= checkpointsLength) revert MinimeTokenHarness_InvalidCheckpointLength();
        return balances[user][index];
    }

    function getFromBlockByAddressAndIndex(address user, uint256 index) public view returns (uint256) {
        uint256 checkpointsLength = getCheckpointsLengthByAddress(user);
        if (checkpointsLength == 0 || index >= checkpointsLength) revert MinimeTokenHarness_InvalidCheckpointLength();
        return balances[user][index].fromBlock;
    }
}