mirror of
https://github.com/FlipsideCrypto/web3.py.git
synced 2026-02-06 10:56:47 +00:00
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
import pytest
|
|
|
|
from web3.utils.abi import (
|
|
function_abi_to_4byte_selector,
|
|
)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def wait_for_first_block(web3, wait_for_block):
|
|
wait_for_block(web3)
|
|
|
|
|
|
@pytest.fixture()
|
|
def math_contract(web3, MATH_ABI, MATH_CODE, MATH_RUNTIME, MATH_SOURCE,
|
|
wait_for_transaction):
|
|
MathContract = web3.eth.contract(
|
|
abi=MATH_ABI,
|
|
bytecode=MATH_CODE,
|
|
bytecode_runtime=MATH_RUNTIME,
|
|
source=MATH_SOURCE,
|
|
)
|
|
deploy_txn = MathContract.deploy({'from': web3.eth.coinbase})
|
|
deploy_receipt = wait_for_transaction(web3, deploy_txn)
|
|
|
|
assert deploy_receipt is not None
|
|
contract_address = deploy_receipt['contractAddress']
|
|
web3.isAddress(contract_address)
|
|
|
|
_math_contract = MathContract(address=contract_address)
|
|
return _math_contract
|
|
|
|
|
|
def test_contract_estimateGas(web3, math_contract):
|
|
increment_abi = math_contract._find_matching_fn_abi('increment', [])
|
|
call_data = function_abi_to_4byte_selector(increment_abi)
|
|
gas_estimate = math_contract.estimateGas().increment()
|
|
|
|
try:
|
|
assert abs(gas_estimate - 21472) < 200 # Geth
|
|
except AssertionError:
|
|
assert abs(gas_estimate - 43020) < 200 # TestRPC
|
|
pass
|