chaintool/chaind-eth
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix bug in dispatch test failing on no gas balance in example tx

Browse Source
This commit is contained in:
nolash 2021-07-19 14:49:34 +02:00
parent 05e0b6ef19
commit 91b7195262
Signed by: lash
GPG Key ID: 21D2E7BB88C2A746
1 changed files with 22 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tests/test_dispatch.py
View File

@ -3,9 +3,20 @@ import logging
import unittest import unittest
# external imports # external imports
from chainlib.eth.tx import unpack from hexathon import strip_0x
from chainlib.eth.tx import (
unpack,
TxFormat,
)
from chainqueue.sql.query import get_tx from chainqueue.sql.query import get_tx
from chainqueue.enum import StatusBits from chainqueue.enum import StatusBits
from chainlib.eth.gas import (
RPCGasOracle,
Gas,
)
from chainlib.eth.nonce import (
RPCNonceOracle,
)
# local imports # local imports
from chaind_eth.dispatch import Dispatcher from chaind_eth.dispatch import Dispatcher
@ -16,19 +27,25 @@ from tests.chaind_eth_base import TestSQLBase
logging.basicConfig(level=logging.DEBUG) logging.basicConfig(level=logging.DEBUG)
class TestDispatcher(TestSQLBase): class TestDispatcher(TestSQLBase):
def test_dispatch_process(self): def test_dispatch_process(self):
gas_oracle = RPCGasOracle(conn=self.rpc)
nonce_oracle = RPCNonceOracle(self.accounts[0], conn=self.rpc)
c = Gas(self.chain_spec, signer=self.signer, gas_oracle=gas_oracle, nonce_oracle=nonce_oracle)
(tx_hash, tx_raw_rlp_signed) = c.create(self.accounts[0], self.accounts[1], 1024, tx_format=TxFormat.RLP_SIGNED)
tx_raw_rlp_signed_bytes = bytes.fromhex(strip_0x(tx_raw_rlp_signed))
dispatcher = Dispatcher(self.chain_spec, self.adapter, 1) dispatcher = Dispatcher(self.chain_spec, self.adapter, 1)
self.adapter.add(self.example_tx, self.chain_spec, session=self.session_chainqueue) self.adapter.add(tx_raw_rlp_signed_bytes, self.chain_spec, session=self.session_chainqueue)
assert dispatcher.get_count(self.example_tx_sender, self.session_chainqueue) == 1 assert dispatcher.get_count(self.example_tx_sender, self.session_chainqueue) == 1
dispatcher.process(self.rpc, self.session_chainqueue) dispatcher.process(self.rpc, self.session_chainqueue)
tx_obj = unpack(self.example_tx, self.chain_spec) tx_obj = unpack(tx_raw_rlp_signed_bytes, self.chain_spec)
o = get_tx(self.chain_spec, tx_obj['hash'], session=self.session_chainqueue) o = get_tx(self.chain_spec, tx_obj['hash'], session=self.session_chainqueue)
assert o['status'] & StatusBits.IN_NETWORK > 0 assert o['status'] & StatusBits.IN_NETWORK > 0
assert dispatcher.get_count(self.example_tx_sender, self.session_chainqueue) == 0 assert dispatcher.get_count(self.accounts[0], self.session_chainqueue) == 0
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()