from subprocess import check_output
from tests import *
MYTH = str ( PROJECT_DIR / " myth " )
def output_of ( command ) :
return check_output ( command , shell = True ) . decode ( " UTF-8 " )
class CommandLineToolTestCase ( BaseTestCase ) :
def test_disassemble_code_correctly ( self ) :
command = " python3 {} MYTH -d --bin-runtime -c 0x5050 --solv 0.5.0 " . format ( MYTH )
self . assertIn ( " 0 POP \n 1 POP \n " , output_of ( command ) )
def test_disassemble_solidity_file_correctly ( self ) :
solidity_file = str ( TESTDATA / " input_contracts " / " metacoin.sol " )
command = " python3 {} -d {} --solv 0.5.0 " . format ( MYTH , solidity_file )
self . assertIn ( " 2 PUSH1 0x40 \n 4 MSTORE " , output_of ( command ) )
def test_hash_a_function_correctly ( self ) :
command = " python3 {} --solv 0.5.0 --hash ' setOwner(address) ' " . format ( MYTH )
self . assertIn ( " 0x13af4035 \n " , output_of ( command ) )
class TruffleTestCase ( BaseTestCase ) :
def test_analysis_truffle_project ( self ) :
truffle_project_root = str ( TESTS_DIR / " truffle_project " )
command = " cd {} ; truffle compile; python3 {} --truffle -t 1 " . format (
truffle_project_root , MYTH
)
self . assertIn ( " === Ether thief ==== " , output_of ( command ) )
class InfuraTestCase ( BaseTestCase ) :
def test_infura_mainnet ( self ) :
command = " python3 {} --rpc infura-mainnet -d -a 0x2a0c0dbecc7e4d658f48e01e3fa353f44050c208 " . format (
MYTH
)
output = output_of ( command )
self . assertIn ( " 0 PUSH1 0x60 \n 2 PUSH1 0x40 \n 4 MSTORE " , output )
self . assertIn ( " 7278 POP \n 7279 POP \n 7280 JUMP \n 7281 STOP " , output )
def test_infura_rinkeby ( self ) :
command = " python3 {} --rpc infura-rinkeby -d -a 0xB6f2bFED892a662bBF26258ceDD443f50Fa307F5 " . format (
MYTH
)
output = output_of ( command )
self . assertIn ( " 34 JUMPDEST \n 35 CALLVALUE " , output )
def test_infura_kovan ( self ) :
command = " python3 {} --rpc infura-kovan -d -a 0xE6bBF9B5A3451242F82f8cd458675092617a1235 " . format (
MYTH
)
output = output_of ( command )
self . assertIn ( " 9999 PUSH1 0x00 \n 10001 NOT \n 10002 AND \n 10003 PUSH1 0x00 " , output )
def test_infura_ropsten ( self ) :
command = " python3 {} --rpc infura-ropsten -d -a 0x6e0E0e02377Bc1d90E8a7c21f12BA385C2C35f78 " . format (
MYTH
)
output = output_of ( command )
self . assertIn ( " 1821 PUSH1 0x01 \n 1823 PUSH2 0x070c " , output )