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.
27 lines
787 B
27 lines
787 B
1 year ago
|
from typing import Dict
|
||
|
|
from pathlib import Path
|
||
|
|
from slither import Slither
|
||
|
|
|
||
|
|
TEST_DATA_DIR = Path(__file__).resolve().parent / "test_data"
|
||
|
|
|
||
|
|
ALL_TESTS = list(Path(TEST_DATA_DIR).glob("*.vy"))
|
||
|
|
|
||
|
|
|
||
|
|
def pytest_generate_tests(metafunc):
|
||
|
|
test_cases = []
|
||
|
|
for test_file in ALL_TESTS:
|
||
|
|
sl = Slither(test_file.as_posix())
|
||
|
|
for contract in sl.contracts:
|
||
|
|
if contract.is_interface:
|
||
|
|
continue
|
||
|
|
for func_or_modifier in contract.functions:
|
||
|
|
test_cases.append(
|
||
|
|
(func_or_modifier.canonical_name, func_or_modifier.slithir_cfg_to_dot_str())
|
||
|
|
)
|
||
|
|
|
||
|
|
metafunc.parametrize("test_case", test_cases, ids=lambda x: x[0])
|
||
|
|
|
||
|
|
|
||
|
|
def test_vyper_cfgir(test_case, snapshot):
|
||
|
|
assert snapshot() == test_case[1]
|