summary refs log tree commit diff
path: root/tests/replication/_base.py
diff options
context:
space:
mode:
authorErik Johnston <erik@matrix.org>2021-04-14 17:06:06 +0100
committerGitHub <noreply@github.com>2021-04-14 17:06:06 +0100
commit00a6db967655daf1d6db290b7e0d2bb53827ade9 (patch)
tree154989bf3a764542ba34147a6b27758dde7b1560 /tests/replication/_base.py
parentMore robust handling of the Content-Type header for thumbnail generation (#9788) (diff)
downloadsynapse-00a6db967655daf1d6db290b7e0d2bb53827ade9.tar.xz
Move some replication processing out of generic_worker (#9796)
Co-authored-by: Richard van der Hoff <1389908+richvdh@users.noreply.github.com>
Diffstat (limited to '')
-rw-r--r--tests/replication/_base.py8
1 files changed, 3 insertions, 5 deletions
diff --git a/tests/replication/_base.py b/tests/replication/_base.py
index 36138d69aa..c9d04aef29 100644
--- a/tests/replication/_base.py
+++ b/tests/replication/_base.py
@@ -21,13 +21,11 @@ from twisted.web.http import HTTPChannel
 from twisted.web.resource import Resource
 from twisted.web.server import Request, Site
 
-from synapse.app.generic_worker import (
-    GenericWorkerReplicationHandler,
-    GenericWorkerServer,
-)
+from synapse.app.generic_worker import GenericWorkerServer
 from synapse.http.server import JsonResource
 from synapse.http.site import SynapseRequest, SynapseSite
 from synapse.replication.http import ReplicationRestResource
+from synapse.replication.tcp.client import ReplicationDataHandler
 from synapse.replication.tcp.handler import ReplicationCommandHandler
 from synapse.replication.tcp.protocol import ClientReplicationStreamProtocol
 from synapse.replication.tcp.resource import (
@@ -431,7 +429,7 @@ class BaseMultiWorkerStreamTestCase(unittest.HomeserverTestCase):
             server_protocol.makeConnection(server_to_client_transport)
 
 
-class TestReplicationDataHandler(GenericWorkerReplicationHandler):
+class TestReplicationDataHandler(ReplicationDataHandler):
     """Drop-in for ReplicationDataHandler which just collects RDATA rows"""
 
     def __init__(self, hs: HomeServer):