diff options
author | Erik Johnston <erik@matrix.org> | 2022-07-15 13:59:45 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-07-15 12:59:45 +0000 |
commit | 0731e0829c08aec7f31fdc72c236757e4cc38747 (patch) | |
tree | b1a7046cf12a1bf24f9affd9fdca26180b863d0f /tests | |
parent | Use a real room in the notification rotation tests. (#13260) (diff) | |
download | synapse-0731e0829c08aec7f31fdc72c236757e4cc38747.tar.xz |
Don't pull out the full state when storing state (#13274)
Diffstat (limited to 'tests')
-rw-r--r-- | tests/rest/client/test_rooms.py | 4 | ||||
-rw-r--r-- | tests/test_state.py | 4 |
2 files changed, 6 insertions, 2 deletions
diff --git a/tests/rest/client/test_rooms.py b/tests/rest/client/test_rooms.py index 8ed5272b16..06221b806a 100644 --- a/tests/rest/client/test_rooms.py +++ b/tests/rest/client/test_rooms.py @@ -709,7 +709,7 @@ class RoomsCreateTestCase(RoomBase): self.assertEqual(200, channel.code, channel.result) self.assertTrue("room_id" in channel.json_body) assert channel.resource_usage is not None - self.assertEqual(32, channel.resource_usage.db_txn_count) + self.assertEqual(36, channel.resource_usage.db_txn_count) def test_post_room_initial_state(self) -> None: # POST with initial_state config key, expect new room id @@ -722,7 +722,7 @@ class RoomsCreateTestCase(RoomBase): self.assertEqual(200, channel.code, channel.result) self.assertTrue("room_id" in channel.json_body) assert channel.resource_usage is not None - self.assertEqual(35, channel.resource_usage.db_txn_count) + self.assertEqual(40, channel.resource_usage.db_txn_count) def test_post_room_visibility_key(self) -> None: # POST with visibility config key, expect new room id diff --git a/tests/test_state.py b/tests/test_state.py index e2c0013671..bafd6d1750 100644 --- a/tests/test_state.py +++ b/tests/test_state.py @@ -99,6 +99,10 @@ class _DummyStore: state_group = self._next_group self._next_group += 1 + if current_state_ids is None: + current_state_ids = dict(self._group_to_state[prev_group]) + current_state_ids.update(delta_ids) + self._group_to_state[state_group] = dict(current_state_ids) return state_group |