diff --git a/chaind/filter.py b/chaind/filter.py index b143b91..1e5386f 100644 --- a/chaind/filter.py +++ b/chaind/filter.py @@ -115,7 +115,7 @@ class StateFilter(SyncFilter): queue_adapter = self.__get_adapter(block, force_reload=True) continue - logg.info('filter registered {} for {} in {}'.format(tx.status_name, tx.hash, block)) + logg.info('filter registered {} for {} in {}'.format(tx.status.name, tx.hash, block)) if self.throttler != None: self.throttler.dec(tx.hash) diff --git a/chaind/unittest/common.py b/chaind/unittest/common.py index 949fd90..39e389d 100644 --- a/chaind/unittest/common.py +++ b/chaind/unittest/common.py @@ -7,11 +7,6 @@ from chainqueue.cache import CacheTokenTx from chainlib.status import Status as TxStatus from chainlib.chain import ChainSpec from chainlib.error import RPCException -from chainlib.tx import ( - Tx, - TxResult, - ) -from chainlib.block import Block class MockCacheAdapter(CacheTokenTx): @@ -39,17 +34,14 @@ class MockDispatcher: pass -class MockTx(Tx): +class MockTx: def __init__(self, tx_hash, status=TxStatus.SUCCESS): - result = TxResult() - result.status = status - super(MockTx, self).__init__(result=result) - self.set_hash(tx_hash) + self.hash = tx_hash + self.status = status -class MockBlock(Block): +class MockBlock: def __init__(self, number): - super(MockBlock, self).__init__() self.number = number