cic-internal-integration/apps/cic-eth/tests/task/api/test_list.py

# standard imports
import logging

# local imports
from chainlib.eth.nonce import RPCNonceOracle
from chainlib.eth.erc20 import ERC20
from chainlib.eth.tx import receipt
from cic_eth.api.api_task import Api
from tests.mock.filter import (
        block_filter,
        tx_filter,
        )
from cic_eth.db.models.nonce import (
        Nonce,
        NonceReservation,
        )

logg = logging.getLogger()


def test_list_tx(
        default_chain_spec,
        init_database,
        cic_registry,
        eth_rpc,
        eth_signer,
        custodial_roles,
        agent_roles,
        foo_token,
        register_tokens,
        init_eth_tester,
        celery_session_worker,
        ):

    chain_id = default_chain_spec.chain_id()

    tx_hashes = []

    # external tx
    nonce_oracle = RPCNonceOracle(custodial_roles['FOO_TOKEN_GIFTER'], eth_rpc)
    nonce = nonce_oracle.get_nonce()
    
    q = init_database.query(Nonce)
    q = q.filter(Nonce.address_hex==agent_roles['ALICE'])
    o = q.first()
    o.nonce = nonce
    init_database.add(o)
    init_database.commit()

    # TODO: implement cachenonceoracle instead, this is useless
    # external tx one
    Nonce.next(custodial_roles['FOO_TOKEN_GIFTER'], 'foo', session=init_database)
    init_database.commit()

    init_eth_tester.mine_blocks(13)
    c = ERC20(signer=eth_signer, nonce_oracle=nonce_oracle, chain_id=chain_id)
    (tx_hash_hex, o) = c.transfer(foo_token, custodial_roles['FOO_TOKEN_GIFTER'], agent_roles['ALICE'], 1024)
    eth_rpc.do(o)
    o = receipt(tx_hash_hex)
    r = eth_rpc.do(o)
    assert r['status'] == 1
    a = r['block_number']
    block_filter.add(a.to_bytes(4, 'big'))

    a = r['block_number'] + r['transaction_index']
    tx_filter.add(a.to_bytes(4, 'big'))

    tx_hashes.append(tx_hash_hex)

    # external tx two
    Nonce.next(agent_roles['ALICE'], 'foo', session=init_database)
    init_database.commit()

    init_eth_tester.mine_blocks(13)
    nonce_oracle = RPCNonceOracle(agent_roles['ALICE'], eth_rpc)
    c = ERC20(signer=eth_signer, nonce_oracle=nonce_oracle, chain_id=chain_id)
    (tx_hash_hex, o) = c.transfer(foo_token, agent_roles['ALICE'], agent_roles['BOB'], 256)
    eth_rpc.do(o)
    o = receipt(tx_hash_hex)
    r = eth_rpc.do(o)
    assert r['status'] == 1
    a = r['block_number']
    block_filter.add(a.to_bytes(4, 'big'))

    a = r['block_number'] + r['transaction_index']
    tx_filter.add(a.to_bytes(4, 'big'))

    tx_hashes.append(tx_hash_hex)

    init_eth_tester.mine_blocks(28)

    # custodial tx 1
    api = Api(str(default_chain_spec), queue=None)
    t = api.transfer(agent_roles['ALICE'], agent_roles['CAROL'], 64, 'FOO') #, 'blinky')
    r = t.get_leaf()
    assert t.successful()
    tx_hashes.append(r)

    # custodial tx 2
    api = Api(str(default_chain_spec), queue=None)
    t = api.transfer(agent_roles['ALICE'], agent_roles['DAVE'], 16, 'FOO') #, 'blinky')
    r = t.get_leaf()
    assert t.successful()
    tx_hashes.append(r)

    logg.debug('r {}'.format(r))

    # test the api
    t = api.list(agent_roles['ALICE'], external_task='tests.mock.filter.filter')
    r = t.get_leaf()
    assert t.successful()


    assert len(r) == 3
    logg.debug('rrrr {}'.format(r))

    for tx in r:
        logg.debug('have tx {}'.format(tx))
        tx_hashes.remove(tx['hash'])
    assert len(tx_hashes) == 1