summary refs log tree commit diff
path: root/tests/federation
diff options
context:
space:
mode:
authorErik Johnston <erik@matrix.org>2022-05-20 12:33:21 +0100
committerErik Johnston <erik@matrix.org>2022-05-20 12:51:53 +0100
commit11efe7231f4c5320493460ec41f4e9773c6ca120 (patch)
treeaf97ead1f5193cb20b727db17f0e7241f91d3afd /tests/federation
parentAdd helper methods to store (diff)
downloadsynapse-11efe7231f4c5320493460ec41f4e9773c6ca120.tar.xz
Use new helper functions
Diffstat (limited to 'tests/federation')
-rw-r--r--tests/federation/test_federation_server.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/federation/test_federation_server.py b/tests/federation/test_federation_server.py
index b19365b81a..0f9fb0abf1 100644
--- a/tests/federation/test_federation_server.py
+++ b/tests/federation/test_federation_server.py
@@ -207,7 +207,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.hs.get_datastores().main.get_current_state(self._room_id)
         )
         self.assertEqual(r[("m.room.member", joining_user)].membership, "join")
 
@@ -258,7 +258,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.hs.get_datastores().main.get_current_state(self._room_id)
         )
         self.assertEqual(r[("m.room.member", joining_user)].membership, "join")