76 lines
1.7 KiB
Python
76 lines
1.7 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 MockIndexStore:
|
|
|
|
def __init__(self):
|
|
self.store = {}
|
|
|
|
|
|
def put(self, k, v):
|
|
self.store[k] = v
|
|
|
|
|
|
def get(self, k):
|
|
return self.store.get(k)
|
|
|
|
|
|
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
|
|
mock_store = MockIndexStore()
|
|
tx = Tx(self.state, mock_store, tx_hash)
|
|
tx.create(nonce, signed_tx)
|
|
|
|
tx_retrieved = Tx(self.state, mock_store, tx_hash)
|
|
tx_retrieved.load()
|
|
self.assertEqual(tx_retrieved.signed_tx, signed_tx)
|
|
|
|
|
|
def test_shep_valid(self):
|
|
self.state.put('foo', 'bar')
|
|
self.state.set('foo', self.state.IN_NETWORK)
|
|
self.state.set('foo', self.state.FINAL)
|
|
|
|
|
|
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.INSUFFICIENT_FUNDS)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|