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/federation | |
| 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/federation')
| -rw-r--r-- | tests/federation/test_federation_server.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/federation/test_federation_server.py b/tests/federation/test_federation_server.py index b19365b81a..413b3c9426 100644 --- a/tests/federation/test_federation_server.py +++ b/tests/federation/test_federation_server.py @@ -134,6 +134,8 @@ class SendJoinFederationTests(unittest.FederatingHomeserverTestCase): def prepare(self, reactor: MemoryReactor, clock: Clock, hs: HomeServer): super().prepare(reactor, clock, hs) + self._storage_controllers = hs.get_storage_controllers() + # create the room creator_user_id = self.register_user("kermit", "test") tok = self.login("kermit", "test") @@ -207,7 +209,7 @@ class SendJoinFederationTests(unittest.FederatingHomeserverTestCase): # the room should show that the new user is a member r = self.get_success( - self.hs.get_state_handler().get_current_state(self._room_id) + self._storage_controllers.state.get_current_state(self._room_id) ) self.assertEqual(r[("m.room.member", joining_user)].membership, "join") @@ -258,7 +260,7 @@ class SendJoinFederationTests(unittest.FederatingHomeserverTestCase): # the room should show that the new user is a member r = self.get_success( - self.hs.get_state_handler().get_current_state(self._room_id) + self._storage_controllers.state.get_current_state(self._room_id) ) self.assertEqual(r[("m.room.member", joining_user)].membership, "join") |