diff options
| author | Erik Johnston <erik@matrix.org> | 2022-06-06 11:24:12 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-06-06 09:24:12 +0100 |
| commit | e3163e2e11cf8bffa4cb3e58ac0b86a83eca314c (patch) | |
| tree | 639271e4b4d4157b95047f801f6b22bf39a24f08 /tests/handlers | |
| parent | Remove groups code from synapse_port_db. (#12899) (diff) | |
| download | synapse-e3163e2e11cf8bffa4cb3e58ac0b86a83eca314c.tar.xz | |
Reduce the amount of state we pull from the DB (#12811)
Diffstat (limited to 'tests/handlers')
| -rw-r--r-- | tests/handlers/test_directory.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/handlers/test_directory.py b/tests/handlers/test_directory.py index 11ad44223d..53d49ca896 100644 --- a/tests/handlers/test_directory.py +++ b/tests/handlers/test_directory.py @@ -298,6 +298,7 @@ class CanonicalAliasTestCase(unittest.HomeserverTestCase): self.store = hs.get_datastores().main self.handler = hs.get_directory_handler() self.state_handler = hs.get_state_handler() + self._storage_controllers = hs.get_storage_controllers() # Create user self.admin_user = self.register_user("admin", "pass", admin=True) @@ -335,7 +336,7 @@ class CanonicalAliasTestCase(unittest.HomeserverTestCase): def _get_canonical_alias(self): """Get the canonical alias state of the room.""" return self.get_success( - self.state_handler.get_current_state( + self._storage_controllers.state.get_current_state_event( self.room_id, EventTypes.CanonicalAlias, "" ) ) |