slither/tests/unit/core/test_inheritance.py

from pathlib import Path

from crytic_compile import CryticCompile
from crytic_compile.platform.solc_standard_json import SolcStandardJson
from slither import Slither

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

# https://github.com/crytic/slither/issues/2304
def test_inheritance_with_renaming(solc_binary_path) -> None:
    solc_path = solc_binary_path("0.8.15")
    standard_json = SolcStandardJson()
    for source_file in Path(TEST_DATA_DIR).rglob("*.sol"):
        print(source_file)
        standard_json.add_source_file(Path(source_file).as_posix())
    compilation = CryticCompile(standard_json, solc=solc_path)
    slither = Slither(compilation)

    a = slither.get_contract_from_name("A")[0]
    b = slither.get_contract_from_name("B")[0]
    c = slither.get_contract_from_name("C")[0]

    assert len(a.immediate_inheritance) == 1
    assert a.immediate_inheritance[0] == b
    assert len(a.inheritance) == 2
    assert a.inheritance[0] == b
    assert a.inheritance[1] == c
    assert len(a.explicit_base_constructor_calls) == 1
    a_base_constructor_call = a.explicit_base_constructor_calls[0]
    assert a_base_constructor_call == b.constructor

    assert len(b.inheritance) == 1
    assert b.inheritance[0] == c
    assert len(b.immediate_inheritance) == 1
    assert b.immediate_inheritance[0] == c
    assert len(b.explicit_base_constructor_calls) == 0