# standard imports
import unittest

# local imports
from chainqueue.db.models.tx import TxCache
from chainqueue.error import NotLocalTxError
from chainqueue.state import *
from chainqueue.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)
        set_reserved(self.chain_spec, self.tx_hash)
        set_sent(self.chain_spec, self.tx_hash)
        set_final(self.chain_spec, self.tx_hash, block=1024)

        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'], self.tx_hash)


if __name__ == '__main__':
    unittest.main()