slither/tests/unit/utils/test_code_generation.py

import os
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() -> None:
    solc_select.switch_global_version("0.8.4", always_install=True)

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

    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

    actual = generate_interface(sl.get_contract_from_name("TestContract")[0], unroll_structs=False)
    expected_path = os.path.join(CODE_TEST_ROOT, "TEST_generated_code_not_unrolled.sol")

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

    assert actual == expected