mythril/tests/integration_tests/version_test.py

import pytest
import json
import sys

from tests import PROJECT_DIR, TESTDATA
from utils import output_of

MYTH = str(PROJECT_DIR / "myth")
test_data = (
    ("version_contract.sol", "v0.7.0", True),
    ("version_contract.sol", "v0.8.0", False),
    ("version_contract_0.8.0.sol", None, False),
    ("version_contract_0.7.0.sol", None, True),
    ("version_chaos.sol", None, True),
    ("version_2.sol", None, True),
    ("version_3.sol", None, True),
    ("version_patch.sol", None, False),
    ("integer_edge_case.sol", None, True),
    ("integer_edge_case.sol", "v0.8.19", True),
)


@pytest.mark.parametrize("file_name, version, has_overflow", test_data)
def test_analysis(file_name, version, has_overflow):
    file = str(TESTDATA / "input_contracts" / file_name)
    if version:
        command = f"python3 {MYTH} analyze {file} --solv {version}"
    else:
        command = f"python3 {MYTH} analyze {file}"
    output = output_of(command)
    if has_overflow:
        assert f"SWC ID: 101" in output
    else:
        assert (
            "The analysis was completed successfully. No issues were detected."
            in output
        )