1 files changed, 5 insertions, 2 deletions
diff --git a/tests/replication/_base.py b/tests/replication/_base.py
index cb02eddf07..9fc50f8852 100644
--- a/tests/replication/_base.py
+++ b/tests/replication/_base.py
@@ -14,6 +14,7 @@
import logging
from typing import Any, Dict, List, Optional, Tuple
+from twisted.internet.address import IPv4Address
from twisted.internet.protocol import Protocol
from twisted.web.resource import Resource
@@ -53,7 +54,7 @@ class BaseStreamTestCase(unittest.HomeserverTestCase):
server_factory = ReplicationStreamProtocolFactory(hs)
self.streamer = hs.get_replication_streamer()
self.server: ServerReplicationStreamProtocol = server_factory.buildProtocol(
- None
+ IPv4Address("TCP", "127.0.0.1", 0)
)
# Make a new HomeServer object for the worker
@@ -345,7 +346,9 @@ class BaseMultiWorkerStreamTestCase(unittest.HomeserverTestCase):
self.clock,
repl_handler,
)
- server = self.server_factory.buildProtocol(None)
+ server = self.server_factory.buildProtocol(
+ IPv4Address("TCP", "127.0.0.1", 0)
+ )
client_transport = FakeTransport(server, self.reactor)
client.makeConnection(client_transport)
|