Fix tests

synapse/storage/controllers/state.py

@@ -487,6 +487,6 @@ async def get_current_state_event(
  async def get_current_hosts_in_room(self, room_id: str) -> Set[str]:
  """Get current hosts in room based on current state."""
 
- await self._partial_state_room_tracker.await_full_state(room_id)
+ # await self._partial_state_room_tracker.await_full_state(room_id)
 
  return await self.stores.main.get_current_hosts_in_room(room_id)

tests/federation/test_federation_sender.py

@@ -34,7 +34,7 @@ def make_homeserver(self, reactor, clock):
  federation_transport_client=Mock(spec=["send_transaction"]),
  )
 
- hs.get_datastores().main.get_current_hosts_in_room = Mock(
+ hs.get_storage_controllers().state.get_current_hosts_in_room = Mock(
  return_value=make_awaitable(["test", "host2"])
  )
 

tests/handlers/test_typing.py

@@ -129,10 +129,12 @@ async def check_host_in_room(room_id: str, server_name: str) -> bool:
 
  hs.get_event_auth_handler().check_host_in_room = check_host_in_room
 
- def get_joined_hosts_for_room(room_id: str):
+ async def get_current_hosts_in_room(room_id: str):
  return {member.domain for member in self.room_members}
 
- self.datastore.get_joined_hosts_for_room = get_joined_hosts_for_room
+ hs.get_storage_controllers().state.get_current_hosts_in_room = (
+ get_current_hosts_in_room
+ )
 
  async def get_users_in_room(room_id: str):
  return {str(u) for u in self.room_members}