Add missing type hints to new tests

2 files changed, 9 insertions, 2 deletions

diff --git a/tests/handlers/test_space_hierarchy.py b/tests/handlers/test_space_hierarchy.py
index 8e46b2c93e..548173d1db 100644
--- a/tests/handlers/test_space_hierarchy.py
+++ b/tests/handlers/test_space_hierarchy.py
@@ -14,11 +14,14 @@
 
 from typing import Dict, Optional
 
+from twisted.test.proto_helpers import MemoryReactor
+
 from synapse.api.constants import EventContentFields, EventTypes, RoomTypes
 from synapse.rest import admin
 from synapse.rest.client import login, room
 from synapse.server import HomeServer
 from synapse.types import JsonDict
+from synapse.util import Clock
 
 from tests import unittest
 
@@ -32,7 +35,7 @@ class SpaceDescendantsTestCase(unittest.HomeserverTestCase):
         room.register_servlets,
     ]
 
-    def prepare(self, reactor, clock, hs: HomeServer):
+    def prepare(self, reactor: MemoryReactor, clock: Clock, hs: HomeServer):
         self.hs = hs
         self.handler = self.hs.get_space_hierarchy_handler()
 
diff --git a/tests/rest/admin/test_space.py b/tests/rest/admin/test_space.py
index 59c6c3b284..46267b0b89 100644
--- a/tests/rest/admin/test_space.py
+++ b/tests/rest/admin/test_space.py
@@ -16,6 +16,8 @@ from typing import Dict, Optional, Tuple, Union
 
 from typing_extensions import Literal
 
+from twisted.test.proto_helpers import MemoryReactor
+
 import synapse.rest.admin
 from synapse.api.constants import (
     EventContentFields,
@@ -27,7 +29,9 @@ from synapse.api.constants import (
 )
 from synapse.api.room_versions import RoomVersions
 from synapse.rest.client import login, room
+from synapse.server import HomeServer
 from synapse.types import JsonDict
+from synapse.util import Clock
 
 from tests import unittest
 
@@ -41,7 +45,7 @@ class RemoveSpaceMemberTestCase(unittest.HomeserverTestCase):
         room.register_servlets,
     ]
 
-    def prepare(self, reactor, clock, hs):
+    def prepare(self, reactor: MemoryReactor, clock: Clock, hs: HomeServer):
         self.store = hs.get_datastore()
 
         # Create users