slither/tests/unit/utils/test_code_generation.py

from pathlib import Path
from solc_select import solc_select

from slither import Slither
from slither.utils.code_generation import (
    generate_interface,
)

TEST_DATA_DIR = Path(__file__).resolve().parent / "test_data" / "code_generation"


def test_interface_generation(use_solc_version) -> None:
    solc_path = next(use_solc_version("0.8.4"))

    sl = Slither(Path(TEST_DATA_DIR, "CodeGeneration.sol").as_posix(), solc=solc_path)

    actual = generate_interface(sl.get_contract_from_name("TestContract")[0])
    expected_path = Path(TEST_DATA_DIR, "TEST_generated_code.sol").as_posix()

    with open(expected_path, "r", encoding="utf-8") as file:
        expected = file.read()

    assert actual == expected