openzeppelin-contracts/contracts/mocks/ERC4626Mock.sol

// SPDX-License-Identifier: MIT

pragma solidity ^0.8.0;

import "../token/ERC20/extensions/ERC4626.sol";

contract ERC4626Mock is ERC4626 {
    constructor(IERC20Metadata asset, string memory name, string memory symbol) ERC20(name, symbol) ERC4626(asset) {}

    function mockMint(address account, uint256 amount) public {
        _mint(account, amount);
    }

    function mockBurn(address account, uint256 amount) public {
        _burn(account, amount);
    }
}

contract ERC4626DecimalMock is ERC4626Mock {
    using Math for uint256;

    uint8 private immutable _decimals;

    constructor(
        IERC20Metadata asset,
        string memory name,
        string memory symbol,
        uint8 decimalsOverride
    ) ERC4626Mock(asset, name, symbol) {
        _decimals = decimalsOverride;
    }

    function decimals() public view virtual override returns (uint8) {
        return _decimals;
    }

    function _initialConvertToShares(
        uint256 assets,
        Math.Rounding rounding
    ) internal view virtual override returns (uint256 shares) {
        return assets.mulDiv(10 ** decimals(), 10 ** super.decimals(), rounding);
    }

    function _initialConvertToAssets(
        uint256 shares,
        Math.Rounding rounding
    ) internal view virtual override returns (uint256 assets) {
        return shares.mulDiv(10 ** super.decimals(), 10 ** decimals(), rounding);
    }
}