From 575e1c4309d60ef2c2146f81029265731d2ced0b Mon Sep 17 00:00:00 2001 From: David Robertson Date: Sat, 11 Jun 2022 18:00:24 +0100 Subject: [PATCH] Annotate create_room --- tests/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/utils.py b/tests/utils.py index 0666458862..c93fc8990b 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -22,6 +22,7 @@ from synapse.api.room_versions import RoomVersions from synapse.config.homeserver import HomeServerConfig from synapse.config.server import DEFAULT_ROOM_VERSION from synapse.logging.context import current_context, set_current_context +from synapse.server import HomeServer from synapse.storage.database import LoggingDatabaseConnection from synapse.storage.engines import PostgresEngine, create_engine from synapse.storage.prepare_database import prepare_database @@ -270,10 +271,11 @@ class MockClock: self.advance_time(ms / 1000.0) -async def create_room(hs, room_id: str, creator_id: str): +async def create_room(hs: HomeServer, room_id: str, creator_id: str) -> None: """Creates and persist a creation event for the given room""" persistence_store = hs.get_storage_controllers().persistence + assert persistence_store is not None store = hs.get_datastores().main event_builder_factory = hs.get_event_builder_factory() event_creation_handler = hs.get_event_creation_handler()