Add dispatch tests

tests/chaind_eth_base.py

@@ -1,15 +1,23 @@
 # standard imports
+import os
 import unittest
 
 # external imports
 from chainsyncer.unittest.db import ChainSyncerDb
 from chainqueue.unittest.db import ChainQueueDb
 from chainlib.eth.unittest.ethtester import EthTesterCase
+from chainqueue.adapters.eth import EthAdapter
+from chainqueue.unittest.db import (
+        db_config,
+        dsn_from_config,
+        )
+from chainqueue.sql.backend import SQLBackend
+from chainlib.eth.address import to_checksum_address
+from hexathon import add_0x
 
 # local imports
 from chaind_eth.chain import EthChainInterface
 
-
 class TestBase(EthTesterCase):
 
     def setUp(self):
@@ -28,3 +36,15 @@ class TestBase(EthTesterCase):
         self.session_chainqueue.commit()
         self.db_chainqueue.release_session(self.session_chainqueue)
         super(TestBase, self).tearDown()
+
+
+class TestSQLBase(TestBase):
+
+    example_tx = bytes.fromhex('f8650d8405f5e10082520894ee38d3a40e177608d41978778206831f60dd0fa88204008077a040adee2ad0a0e566bced4b76a8899549e86719eb8866b87674b6fdc88479c201a030b3ca061bb330f4d78bc9cb8144c8e570339496f56b7809387de2ffeaa585d5')
+    example_tx_sender = add_0x(to_checksum_address('eb3907ecad74a0013c259d5874ae7f22dcbcc95c'))
+    dsn = dsn_from_config(db_config)
+
+    def setUp(self):
+        super(TestSQLBase, self).setUp()
+        self.backend = SQLBackend(self.dsn, debug=bool(os.environ.get('DATABASE_DEBUG')))
+        self.adapter = EthAdapter(self.backend)

tests/test_adapter.py

@@ -1,44 +1,25 @@
 # stanndard imports
 import logging
 import unittest
-import os
 
 # external imports
-from chainqueue.sql.backend import SQLBackend
 from chainlib.chain import ChainSpec
-from chainqueue.unittest.db import (
-        db_config,
-        dsn_from_config,
-        )
-
-# local imports
-from chainqueue.adapters.eth import EthAdapter
 
 # test imports
-from tests.chaind_eth_base import TestBase
+from tests.chaind_eth_base import TestSQLBase
 
 logging.basicConfig(level=logging.DEBUG)
 
 
-class TestAdapter(TestBase):
-
-    example_tx = bytes.fromhex('f8640183989680825208948311ad69b3429400ab795d45af85d204f73329ae8204d38026a097a7fd66548e4c116270b547ac7ed8cb531b0b97f80d49b45986144e47dbe44da07cc4345741dc0fabf65a473c0d3a1536cd501961f7e01b07dd8e107ff87d1556')
-    dsn = dsn_from_config(db_config)
-
-    def setUp(self):
-        super(TestAdapter, self).setUp()
-        self.chain_spec = ChainSpec.from_chain_str('foo:bar:1:baz')
-        self.backend = SQLBackend(self.dsn, debug=bool(os.environ.get('DATABASE_DEBUG')))
-        self.adapter = EthAdapter(self.backend)
-
+class TestAdapter(TestSQLBase):
 
     def test_eth_adapter_translate(self):
-        self.adapter.translate(self.chain_spec, self.example_tx)
+        self.adapter.translate(self.example_tx, self.chain_spec)
         # succesful decode means translate is working, no further checks needed
 
 
     def test_eth_adapter_add(self):
-        self.adapter.add(self.chain_spec, self.example_tx, session=self.session_chainqueue)
+        self.adapter.add(self.example_tx, self.chain_spec, session=self.session_chainqueue)
             
 
 if __name__ == '__main__':

tests/test_dispatch.py

@@ -0,0 +1,34 @@
+# stanndard imports
+import logging
+import unittest
+
+# external imports
+from chainlib.eth.tx import unpack
+from chainqueue.sql.query import get_tx
+from chainqueue.enum import StatusBits
+
+# local imports
+from chaind_eth.dispatch import Dispatcher
+
+# test imports
+from tests.chaind_eth_base import TestSQLBase
+
+logging.basicConfig(level=logging.DEBUG)
+
+
+
+class TestDispatcher(TestSQLBase):
+
+    def test_dispatch_process(self):
+        dispatcher = Dispatcher(self.chain_spec, self.adapter, 1)
+        self.adapter.add(self.example_tx, self.chain_spec, session=self.session_chainqueue)
+        assert dispatcher.get_count(self.example_tx_sender, self.session_chainqueue) == 1
+        dispatcher.process(self.rpc, self.session_chainqueue)
+        tx_obj = unpack(self.example_tx, self.chain_spec)
+        o = get_tx(self.chain_spec, tx_obj['hash'], session=self.session_chainqueue)
+        assert o['status'] & StatusBits.IN_NETWORK > 0
+        
+        assert dispatcher.get_count(self.example_tx_sender, self.session_chainqueue) == 0
+
+if __name__ == '__main__':
+    unittest.main()

tests/test_filter.py

@@ -76,7 +76,7 @@ class TestFilter(TestBase):
         adapter = EthAdapter(backend)
 
         tx_raw_rlp_signed_bytes = bytes.fromhex(strip_0x(tx_raw_rlp_signed))
-        adapter.add(self.chain_spec, tx_raw_rlp_signed_bytes, session=self.session_chainqueue)
+        adapter.add(tx_raw_rlp_signed_bytes, self.chain_spec, session=self.session_chainqueue)
  
         set_ready(self.chain_spec, tx_hash, session=self.session_chainqueue)
         set_reserved(self.chain_spec, tx_hash, session=self.session_chainqueue)