Introduce driver object

tests/test_filter.py

@@ -0,0 +1,75 @@
+# standard imports
+import unittest
+import tempfile
+import shutil
+import logging
+import stat
+import os
+
+# local imports
+from chainsyncer.store.fs import SyncFsStore
+from chainsyncer.session import SyncSession
+from chainsyncer.error import (
+        LockError,
+        FilterDone,
+        IncompleteFilterError,
+        )
+from chainsyncer.unittest import (
+        MockFilter,
+        MockConn,
+        MockTx,
+        MockBlock,
+        )
+
+logging.basicConfig(level=logging.DEBUG)
+logg = logging.getLogger()
+
+
+class TestFilter(unittest.TestCase):
+
+    def setUp(self):
+        self.path = tempfile.mkdtemp()
+        self.store = SyncFsStore(self.path)
+        self.session = SyncSession(self.store)
+        self.session.start()
+        self.conn = MockConn()
+
+
+    def tearDown(self):
+        shutil.rmtree(self.path)
+
+
+    def test_filter_basic(self):
+        fltr_one = MockFilter('foo')
+        self.session.register(fltr_one)
+        fltr_two = MockFilter('bar')
+        self.session.register(fltr_two)
+
+        tx_hash = os.urandom(32).hex()
+        tx = MockTx(42, tx_hash)
+        block = MockBlock(13, [tx_hash])
+        self.session.filter(self.conn, block, tx)
+        
+        self.assertEqual(len(fltr_one.contents), 1)
+        self.assertEqual(len(fltr_two.contents), 1)
+        
+
+
+    def test_filter_interrupt(self):
+        fltr_one = MockFilter('foo', brk=True)
+        self.session.register(fltr_one)
+        fltr_two = MockFilter('bar')
+        self.session.register(fltr_two)
+
+        tx_hash = os.urandom(32).hex()
+        tx = MockTx(42, tx_hash)
+        block = MockBlock(13, [tx_hash])
+        self.session.filter(self.conn, block, tx)
+        
+        self.assertEqual(len(fltr_one.contents), 1)
+        self.assertEqual(len(fltr_two.contents), 0)
+
+
+if __name__ == '__main__':
+    unittest.main()
+

tests/test_fs.py

@@ -13,6 +13,7 @@ from chainsyncer.error import (
         LockError,
         FilterDone,
         IncompleteFilterError,
+        SyncDone,
         )
 from chainsyncer.unittest import MockFilter
 
@@ -25,6 +26,7 @@ class TestFs(unittest.TestCase):
     def setUp(self):
         self.path = tempfile.mkdtemp()
 
+
     def tearDown(self):
         shutil.rmtree(self.path)
 
@@ -167,6 +169,23 @@ class TestFs(unittest.TestCase):
 
         o.reset()
 
+    
+    def test_sync_process_done(self):
+        store = SyncFsStore(self.path)
+        session = SyncSession(store)
+
+        fltr_one = MockFilter('foo')
+        session.register(fltr_one)
+
+        session.start(target=0)
+        o = session.get(0)
+        o.advance()
+        o.release()
+        with self.assertRaises(FilterDone):
+            o.advance()
+        with self.assertRaises(SyncDone):
+            o.reset()
+
 
 if __name__ == '__main__':
     unittest.main()

tests/test_session.py

@@ -0,0 +1,68 @@
+# standard imports
+import unittest
+import tempfile
+import shutil
+import logging
+import stat
+import os
+
+# local imports
+from chainsyncer.store.fs import SyncFsStore
+from chainsyncer.session import SyncSession
+from chainsyncer.error import (
+        LockError,
+        FilterDone,
+        IncompleteFilterError,
+        SyncDone,
+        )
+from chainsyncer.unittest import (
+        MockFilter,
+        MockConn,
+        MockTx,
+        MockBlock,
+        )
+from chainsyncer.driver import SyncDriver
+
+logging.basicConfig(level=logging.DEBUG)
+logg = logging.getLogger()
+
+
+class TestFilter(unittest.TestCase):
+
+    def setUp(self):
+        self.path = tempfile.mkdtemp()
+        self.store = SyncFsStore(self.path)
+        self.conn = MockConn()
+
+
+    def tearDown(self):
+        shutil.rmtree(self.path)
+
+
+    def test_filter_basic(self):
+        session = SyncSession(self.store)
+        session.start(target=1)
+        fltr_one = MockFilter('foo')
+        session.register(fltr_one)
+
+        tx_hash = os.urandom(32).hex()
+        tx = MockTx(42, tx_hash)
+        block = MockBlock(0, [tx_hash])
+        session.filter(self.conn, block, tx)
+ 
+        tx_hash = os.urandom(32).hex()
+        tx = MockTx(42, tx_hash)
+        block = MockBlock(1, [tx_hash])
+        with self.assertRaises(SyncDone):
+            session.filter(self.conn, block, tx)
+        self.assertEqual(len(fltr_one.contents), 2)
+
+
+
+    def test_driver(self):
+        drv = SyncDriver(self.conn, self.store)
+        drv.run()
+
+
+if __name__ == '__main__':
+    unittest.main()