89 lines
2.4 KiB
Python
89 lines
2.4 KiB
Python
import pytest
|
|
|
|
from vyper import compile_code
|
|
|
|
from eth_tester import (
|
|
EthereumTester,
|
|
)
|
|
from web3.providers.eth_tester import (
|
|
EthereumTesterProvider,
|
|
)
|
|
from web3 import (
|
|
Web3,
|
|
)
|
|
from vdb.vdb import (
|
|
VyperDebugCmd
|
|
)
|
|
from vdb.eth_tester_debug_backend import (
|
|
PyEVMDebugBackend,
|
|
set_debug_info
|
|
)
|
|
from vdb.source_map import (
|
|
produce_source_map
|
|
)
|
|
|
|
|
|
@pytest.fixture()
|
|
def tester():
|
|
t = EthereumTester(backend=PyEVMDebugBackend())
|
|
return t
|
|
|
|
|
|
def zero_gas_price_strategy(web3, transaction_params=None):
|
|
return 0 # zero gas price makes testing simpler.
|
|
|
|
|
|
@pytest.fixture()
|
|
def w3(tester):
|
|
w3 = Web3(EthereumTesterProvider(tester))
|
|
w3.eth.setGasPriceStrategy(zero_gas_price_strategy)
|
|
return w3
|
|
|
|
|
|
def _get_contract(w3, source_code, *args, **kwargs):
|
|
compiler_output = compile_code(source_code, ['bytecode', 'abi'])
|
|
abi = compiler_output['abi']
|
|
bytecode = compiler_output['bytecode']
|
|
contract = w3.eth.contract(abi=abi, bytecode=bytecode)
|
|
|
|
stdin = kwargs['stdin'] if 'stdin' in kwargs else None
|
|
stdout = kwargs['stdout'] if 'stdout' in kwargs else None
|
|
|
|
source_map = produce_source_map(source_code)
|
|
set_debug_info(source_code, source_map, stdin, stdout)
|
|
import vdb
|
|
setattr(vdb.debug_computation.DebugComputation, 'enable_debug', True)
|
|
value = kwargs.pop('value', 0)
|
|
value_in_eth = kwargs.pop('value_in_eth', 0)
|
|
value = value_in_eth * 10**18 if value_in_eth else value # Handle deploying with an eth value.
|
|
gasPrice = kwargs.pop('gasPrice', 0)
|
|
deploy_transaction = {
|
|
'from': w3.eth.accounts[0],
|
|
'data': contract._encode_constructor_data(args, kwargs),
|
|
'value': value,
|
|
'gasPrice': gasPrice
|
|
}
|
|
tx = w3.eth.sendTransaction(deploy_transaction)
|
|
address = w3.eth.getTransactionReceipt(tx)['contractAddress']
|
|
contract = w3.eth.contract(address, abi=abi, bytecode=bytecode)
|
|
# Filter logs.
|
|
contract._logfilter = w3.eth.filter({
|
|
'fromBlock': w3.eth.blockNumber - 1,
|
|
'address': contract.address
|
|
})
|
|
return contract
|
|
|
|
|
|
@pytest.fixture
|
|
def get_contract(w3):
|
|
def get_contract(source_code, *args, **kwargs):
|
|
return _get_contract(w3, source_code, *args, **kwargs)
|
|
return get_contract
|
|
|
|
|
|
@pytest.fixture
|
|
def get_last_out():
|
|
def _get_last_out(stdout):
|
|
return stdout.getvalue().splitlines()[-2].split(VyperDebugCmd.prompt)[1]
|
|
return _get_last_out
|