chaind

commit dd1879bb91ae2d99aae94bd2c5d2e447ea91dd29
parent e264ed5c379f5e588dc29cc7b832d5a3825b1a43
Author: lash <dev@holbrook.no>
Date:   Mon,  9 May 2022 19:25:31 +0000

Implement mocks on generic tx, block

Diffstat:
M
chaind/filter.py
 | 
2
+-
M
chaind/unittest/common.py
 | 
16
++++++++++++----

2 files changed, 13 insertions(+), 5 deletions(-)
diff --git 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
@@ -7,6 +7,11 @@ 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):
@@ -34,14 +39,17 @@ class MockDispatcher:
         pass
 
 
-class MockTx:
+class MockTx(Tx):
 
     def __init__(self, tx_hash, status=TxStatus.SUCCESS):
-        self.hash = tx_hash
-        self.status = status
+        result = TxResult()
+        result.status = status
+        super(MockTx, self).__init__(result=result)
+        self.set_hash(tx_hash)
 
 
-class MockBlock:
+class MockBlock(Block):
 
     def __init__(self, number):
+        super(MockBlock, self).__init__()
         self.number = number