Fix up state_store naming (#12871)

This commit is contained in:
Erik Johnston 2022-05-25 12:59:04 +01:00 committed by GitHub
parent a8db8c6eba
commit 4660d9fdcf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
15 changed files with 58 additions and 49 deletions

View file

@ -50,7 +50,7 @@ class FederationTestCase(unittest.FederatingHomeserverTestCase):
hs = self.setup_test_homeserver(federation_http_client=None)
self.handler = hs.get_federation_handler()
self.store = hs.get_datastores().main
self.state_store = hs.get_storage().state
self.state_storage = hs.get_storage().state
self._event_auth_handler = hs.get_event_auth_handler()
return hs
@ -334,7 +334,7 @@ class FederationTestCase(unittest.FederatingHomeserverTestCase):
# mapping from (type, state_key) -> state_event_id
assert most_recent_prev_event_id is not None
prev_state_map = self.get_success(
self.state_store.get_state_ids_for_event(most_recent_prev_event_id)
self.state_storage.get_state_ids_for_event(most_recent_prev_event_id)
)
# List of state event ID's
prev_state_ids = list(prev_state_map.values())