WIP implement sync done in resume sync for mock driver

chainsyncer/driver.py

@@ -110,7 +110,6 @@ class SyncDriver:
                 if self.block_callback != None:
                     self.block_callback(block, None)
 
-                last_block = block
                 try:
                     self.process(conn, item, block, tx_start)
                 except IndexError:
@@ -119,6 +118,9 @@ class SyncDriver:
                 time.sleep(self.yield_delay)
             if self.post_callback != None:
                 self.post_callback()
+    
+            if self.store.target > -1 and block.number >= self.store.target:
+                self.running = False
 
             self.idle(interval)
 

chainsyncer/unittest/base.py

@@ -195,7 +195,10 @@ class MockDriver(SyncDriver):
 
 
     def get(self, conn, item):
-        return self.blocks[item.cursor]
+        try:
+            return self.blocks[item.cursor]
+        except KeyError:
+            raise NoBlockForYou()
 
 
     def process(self, conn, item, block, tx_start):

tests/test_session.py

@@ -128,7 +128,9 @@ class TestFilter(unittest.TestCase):
 
 
         store = SyncFsStore(self.path, state_event_callback=state_event_handler, filter_state_event_callback=filter_state_event_handler)
-        drv = MockDriver(store)
+        fltr_one = MockFilter('foo')
+        store.register(fltr_one)
+        drv = MockDriver(store, target=1)
         generator.apply(drv, offset=1)
         drv.run(self.conn, interval=0.1)