pragma solidity ^0.8.4;

import { DSTest } from "ds-test/test.sol";
import { CheatCodes } from "./utils/CheatCodes.sol";

library Library {
    function willRevert() internal pure {
        revert("reason");
    }
}

contract RevertTest is DSTest {
    CheatCodes private constant VM = CheatCodes(HEVM_ADDRESS);

    function testExternal() external {
        VM.expectRevert("reason");
        this.willRevert();
    }

    function testInternal() external {
        VM.expectRevert("reason");
        willRevert();
    }

    function testLibrary() external {
        VM.expectRevert("reason");
        Library.willRevert();
    }

    function willRevert() public pure {
        revert("reason");
    }
}