diff options
| author | Patrick Cloke <clokep@users.noreply.github.com> | 2022-02-08 11:03:08 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-02-08 11:03:08 -0500 |
| commit | d0e78af35e519ff76bd23e786007f3e7130d90f7 (patch) | |
| tree | 5755ad619669784608934fbd74d29cc92eaa4bae /tests | |
| parent | Experimental support to include bundled aggregations in search results (MSC36... (diff) | |
| download | synapse-d0e78af35e519ff76bd23e786007f3e7130d90f7.tar.xz | |
Add missing type hints to synapse.replication. (#11938)
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/replication/_base.py | 7 | ||||
| -rw-r--r-- | tests/replication/tcp/test_remote_server_up.py | 3 |
2 files changed, 7 insertions, 3 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) diff --git a/tests/replication/tcp/test_remote_server_up.py b/tests/replication/tcp/test_remote_server_up.py index 262c35cef3..545f11acd1 100644 --- a/tests/replication/tcp/test_remote_server_up.py +++ b/tests/replication/tcp/test_remote_server_up.py @@ -14,6 +14,7 @@ from typing import Tuple +from twisted.internet.address import IPv4Address from twisted.internet.interfaces import IProtocol from twisted.test.proto_helpers import StringTransport @@ -29,7 +30,7 @@ class RemoteServerUpTestCase(HomeserverTestCase): def _make_client(self) -> Tuple[IProtocol, StringTransport]: """Create a new direct TCP replication connection""" - proto = self.factory.buildProtocol(("127.0.0.1", 0)) + proto = self.factory.buildProtocol(IPv4Address("TCP", "127.0.0.1", 0)) transport = StringTransport() proto.makeConnection(transport) |