mirror of https://github.com/ConsenSys/mythril
blockchainethereumsmart-contractssoliditysecurityprogram-analysissecurity-analysissymbolic-execution
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.
51 lines
1.6 KiB
51 lines
1.6 KiB
from pathlib import Path
|
|
from unittest import TestCase
|
|
|
|
TESTS_DIR = Path(__file__).parent
|
|
PROJECT_DIR = TESTS_DIR.parent
|
|
TESTDATA = TESTS_DIR / "testdata"
|
|
TESTDATA_INPUTS = TESTDATA / "inputs"
|
|
TESTDATA_INPUTS_CONTRACTS = TESTDATA / "input_contracts"
|
|
TESTDATA_OUTPUTS_EXPECTED = TESTDATA / "outputs_expected"
|
|
TESTDATA_OUTPUTS_CURRENT = TESTDATA / "outputs_current"
|
|
TESTDATA_OUTPUTS_CURRENT_LASER_RESULT = TESTDATA / "outputs_current_laser_result"
|
|
TESTDATA_OUTPUTS_EXPECTED_LASER_RESULT = TESTDATA / "outputs_expected_laser_result"
|
|
MYTHRIL_DIR = TESTS_DIR / "mythril_dir"
|
|
|
|
TESTDATA_OUTPUTS_CURRENT.mkdir(exist_ok=True)
|
|
TESTDATA_OUTPUTS_CURRENT_LASER_RESULT.mkdir(exist_ok=True)
|
|
|
|
|
|
class BaseTestCase(TestCase):
|
|
def setUp(self):
|
|
""""""
|
|
self.changed_files = []
|
|
|
|
def compare_files_error_message(self):
|
|
"""
|
|
|
|
:return:
|
|
"""
|
|
message = "Following output files are changed, compare them manually to see differences: \n"
|
|
|
|
for input_file, expected, current in self.changed_files:
|
|
message += "{}:\n".format(input_file.name)
|
|
message += "- {}\n".format(str(expected))
|
|
message += "- {}\n".format(str(current))
|
|
|
|
return message
|
|
|
|
def found_changed_files(self, input_file, output_expected, output_current):
|
|
"""
|
|
|
|
:param input_file:
|
|
:param output_expected:
|
|
:param output_current:
|
|
"""
|
|
self.changed_files.append((input_file, output_expected, output_current))
|
|
|
|
def assert_and_show_changed_files(self):
|
|
""""""
|
|
self.assertEqual(
|
|
0, len(self.changed_files), msg=self.compare_files_error_message()
|
|
)
|
|
|