374 lines
12 KiB
Python
374 lines
12 KiB
Python
|
# standard imports
|
||
|
|
import logging
|
||
|
|
import io
|
||
|
|
import json
|
||
|
|
|
||
|
|
# external imports
|
||
|
|
import pytest
|
||
|
|
from chainlib.connection import RPCConnection
|
||
|
|
from chainlib.eth.nonce import (
|
||
|
|
nonce,
|
||
|
|
OverrideNonceOracle,
|
||
|
|
RPCNonceOracle,
|
||
|
|
)
|
||
|
|
from chainqueue.tx import create as queue_create
|
||
|
|
from chainlib.eth.tx import (
|
||
|
|
raw,
|
||
|
|
receipt,
|
||
|
|
TxFormat,
|
||
|
|
Tx,
|
||
|
|
)
|
||
|
|
from chainlib.eth.block import block_latest
|
||
|
|
from chainlib.eth.gas import (
|
||
|
|
Gas,
|
||
|
|
OverrideGasOracle,
|
||
|
|
)
|
||
|
|
from chainqueue.state import (
|
||
|
|
set_reserved,
|
||
|
|
set_sent,
|
||
|
|
set_ready,
|
||
|
|
)
|
||
|
|
from chainqueue.db.models.otx import Otx
|
||
|
|
from chainqueue.db.enum import StatusBits
|
||
|
|
from chainqueue.query import get_nonce_tx_cache
|
||
|
|
from eth_erc20 import ERC20
|
||
|
|
from cic_eth_registry import CICRegistry
|
||
|
|
|
||
|
|
# local imports
|
||
|
|
from cic_eth.api.api_admin import AdminApi
|
||
|
|
from cic_eth.eth.gas import cache_gas_data
|
||
|
|
from cic_eth.eth.erc20 import cache_transfer_data
|
||
|
|
|
||
|
|
logg = logging.getLogger()
|
||
|
|
|
||
|
|
|
||
|
|
def test_admin_api_tx(
|
||
|
|
default_chain_spec,
|
||
|
|
init_database,
|
||
|
|
init_celery_tasks,
|
||
|
|
eth_rpc,
|
||
|
|
eth_signer,
|
||
|
|
agent_roles,
|
||
|
|
contract_roles,
|
||
|
|
custodial_roles,
|
||
|
|
celery_session_worker,
|
||
|
|
foo_token,
|
||
|
|
address_declarator,
|
||
|
|
cic_registry,
|
||
|
|
register_tokens,
|
||
|
|
register_lookups,
|
||
|
|
caplog,
|
||
|
|
):
|
||
|
|
|
||
|
|
nonce_oracle = RPCNonceOracle(custodial_roles['FOO_TOKEN_GIFTER'], conn=eth_rpc)
|
||
|
|
gas_oracle = OverrideGasOracle(limit=100000, conn=eth_rpc)
|
||
|
|
|
||
|
|
o = nonce(custodial_roles['FOO_TOKEN_GIFTER'])
|
||
|
|
r = eth_rpc.do(o)
|
||
|
|
gifter_nonce = int(r, 16)
|
||
|
|
|
||
|
|
#c = Gas(default_chain_spec, signer=eth_signer, nonce_oracle=nonce_oracle, gas_oracle=gas_oracle)
|
||
|
|
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, custodial_roles['FOO_TOKEN_GIFTER'], agent_roles['ALICE'], 100 * (10 ** 6), tx_format=TxFormat.RLP_SIGNED)
|
||
|
|
queue_create(
|
||
|
|
default_chain_spec,
|
||
|
|
gifter_nonce, # will only work if agent starts at 0
|
||
|
|
agent_roles['ALICE'],
|
||
|
|
tx_hash_hex,
|
||
|
|
tx_signed_raw_hex,
|
||
|
|
session=init_database,
|
||
|
|
)
|
||
|
|
cache_transfer_data(
|
||
|
|
tx_hash_hex,
|
||
|
|
tx_signed_raw_hex,
|
||
|
|
default_chain_spec.asdict(),
|
||
|
|
)
|
||
|
|
|
||
|
|
init_database.commit()
|
||
|
|
|
||
|
|
o = raw(tx_signed_raw_hex)
|
||
|
|
eth_rpc.do(o)
|
||
|
|
|
||
|
|
o = receipt(tx_hash_hex)
|
||
|
|
r = eth_rpc.do(o)
|
||
|
|
assert r['status'] == 1
|
||
|
|
|
||
|
|
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)
|
||
|
|
|
||
|
|
# lookup by transaction hash, without registry
|
||
|
|
api = AdminApi(eth_rpc, queue=None, call_address=contract_roles['CONTRACT_DEPLOYER'])
|
||
|
|
tx = api.tx(default_chain_spec, tx_hash=tx_hash_hex)
|
||
|
|
logg.debug('deployed {}'.format(contract_roles['CONTRACT_DEPLOYER']))
|
||
|
|
assert tx['tx_hash'] == tx_hash_hex
|
||
|
|
|
||
|
|
# lookup by RLP transaction, without registry
|
||
|
|
tx = api.tx(default_chain_spec, tx_raw=tx_signed_raw_hex)
|
||
|
|
assert tx['tx_hash'] == tx_hash_hex
|
||
|
|
|
||
|
|
# lookup by transaction hash, with registry
|
||
|
|
registry = CICRegistry(default_chain_spec, eth_rpc)
|
||
|
|
tx = api.tx(default_chain_spec, tx_hash=tx_hash_hex, registry=registry)
|
||
|
|
assert tx['tx_hash'] == tx_hash_hex
|
||
|
|
|
||
|
|
# lookup by transaction hash, using writer
|
||
|
|
buf = io.StringIO()
|
||
|
|
api.tx(default_chain_spec, tx_hash=tx_hash_hex, renderer=json.dumps, w=buf)
|
||
|
|
tx = json.loads(buf.getvalue())
|
||
|
|
assert tx['tx_hash'] == tx_hash_hex
|
||
|
|
|
||
|
|
|
||
|
|
def test_admin_api_account(
|
||
|
|
default_chain_spec,
|
||
|
|
init_database,
|
||
|
|
eth_rpc,
|
||
|
|
eth_signer,
|
||
|
|
agent_roles,
|
||
|
|
contract_roles,
|
||
|
|
celery_session_worker,
|
||
|
|
caplog,
|
||
|
|
):
|
||
|
|
|
||
|
|
nonce_oracle = OverrideNonceOracle(agent_roles['ALICE'], 42)
|
||
|
|
gas_oracle = OverrideGasOracle(limit=21000, conn=eth_rpc)
|
||
|
|
|
||
|
|
tx_hashes_alice = []
|
||
|
|
txs_alice = []
|
||
|
|
|
||
|
|
for i in range(3):
|
||
|
|
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+i,
|
||
|
|
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(),
|
||
|
|
)
|
||
|
|
tx_hashes_alice.append(tx_hash_hex)
|
||
|
|
txs_alice.append(tx_signed_raw_hex)
|
||
|
|
|
||
|
|
init_database.commit()
|
||
|
|
|
||
|
|
nonce_oracle = OverrideNonceOracle(agent_roles['BOB'], 13)
|
||
|
|
tx_hashes_bob = []
|
||
|
|
txs_bob = []
|
||
|
|
|
||
|
|
for i in range(2):
|
||
|
|
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['BOB'], agent_roles['ALICE'], 100 * (10 ** 6), tx_format=TxFormat.RLP_SIGNED)
|
||
|
|
queue_create(
|
||
|
|
default_chain_spec,
|
||
|
|
13+i,
|
||
|
|
agent_roles['BOB'],
|
||
|
|
tx_hash_hex,
|
||
|
|
tx_signed_raw_hex,
|
||
|
|
session=init_database,
|
||
|
|
)
|
||
|
|
cache_gas_data(
|
||
|
|
tx_hash_hex,
|
||
|
|
tx_signed_raw_hex,
|
||
|
|
default_chain_spec.asdict(),
|
||
|
|
)
|
||
|
|
tx_hashes_bob.append(tx_hash_hex)
|
||
|
|
txs_bob.append(tx_signed_raw_hex)
|
||
|
|
|
||
|
|
init_database.commit()
|
||
|
|
|
||
|
|
|
||
|
|
api = AdminApi(eth_rpc, queue=None, call_address=contract_roles['CONTRACT_DEPLOYER'])
|
||
|
|
r = api.account(default_chain_spec, agent_roles['ALICE'])
|
||
|
|
assert len(r) == 5
|
||
|
|
|
||
|
|
api = AdminApi(eth_rpc, queue=None, call_address=contract_roles['CONTRACT_DEPLOYER'])
|
||
|
|
r = api.account(default_chain_spec, agent_roles['ALICE'], include_sender=False)
|
||
|
|
assert len(r) == 2
|
||
|
|
|
||
|
|
api = AdminApi(eth_rpc, queue=None, call_address=contract_roles['CONTRACT_DEPLOYER'])
|
||
|
|
r = api.account(default_chain_spec, agent_roles['ALICE'], include_recipient=False)
|
||
|
|
assert len(r) == 3
|
||
|
|
|
||
|
|
|
||
|
|
def test_admin_api_account_writer(
|
||
|
|
default_chain_spec,
|
||
|
|
init_database,
|
||
|
|
eth_rpc,
|
||
|
|
eth_signer,
|
||
|
|
agent_roles,
|
||
|
|
contract_roles,
|
||
|
|
celery_session_worker,
|
||
|
|
caplog,
|
||
|
|
):
|
||
|
|
|
||
|
|
nonce_oracle = OverrideNonceOracle(agent_roles['ALICE'], 42)
|
||
|
|
gas_oracle = OverrideGasOracle(limit=21000, conn=eth_rpc)
|
||
|
|
|
||
|
|
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(),
|
||
|
|
)
|
||
|
|
|
||
|
|
init_database.commit()
|
||
|
|
|
||
|
|
buf = io.StringIO()
|
||
|
|
api = AdminApi(eth_rpc, queue=None, call_address=contract_roles['CONTRACT_DEPLOYER'])
|
||
|
|
api.account(default_chain_spec, agent_roles['ALICE'], renderer=json.dumps, w=buf)
|
||
|
|
|
||
|
|
# TODO: improve eval
|
||
|
|
tx = json.loads(buf.getvalue())
|
||
|
|
assert tx['tx_hash'] == tx_hash_hex
|
||
|
|
|
||
|
|
|
||
|
|
def test_registry(
|
||
|
|
eth_rpc,
|
||
|
|
cic_registry,
|
||
|
|
contract_roles,
|
||
|
|
celery_session_worker,
|
||
|
|
):
|
||
|
|
|
||
|
|
api = AdminApi(eth_rpc, queue=None, call_address=contract_roles['CONTRACT_DEPLOYER'])
|
||
|
|
t = api.registry()
|
||
|
|
r = t.get_leaf()
|
||
|
|
assert r == cic_registry
|
||
|
|
|
||
|
|
|
||
|
|
def test_proxy_do(
|
||
|
|
default_chain_spec,
|
||
|
|
eth_rpc,
|
||
|
|
contract_roles,
|
||
|
|
celery_session_worker,
|
||
|
|
):
|
||
|
|
|
||
|
|
o = block_latest()
|
||
|
|
r = eth_rpc.do(o)
|
||
|
|
|
||
|
|
api = AdminApi(eth_rpc, queue=None, call_address=contract_roles['CONTRACT_DEPLOYER'])
|
||
|
|
t = api.proxy_do(default_chain_spec, o)
|
||
|
|
rr = t.get_leaf()
|
||
|
|
|
||
|
|
assert r == rr
|
||
|
|
|
||
|
|
|
||
|
|
def test_resend_inplace(
|
||
|
|
init_database,
|
||
|
|
default_chain_spec,
|
||
|
|
eth_rpc,
|
||
|
|
eth_signer,
|
||
|
|
agent_roles,
|
||
|
|
contract_roles,
|
||
|
|
celery_session_worker,
|
||
|
|
):
|
||
|
|
|
||
|
|
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)
|
||
|
|
|
||
|
|
init_database.commit()
|
||
|
|
|
||
|
|
api = AdminApi(eth_rpc, queue=None, call_address=contract_roles['CONTRACT_DEPLOYER'])
|
||
|
|
t = api.resend(tx_hash_hex, default_chain_spec, unlock=True)
|
||
|
|
r = t.get_leaf()
|
||
|
|
assert t.successful()
|
||
|
|
|
||
|
|
|
||
|
|
otx = Otx.load(tx_hash_hex, session=init_database)
|
||
|
|
assert otx.status & StatusBits.OBSOLETE == StatusBits.OBSOLETE
|
||
|
|
|
||
|
|
txs = get_nonce_tx_cache(default_chain_spec, otx.nonce, agent_roles['ALICE'], session=init_database)
|
||
|
|
assert len(txs) == 2
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.xfail()
|
||
|
|
def test_resend_clone(
|
||
|
|
init_database,
|
||
|
|
default_chain_spec,
|
||
|
|
eth_rpc,
|
||
|
|
eth_signer,
|
||
|
|
agent_roles,
|
||
|
|
contract_roles,
|
||
|
|
celery_session_worker,
|
||
|
|
):
|
||
|
|
|
||
|
|
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)
|
||
|
|
|
||
|
|
init_database.commit()
|
||
|
|
|
||
|
|
api = AdminApi(eth_rpc, queue=None, call_address=contract_roles['CONTRACT_DEPLOYER'])
|
||
|
|
t = api.resend(tx_hash_hex, default_chain_spec, in_place=False)
|
||
|
|
r = t.get_leaf()
|
||
|
|
assert t.successful()
|
||
|
|
|
||
|
|
otx = Otx.load(tx_hash_hex, session=init_database)
|
||
|
|
assert otx.status & StatusBits.IN_NETWORK == StatusBits.IN_NETWORK
|
||
|
|
assert otx.status & StatusBits.OBSOLETE == StatusBits.OBSOLETE
|
||
|
|
|
||
|
|
txs = get_nonce_tx_cache(default_chain_spec, otx.nonce, agent_roles['ALICE'], session=init_database)
|
||
|
|
assert len(txs) == 1
|
||
|
|
|
||
|
|
txs = get_nonce_tx_cache(default_chain_spec, otx.nonce + 1, agent_roles['ALICE'], session=init_database)
|
||
|
|
assert len(txs) == 1
|
||
|
|
|
||
|
|
otx = Otx.load(txs[0], session=init_database)
|
||
|
|
assert otx.status == 0
|