Use mock conn and process method from headsyncer in testsyncer

This commit is contained in:
nolash
2021-04-15 15:27:19 +02:00
parent 987a18fd6b
commit 6201420ad2
3 changed files with 22 additions and 17 deletions

View File

@@ -14,6 +14,7 @@ from chainsyncer.backend.sql import SyncerBackend
from tests.base import TestBase
from chainsyncer.unittest.base import (
MockBlock,
MockConn,
TestSyncer,
)
@@ -63,27 +64,28 @@ class TestInterrupt(TestBase):
self.filters = [
CountFilter('foo'),
CountFilter('bar'),
NaughtyCountExceptionFilter('xyzzy', 3),
NaughtyCountExceptionFilter('xyzzy', croak_on=3),
CountFilter('baz'),
]
self.backend = None
self.conn = MockConn()
def assert_filter_interrupt(self):
syncer = TestSyncer(self.backend, [4, 2, 3])
syncer = TestSyncer(self.backend, [4, 3, 2])
for fltr in self.filters:
syncer.add_filter(fltr)
try:
syncer.loop(0.1, None)
syncer.loop(0.1, self.conn)
except RuntimeError:
logg.info('caught croak')
pass
(pair, fltr) = self.backend.get()
self.assertGreater(fltr, 0)
syncer.loop(0.1, None)
syncer.loop(0.1, self.conn)
for fltr in self.filters:
logg.debug('{} {}'.format(str(fltr), fltr.c))