99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
# external imports
|
|
from eth_erc20 import ERC20
|
|
from chainlib.connection import RPCConnection
|
|
from chainlib.eth.nonce import RPCNonceOracle
|
|
from chainlib.eth.gas import (
|
|
Gas,
|
|
OverrideGasOracle,
|
|
)
|
|
from chainlib.eth.tx import (
|
|
TxFormat,
|
|
receipt,
|
|
raw,
|
|
unpack,
|
|
Tx,
|
|
)
|
|
from chainlib.eth.block import (
|
|
Block,
|
|
block_latest,
|
|
block_by_number,
|
|
)
|
|
from chainlib.eth.address import is_same_address
|
|
from chainlib.eth.contract import ABIContractEncoder
|
|
from hexathon import strip_0x
|
|
from eth_token_index import TokenUniqueSymbolIndex
|
|
|
|
# local imports
|
|
from cic_eth.runnable.daemons.filters.token import TokenFilter
|
|
from cic_eth.db.models.gas_cache import GasCache
|
|
from cic_eth.db.models.base import SessionBase
|
|
|
|
|
|
def test_filter_gas(
|
|
default_chain_spec,
|
|
init_database,
|
|
eth_rpc,
|
|
eth_signer,
|
|
contract_roles,
|
|
agent_roles,
|
|
token_roles,
|
|
foo_token,
|
|
token_registry,
|
|
register_lookups,
|
|
register_tokens,
|
|
celery_session_worker,
|
|
cic_registry,
|
|
):
|
|
|
|
rpc = RPCConnection.connect(default_chain_spec, 'default')
|
|
nonce_oracle = RPCNonceOracle(token_roles['FOO_TOKEN_OWNER'], eth_rpc)
|
|
gas_oracle = OverrideGasOracle(price=1000000000, limit=1000000)
|
|
c = ERC20(default_chain_spec, signer=eth_signer, nonce_oracle=nonce_oracle, gas_oracle=gas_oracle)
|
|
(tx_hash_hex, tx_signed_raw_hex) = c.transfer(foo_token, token_roles['FOO_TOKEN_OWNER'], agent_roles['ALICE'], 100, tx_format=TxFormat.RLP_SIGNED)
|
|
o = raw(tx_signed_raw_hex)
|
|
eth_rpc.do(o)
|
|
o = receipt(tx_hash_hex)
|
|
rcpt = eth_rpc.do(o)
|
|
assert rcpt['status'] == 1
|
|
|
|
fltr = TokenFilter(default_chain_spec, queue=None, call_address=agent_roles['ALICE'])
|
|
|
|
o = block_latest()
|
|
r = eth_rpc.do(o)
|
|
o = block_by_number(r, include_tx=False)
|
|
r = eth_rpc.do(o)
|
|
block = Block(r)
|
|
block.txs = [tx_hash_hex]
|
|
|
|
tx_signed_raw_bytes = bytes.fromhex(strip_0x(tx_signed_raw_hex))
|
|
tx_src = unpack(tx_signed_raw_bytes, default_chain_spec)
|
|
tx = Tx(tx_src, block=block)
|
|
tx.apply_receipt(rcpt)
|
|
t = fltr.filter(eth_rpc, block, tx, db_session=init_database)
|
|
assert t.get() == None
|
|
|
|
nonce_oracle = RPCNonceOracle(contract_roles['CONTRACT_DEPLOYER'], eth_rpc)
|
|
c = TokenUniqueSymbolIndex(default_chain_spec, signer=eth_signer, nonce_oracle=nonce_oracle)
|
|
(tx_hash_hex_register, o) = c.register(token_registry, contract_roles['CONTRACT_DEPLOYER'], foo_token)
|
|
eth_rpc.do(o)
|
|
o = receipt(tx_hash_hex)
|
|
r = eth_rpc.do(o)
|
|
assert r['status'] == 1
|
|
|
|
t = fltr.filter(eth_rpc, block, tx, db_session=init_database)
|
|
r = t.get_leaf()
|
|
assert t.successful()
|
|
|
|
q = init_database.query(GasCache)
|
|
q = q.filter(GasCache.tx_hash==strip_0x(tx_hash_hex))
|
|
o = q.first()
|
|
|
|
assert is_same_address(o.address, strip_0x(foo_token))
|
|
assert o.value > 0
|
|
|
|
enc = ABIContractEncoder()
|
|
enc.method('transfer')
|
|
method = enc.get()
|
|
|
|
assert o.method == method
|