diff options
Diffstat (limited to 'tests/storage/test_purge.py')
| -rw-r--r-- | tests/storage/test_purge.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/storage/test_purge.py b/tests/storage/test_purge.py index 92cd0dfc05..8dfaa0559b 100644 --- a/tests/storage/test_purge.py +++ b/tests/storage/test_purge.py @@ -102,9 +102,10 @@ class PurgeTests(HomeserverTestCase): first = self.helper.send(self.room_id, body="test1") # Get the current room state. - state_handler = self.hs.get_state_handler() create_event = self.get_success( - state_handler.get_current_state(self.room_id, "m.room.create", "") + self._storage_controllers.state.get_current_state_event( + self.room_id, "m.room.create", "" + ) ) self.assertIsNotNone(create_event) |