217 lines
6.7 KiB
Python
217 lines
6.7 KiB
Python
# standard imports
|
|
import logging
|
|
|
|
# external imports
|
|
import pytest
|
|
from chainlib.connection import RPCConnection
|
|
from chainlib.eth.nonce import (
|
|
OverrideNonceOracle,
|
|
RPCNonceOracle,
|
|
)
|
|
from chainlib.eth.tx import (
|
|
TxFormat,
|
|
unpack,
|
|
Tx,
|
|
receipt,
|
|
)
|
|
from chainlib.eth.gas import (
|
|
Gas,
|
|
OverrideGasOracle,
|
|
)
|
|
from chainlib.eth.block import (
|
|
block_latest,
|
|
block_by_number,
|
|
Block,
|
|
)
|
|
from chainqueue.db.models.otx import Otx
|
|
from chainqueue.db.enum import StatusBits
|
|
from chainqueue.sql.tx import create as queue_create
|
|
from chainqueue.sql.state import (
|
|
set_reserved,
|
|
set_ready,
|
|
set_sent,
|
|
set_waitforgas,
|
|
)
|
|
from cic_eth.error import OutOfGasError
|
|
|
|
from hexathon import (
|
|
strip_0x,
|
|
uniform as hex_uniform,
|
|
)
|
|
from cic_eth.eth.gas import cache_gas_data
|
|
from cic_eth.queue.query import (
|
|
get_tx_local,
|
|
get_account_tx_local,
|
|
)
|
|
|
|
# local imports
|
|
from cic_sync_filter.straggler import StragglerFilter
|
|
|
|
logg = logging.getLogger()
|
|
|
|
|
|
def test_straggler_tx(
|
|
default_chain_spec,
|
|
init_database,
|
|
eth_rpc,
|
|
eth_signer,
|
|
agent_roles,
|
|
celery_session_worker,
|
|
contract_roles,
|
|
):
|
|
|
|
rpc = RPCConnection.connect(default_chain_spec, 'default')
|
|
nonce_oracle = OverrideNonceOracle(agent_roles['ALICE'], 42)
|
|
gas_oracle = OverrideGasOracle(price=1000000000, limit=21000)
|
|
c = Gas(default_chain_spec, signer=eth_signer, nonce_oracle=nonce_oracle, gas_oracle=gas_oracle)
|
|
(tx_hash_hex, tx_signed_raw_hex) = c.create(agent_roles['ALICE'], agent_roles['BOB'], 100 * (10 ** 6), tx_format=TxFormat.RLP_SIGNED)
|
|
queue_create(
|
|
default_chain_spec,
|
|
42,
|
|
agent_roles['ALICE'],
|
|
tx_hash_hex,
|
|
tx_signed_raw_hex,
|
|
session=init_database,
|
|
)
|
|
cache_gas_data(
|
|
tx_hash_hex,
|
|
tx_signed_raw_hex,
|
|
default_chain_spec.asdict(),
|
|
)
|
|
|
|
set_ready(default_chain_spec, tx_hash_hex, session=init_database)
|
|
set_reserved(default_chain_spec, tx_hash_hex, session=init_database)
|
|
set_sent(default_chain_spec, tx_hash_hex, session=init_database)
|
|
|
|
queue = None
|
|
fltr = StragglerFilter(default_chain_spec, None, queue, caller_address=contract_roles['CONTRACT_DEPLOYER'])
|
|
|
|
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)
|
|
t = fltr.filter(None, block, tx, db_session=init_database)
|
|
logg.debug('foo')
|
|
tx_hash_hex_successor = t.get_leaf()
|
|
logg.debug('bar')
|
|
|
|
assert t.successful()
|
|
assert tx_hash_hex_successor != tx_hash_hex
|
|
|
|
otx = Otx.load(tx_hash_hex, session=init_database)
|
|
assert otx.status & StatusBits.OBSOLETE > 0
|
|
assert otx.status & (StatusBits.FINAL | StatusBits.QUEUED | StatusBits.RESERVED) == 0
|
|
|
|
otx = Otx.load(tx_hash_hex_successor, session=init_database)
|
|
assert otx.status == StatusBits.QUEUED
|
|
|
|
|
|
|
|
def test_waitforgas_tx(
|
|
default_chain_spec,
|
|
init_database,
|
|
eth_rpc,
|
|
eth_signer,
|
|
agent_roles,
|
|
celery_session_worker,
|
|
whoever,
|
|
contract_roles,
|
|
):
|
|
|
|
safe_gas = 1000000000000000000
|
|
|
|
rpc = RPCConnection.connect(default_chain_spec, 'default')
|
|
nonce_oracle = OverrideNonceOracle(whoever, 0)
|
|
gas_oracle = OverrideGasOracle(price=1000000000, limit=21000)
|
|
c = Gas(default_chain_spec, signer=eth_signer, nonce_oracle=nonce_oracle, gas_oracle=gas_oracle)
|
|
(tx_hash_hex, tx_signed_raw_hex) = c.create(whoever, agent_roles['BOB'], 100 * (10 ** 6), tx_format=TxFormat.RLP_SIGNED)
|
|
queue_create(
|
|
default_chain_spec,
|
|
0,
|
|
whoever,
|
|
tx_hash_hex,
|
|
tx_signed_raw_hex,
|
|
session=init_database,
|
|
)
|
|
cache_gas_data(
|
|
tx_hash_hex,
|
|
tx_signed_raw_hex,
|
|
default_chain_spec.asdict(),
|
|
)
|
|
|
|
set_ready(default_chain_spec, tx_hash_hex, session=init_database)
|
|
set_waitforgas(default_chain_spec, tx_hash_hex, session=init_database)
|
|
|
|
queue = None
|
|
StragglerFilter.gas_balance_threshold = safe_gas
|
|
fltr = StragglerFilter(default_chain_spec, None, queue, caller_address=contract_roles['CONTRACT_DEPLOYER'])
|
|
|
|
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)
|
|
|
|
t = fltr.filter(eth_rpc, block, tx, db_session=init_database)
|
|
t.get_leaf()
|
|
assert t.successful()
|
|
|
|
otx = get_tx_local(default_chain_spec, tx.hash, session=init_database)
|
|
assert otx['status'] == StatusBits.GAS_ISSUES
|
|
|
|
nonce_oracle = RPCNonceOracle(agent_roles['CAROL'], conn=eth_rpc)
|
|
gas_oracle = OverrideGasOracle(price=1000000000, limit=21000)
|
|
c = Gas(default_chain_spec, signer=eth_signer, nonce_oracle=nonce_oracle, gas_oracle=gas_oracle)
|
|
(tx_hash_hex, o) = c.create(agent_roles['CAROL'], whoever, safe_gas - 1)
|
|
r = 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)
|
|
t.get_leaf()
|
|
assert t.successful()
|
|
|
|
otx = get_tx_local(default_chain_spec, tx.hash, session=init_database)
|
|
assert otx['status'] == StatusBits.GAS_ISSUES
|
|
|
|
|
|
nonce_oracle = RPCNonceOracle(agent_roles['CAROL'], conn=eth_rpc)
|
|
gas_oracle = OverrideGasOracle(price=1000000000, limit=21000)
|
|
c = Gas(default_chain_spec, signer=eth_signer, nonce_oracle=nonce_oracle, gas_oracle=gas_oracle)
|
|
(tx_hash_hex, o) = c.create(agent_roles['CAROL'], whoever, 1)
|
|
r = eth_rpc.do(o)
|
|
|
|
o = receipt(tx_hash_hex)
|
|
r = eth_rpc.do(o)
|
|
assert r['status'] == 1
|
|
|
|
init_database.commit()
|
|
|
|
t = fltr.filter(eth_rpc, block, tx, db_session=init_database)
|
|
t.get_leaf()
|
|
|
|
otx = get_tx_local(default_chain_spec, tx.hash, session=init_database)
|
|
assert otx['status'] & StatusBits.OBSOLETE > 0
|
|
|
|
txs = get_account_tx_local(default_chain_spec, whoever, session=init_database)
|
|
assert len(txs.keys()) == 2
|
|
for k in txs.keys():
|
|
if hex_uniform(strip_0x(tx.hash)) != hex_uniform(strip_0x(k)):
|
|
otx = get_tx_local(default_chain_spec, k, session=init_database)
|
|
assert otx['status'] == StatusBits.QUEUED
|
|
|