45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
# standard imports
|
|
import unittest
|
|
|
|
# external imports
|
|
from hexathon import add_0x
|
|
|
|
# local imports
|
|
from chainqueue.db.models.tx import TxCache
|
|
from chainqueue.error import NotLocalTxError
|
|
from chainqueue.sql.state import *
|
|
from chainqueue.sql.query import get_tx_cache
|
|
|
|
# test imports
|
|
from tests.base import TestTxBase
|
|
|
|
class TestTxCache(TestTxBase):
|
|
|
|
def test_mine(self):
|
|
with self.assertRaises(NotLocalTxError):
|
|
TxCache.set_final(self.tx_hash, 1024, 13, session=self.session)
|
|
|
|
set_ready(self.chain_spec, self.tx_hash, session=self.session)
|
|
set_reserved(self.chain_spec, self.tx_hash, session=self.session)
|
|
set_sent(self.chain_spec, self.tx_hash, session=self.session)
|
|
set_final(self.chain_spec, self.tx_hash, block=1024, session=self.session)
|
|
|
|
with self.assertRaises(NotLocalTxError):
|
|
TxCache.set_final(self.tx_hash, 1023, 13, session=self.session)
|
|
|
|
TxCache.set_final(self.tx_hash, 1024, 13, session=self.session)
|
|
|
|
self.session.commit()
|
|
|
|
txc = TxCache.load(self.tx_hash)
|
|
self.assertEqual(txc.tx_index, 13)
|
|
|
|
|
|
def test_get(self):
|
|
tx_extended_dict = get_tx_cache(self.chain_spec, self.tx_hash)
|
|
self.assertEqual(tx_extended_dict['tx_hash'], add_0x(self.tx_hash))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|