Fixup tests

pull/7185/head
Erik Johnston 2020-04-01 17:31:29 +01:00
parent cf57d56e39
commit dc91879ee1
1 changed files with 4 additions and 2 deletions

View File

@ -27,7 +27,7 @@ class BaseStreamTestCase(unittest.HomeserverTestCase):
"""Base class for tests of the replication streams""" """Base class for tests of the replication streams"""
def make_homeserver(self, reactor, clock): def make_homeserver(self, reactor, clock):
self.test_handler = Mock(wraps=TestReplicationClientHandler()) self.test_handler = Mock(wraps=TestReplicationDataHandler())
return self.setup_test_homeserver(replication_data_handler=self.test_handler) return self.setup_test_homeserver(replication_data_handler=self.test_handler)
def prepare(self, reactor, clock, hs): def prepare(self, reactor, clock, hs):
@ -75,7 +75,9 @@ class BaseStreamTestCase(unittest.HomeserverTestCase):
self.pump(0.1) self.pump(0.1)
class TestReplicationClientHandler: class TestReplicationDataHandler:
"""Drop-in for ReplicationDataHandler which just collects RDATA rows"""
def __init__(self): def __init__(self):
self.streams = set() self.streams = set()
self._received_rdata_rows = [] self._received_rdata_rows = []