Update scanner test

stakewise · Nov 18, 2024 · ffe7d0e · ffe7d0e
1 parent 98b79ef
commit ffe7d0e
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 10 deletions.
diff --git a/sw_utils/event_scanner.py b/sw_utils/event_scanner.py
@@ -105,7 +105,6 @@ async def _scan_chunk(
                 if i < retries - 1:
                     # Decrease the `eth_getBlocks` range
                     self.chunk_size = self.chunk_size // 2
-                    to_block = BlockNumber(from_block + self.chunk_size)
                     # Let the JSON-RPC to recover e.g. from restart
                     await sleep(self.request_retry_seconds)
                     continue

diff --git a/sw_utils/tests/test_event_scanner.py b/sw_utils/tests/test_event_scanner.py
@@ -27,15 +27,23 @@ class TestExitSignatureCrud:
     async def test_basic(self):
         default_chunk_size = 500000
         p = MockedEventProcessor()
-        s = EventScanner(processor=p)
-        s._contract_call = fetch_events
-        assert s.chunk_size == default_chunk_size
+        scanner = EventScanner(processor=p)
+        scanner._contract_call = fetch_events
+        assert scanner.chunk_size == default_chunk_size
 
-        s._contract_call = fetch_events_broken
-        s.max_request_retries = 1
+        scanner._contract_call = fetch_events_broken
+        scanner.request_retry_seconds = 0
+        scanner.max_request_retries = 1
         with pytest.raises(ConnectionError):
-            await s.process_new_events(888)
-            assert s.chunk_size == default_chunk_size / 2
+            await scanner.process_new_events(888)
+        assert scanner.chunk_size == default_chunk_size
 
-            await s.process_new_events(888)
-            assert s.chunk_size == default_chunk_size / 4
+        scanner.max_request_retries = 2
+        with pytest.raises(ConnectionError):
+            await scanner.process_new_events(888)
+        assert scanner.chunk_size == default_chunk_size // 2
+
+        scanner.max_request_retries = 3
+        with pytest.raises(ConnectionError):
+            await scanner.process_new_events(888)
+        assert scanner.chunk_size == default_chunk_size // 8