diff options
Diffstat (limited to '')
-rw-r--r-- | tests/test_state.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tests/test_state.py b/tests/test_state.py index 651ec1c7d4..74a8ce6096 100644 --- a/tests/test_state.py +++ b/tests/test_state.py @@ -129,6 +129,19 @@ class _DummyStore: async def get_room_version_id(self, room_id): return RoomVersions.V1.identifier + async def get_state_group_for_events(self, event_ids): + res = {} + for event in event_ids: + res[event] = self._event_to_state_group[event] + return res + + async def get_state_for_groups(self, groups): + res = {} + for group in groups: + state = self._group_to_state[group] + res[group] = state + return res + class DictObj(dict): def __init__(self, **kwargs): |