2021-04-15 09:59:45 +02:00
|
|
|
|
# standard imports
|
|
|
|
|
import logging
|
|
|
|
|
import unittest
|
|
|
|
|
import os
|
|
|
|
|
|
|
|
|
|
# external imports
|
|
|
|
|
from chainlib.chain import ChainSpec
|
|
|
|
|
|
|
|
|
|
# local imports
|
|
|
|
|
from chainsyncer.backend.memory import MemBackend
|
2021-04-15 15:06:07 +02:00
|
|
|
|
from chainsyncer.backend.sql import SyncerBackend
|
2021-04-15 09:59:45 +02:00
|
|
|
|
|
|
|
|
|
# test imports
|
|
|
|
|
from tests.base import TestBase
|
2021-04-15 11:15:48 +02:00
|
|
|
|
from chainsyncer.unittest.base import (
|
|
|
|
|
MockBlock,
|
2021-04-15 15:27:19 +02:00
|
|
|
|
MockConn,
|
2021-04-15 11:15:48 +02:00
|
|
|
|
TestSyncer,
|
|
|
|
|
)
|
2021-04-15 09:59:45 +02:00
|
|
|
|
|
|
|
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
|
|
|
logg = logging.getLogger()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class NaughtyCountExceptionFilter:
|
|
|
|
|
|
|
|
|
|
def __init__(self, name, croak_on):
|
|
|
|
|
self.c = 0
|
|
|
|
|
self.croak = croak_on
|
|
|
|
|
self.name = name
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def filter(self, conn, block, tx, db_session=None):
|
|
|
|
|
if self.c == self.croak:
|
2021-04-15 14:11:06 +02:00
|
|
|
|
self.croak = -1
|
2021-04-15 09:59:45 +02:00
|
|
|
|
raise RuntimeError('foo')
|
2021-04-15 14:11:06 +02:00
|
|
|
|
self.c += 1
|
2021-04-15 09:59:45 +02:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def __str__(self):
|
|
|
|
|
return '{} {}'.format(self.__class__.__name__, self.name)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class CountFilter:
|
|
|
|
|
|
|
|
|
|
def __init__(self, name):
|
|
|
|
|
self.c = 0
|
|
|
|
|
self.name = name
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def filter(self, conn, block, tx, db_session=None):
|
|
|
|
|
self.c += 1
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def __str__(self):
|
|
|
|
|
return '{} {}'.format(self.__class__.__name__, self.name)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2021-04-15 15:06:07 +02:00
|
|
|
|
class TestInterrupt(TestBase):
|
2021-04-15 09:59:45 +02:00
|
|
|
|
|
2021-04-15 15:06:07 +02:00
|
|
|
|
def setUp(self):
|
|
|
|
|
super(TestInterrupt, self).setUp()
|
|
|
|
|
self.filters = [
|
2021-04-15 09:59:45 +02:00
|
|
|
|
CountFilter('foo'),
|
|
|
|
|
CountFilter('bar'),
|
2021-04-15 15:27:19 +02:00
|
|
|
|
NaughtyCountExceptionFilter('xyzzy', croak_on=3),
|
2021-04-15 09:59:45 +02:00
|
|
|
|
CountFilter('baz'),
|
2021-04-15 15:06:07 +02:00
|
|
|
|
]
|
|
|
|
|
self.backend = None
|
2021-04-15 15:27:19 +02:00
|
|
|
|
self.conn = MockConn()
|
2021-04-15 15:06:07 +02:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def assert_filter_interrupt(self):
|
|
|
|
|
|
2021-04-15 15:27:19 +02:00
|
|
|
|
syncer = TestSyncer(self.backend, [4, 3, 2])
|
2021-04-15 09:59:45 +02:00
|
|
|
|
|
2021-04-15 15:06:07 +02:00
|
|
|
|
for fltr in self.filters:
|
|
|
|
|
syncer.add_filter(fltr)
|
2021-04-15 09:59:45 +02:00
|
|
|
|
|
2021-04-15 11:15:48 +02:00
|
|
|
|
try:
|
2021-04-15 15:27:19 +02:00
|
|
|
|
syncer.loop(0.1, self.conn)
|
2021-04-15 11:15:48 +02:00
|
|
|
|
except RuntimeError:
|
2021-04-15 14:11:06 +02:00
|
|
|
|
logg.info('caught croak')
|
2021-04-15 11:15:48 +02:00
|
|
|
|
pass
|
2021-04-15 15:06:07 +02:00
|
|
|
|
(pair, fltr) = self.backend.get()
|
|
|
|
|
self.assertGreater(fltr, 0)
|
2021-04-15 15:27:19 +02:00
|
|
|
|
syncer.loop(0.1, self.conn)
|
2021-04-15 09:59:45 +02:00
|
|
|
|
|
2021-04-15 15:06:07 +02:00
|
|
|
|
for fltr in self.filters:
|
2021-04-15 11:15:48 +02:00
|
|
|
|
logg.debug('{} {}'.format(str(fltr), fltr.c))
|
2021-04-15 15:06:07 +02:00
|
|
|
|
self.assertEqual(fltr.c, 9)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_filter_interrupt_memory(self):
|
|
|
|
|
self.backend = MemBackend(self.chain_spec, None, target_block=4)
|
|
|
|
|
self.assert_filter_interrupt()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_filter_interrupt_sql(self):
|
|
|
|
|
self.backend = SyncerBackend.initial(self.chain_spec, 4)
|
|
|
|
|
self.assert_filter_interrupt()
|
2021-04-15 11:15:48 +02:00
|
|
|
|
|
|
|
|
|
|
2021-04-15 09:59:45 +02:00
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
unittest.main()
|