1616
1717from unittest .mock import Mock
1818
19+ from twisted .test .proto_helpers import MemoryReactor
20+
1921import synapse .rest .admin
2022from synapse .rest .client import events , login , room
23+ from synapse .server import HomeServer
24+ from synapse .util import Clock
2125
2226from tests import unittest
2327
@@ -32,7 +36,7 @@ class EventStreamPermissionsTestCase(unittest.HomeserverTestCase):
3236 login .register_servlets ,
3337 ]
3438
35- def make_homeserver (self , reactor , clock ) :
39+ def make_homeserver (self , reactor : MemoryReactor , clock : Clock ) -> HomeServer :
3640
3741 config = self .default_config ()
3842 config ["enable_registration_captcha" ] = False
@@ -41,11 +45,11 @@ def make_homeserver(self, reactor, clock):
4145
4246 hs = self .setup_test_homeserver (config = config )
4347
44- hs .get_federation_handler = Mock ()
48+ hs .get_federation_handler = Mock () # type: ignore[assignment]
4549
4650 return hs
4751
48- def prepare (self , reactor , clock , hs ) :
52+ def prepare (self , reactor : MemoryReactor , clock : Clock , hs : HomeServer ) -> None :
4953
5054 # register an account
5155 self .user_id = self .register_user ("sid1" , "pass" )
@@ -55,7 +59,7 @@ def prepare(self, reactor, clock, hs):
5559 self .other_user = self .register_user ("other2" , "pass" )
5660 self .other_token = self .login (self .other_user , "pass" )
5761
58- def test_stream_basic_permissions (self ):
62+ def test_stream_basic_permissions (self ) -> None :
5963 # invalid token, expect 401
6064 # note: this is in violation of the original v1 spec, which expected
6165 # 403. However, since the v1 spec no longer exists and the v1
@@ -76,7 +80,7 @@ def test_stream_basic_permissions(self):
7680 self .assertTrue ("start" in channel .json_body )
7781 self .assertTrue ("end" in channel .json_body )
7882
79- def test_stream_room_permissions (self ):
83+ def test_stream_room_permissions (self ) -> None :
8084 room_id = self .helper .create_room_as (self .other_user , tok = self .other_token )
8185 self .helper .send (room_id , tok = self .other_token )
8286
@@ -111,7 +115,7 @@ def test_stream_room_permissions(self):
111115
112116 # left to room (expect no content for room)
113117
114- def TODO_test_stream_items (self ):
118+ def TODO_test_stream_items (self ) -> None :
115119 # new user, no content
116120
117121 # join room, expect 1 item (join)
@@ -136,15 +140,15 @@ class GetEventsTestCase(unittest.HomeserverTestCase):
136140 login .register_servlets ,
137141 ]
138142
139- def prepare (self , hs , reactor , clock ) :
143+ def prepare (self , reactor : MemoryReactor , clock : Clock , hs : HomeServer ) -> None :
140144
141145 # register an account
142146 self .user_id = self .register_user ("sid1" , "pass" )
143147 self .token = self .login (self .user_id , "pass" )
144148
145149 self .room_id = self .helper .create_room_as (self .user_id , tok = self .token )
146150
147- def test_get_event_via_events (self ):
151+ def test_get_event_via_events (self ) -> None :
148152 resp = self .helper .send (self .room_id , tok = self .token )
149153 event_id = resp ["event_id" ]
150154
0 commit comments