chainqueue/tests/test_tx_cache.py

# 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.chainqueue_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()