mirror of https://github.com/crytic/slither
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.8 KiB
48 lines
1.8 KiB
import re
|
|
from collections import Counter
|
|
from pathlib import Path
|
|
|
|
from crytic_compile import CryticCompile
|
|
from crytic_compile.platform.solc_standard_json import SolcStandardJson
|
|
|
|
from slither import Slither
|
|
from slither.printers.inheritance.inheritance_graph import PrinterInheritanceGraph
|
|
|
|
|
|
TEST_DATA_DIR = Path(__file__).resolve().parent / "test_data"
|
|
|
|
|
|
def test_inheritance_printer(solc_binary_path) -> None:
|
|
solc_path = solc_binary_path("0.8.0")
|
|
standard_json = SolcStandardJson()
|
|
standard_json.add_source_file(Path(TEST_DATA_DIR, "test_contract_names", "A.sol").as_posix())
|
|
standard_json.add_source_file(Path(TEST_DATA_DIR, "test_contract_names", "B.sol").as_posix())
|
|
standard_json.add_source_file(Path(TEST_DATA_DIR, "test_contract_names", "B2.sol").as_posix())
|
|
standard_json.add_source_file(Path(TEST_DATA_DIR, "test_contract_names", "C.sol").as_posix())
|
|
compilation = CryticCompile(standard_json, solc=solc_path)
|
|
slither = Slither(compilation)
|
|
printer = PrinterInheritanceGraph(slither=slither, logger=None)
|
|
|
|
output = printer.output("test_printer.dot")
|
|
content = output.elements[0]["name"]["content"]
|
|
|
|
pattern = re.compile(r"(?:c\d+_)?(\w+ -> )(?:c\d+_)(\w+)")
|
|
matches = re.findall(pattern, content)
|
|
relations = ["".join(m) for m in matches]
|
|
|
|
counter = Counter(relations)
|
|
|
|
assert counter["B -> A"] == 2
|
|
assert counter["C -> A"] == 1
|
|
|
|
# Lets also test the include/exclude interface behavior
|
|
# Check that the interface is not included
|
|
assert "MyInterfaceX" not in content
|
|
|
|
slither.include_interfaces = True
|
|
output = printer.output("test_printer.dot")
|
|
content = output.elements[0]["name"]["content"]
|
|
assert "MyInterfaceX" in content
|
|
|
|
# Remove test generated files
|
|
Path("test_printer.dot").unlink(missing_ok=True)
|
|
|