mirror of
https://git.anonymousland.org/anonymousland/synapse.git
synced 2025-07-26 20:45:18 -04:00
Add final type hint to tests.unittest. (#15072)
Adds a return type to HomeServerTestCase.make_homeserver and deal with any variables which are no longer Any.
This commit is contained in:
parent
119e0795a5
commit
42aea0d8af
55 changed files with 433 additions and 320 deletions
|
@ -156,11 +156,11 @@ class PresenceRouterTestCase(FederatingHomeserverTestCase):
|
|||
|
||||
def make_homeserver(self, reactor: MemoryReactor, clock: Clock) -> HomeServer:
|
||||
# Mock out the calls over federation.
|
||||
fed_transport_client = Mock(spec=["send_transaction"])
|
||||
fed_transport_client.send_transaction = simple_async_mock({})
|
||||
self.fed_transport_client = Mock(spec=["send_transaction"])
|
||||
self.fed_transport_client.send_transaction = simple_async_mock({})
|
||||
|
||||
hs = self.setup_test_homeserver(
|
||||
federation_transport_client=fed_transport_client,
|
||||
federation_transport_client=self.fed_transport_client,
|
||||
)
|
||||
|
||||
load_legacy_presence_router(hs)
|
||||
|
@ -422,7 +422,7 @@ class PresenceRouterTestCase(FederatingHomeserverTestCase):
|
|||
#
|
||||
# Thus we reset the mock, and try sending all online local user
|
||||
# presence again
|
||||
self.hs.get_federation_transport_client().send_transaction.reset_mock()
|
||||
self.fed_transport_client.send_transaction.reset_mock()
|
||||
|
||||
# Broadcast local user online presence
|
||||
self.get_success(
|
||||
|
@ -447,9 +447,7 @@ class PresenceRouterTestCase(FederatingHomeserverTestCase):
|
|||
}
|
||||
found_users = set()
|
||||
|
||||
calls = (
|
||||
self.hs.get_federation_transport_client().send_transaction.call_args_list
|
||||
)
|
||||
calls = self.fed_transport_client.send_transaction.call_args_list
|
||||
for call in calls:
|
||||
call_args = call[0]
|
||||
federation_transaction: Transaction = call_args[0]
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue