chainqueue/tests/test_shep.py
2022-03-11 11:03:05 +00:00

52 lines
1.2 KiB
Python

# standard imports
import os
import logging
import unittest
import tempfile
# external imports
from hexathon import add_0x
from shep.store.file import SimpleFileStoreFactory
from shep.error import StateTransitionInvalid
# local imports
from chainqueue import Status
from chainqueue.tx import Tx
logging.basicConfig(level=logging.DEBUG)
logg = logging.getLogger()
class TestShepBase(unittest.TestCase):
def setUp(self):
self.path = tempfile.mkdtemp()
factory = SimpleFileStoreFactory(self.path).add
self.state = Status(factory)
class TestShep(TestShepBase):
def test_shep_setup(self):
pass
def test_shep_tx(self):
tx_hash = add_0x(os.urandom(20).hex())
signed_tx = add_0x(os.urandom(128).hex())
nonce = 42
tx = Tx(self.state, nonce, tx_hash, signed_tx)
tx.create()
logg.debug('file {}'.format(self.path))
def test_shep_invalid(self):
self.state.put('foo', 'bar')
self.state.set('foo', self.state.FINAL)
with self.assertRaises(StateTransitionInvalid):
self.state.move('foo', self.state.GAS_ISSUES)
if __name__ == '__main__':
unittest.main()