summary refs log tree commit diff
path: root/tests/replication/test_federation_ack.py
diff options
context:
space:
mode:
authorPatrick Cloke <clokep@users.noreply.github.com>2023-02-06 09:55:00 -0500
committerGitHub <noreply@github.com>2023-02-06 09:55:00 -0500
commit156cd88eefe7db100e5cdba48174c709975b93ca (patch)
treebf4059f81c6ba16439ef6dfa19a4e016057da20d /tests/replication/test_federation_ack.py
parentExpect type stubs from canonicaljson (#14992) (diff)
downloadsynapse-156cd88eefe7db100e5cdba48174c709975b93ca.tar.xz
Add missing type hints to tests.replication. (#14987)
Diffstat (limited to 'tests/replication/test_federation_ack.py')
-rw-r--r--tests/replication/test_federation_ack.py12
1 files changed, 7 insertions, 5 deletions
diff --git a/tests/replication/test_federation_ack.py b/tests/replication/test_federation_ack.py
index 63b1dd40b5..12668b34c5 100644
--- a/tests/replication/test_federation_ack.py
+++ b/tests/replication/test_federation_ack.py
@@ -14,10 +14,14 @@
 
 from unittest import mock
 
+from twisted.test.proto_helpers import MemoryReactor
+
 from synapse.app.generic_worker import GenericWorkerServer
 from synapse.replication.tcp.commands import FederationAckCommand
 from synapse.replication.tcp.protocol import IReplicationConnection
 from synapse.replication.tcp.streams.federation import FederationStream
+from synapse.server import HomeServer
+from synapse.util import Clock
 
 from tests.unittest import HomeserverTestCase
 
@@ -30,12 +34,10 @@ class FederationAckTestCase(HomeserverTestCase):
         config["federation_sender_instances"] = ["federation_sender1"]
         return config
 
-    def make_homeserver(self, reactor, clock):
-        hs = self.setup_test_homeserver(homeserver_to_use=GenericWorkerServer)
-
-        return hs
+    def make_homeserver(self, reactor: MemoryReactor, clock: Clock) -> HomeServer:
+        return self.setup_test_homeserver(homeserver_to_use=GenericWorkerServer)
 
-    def test_federation_ack_sent(self):
+    def test_federation_ack_sent(self) -> None:
         """A FEDERATION_ACK should be sent back after each RDATA federation
 
         This test checks that the federation sender is correctly sending back