fix unit test to use new API

and improve docstring
This commit is contained in:
Hubert Chathi 2019-11-06 20:47:14 -05:00
parent 00fa0c9f14
commit 8721067448
2 changed files with 6 additions and 5 deletions

View file

@ -62,7 +62,8 @@ class EndToEndRoomKeyStore(SQLBaseStore):
Args:
user_id(str): the user whose backup we're adding to
version(str): the version ID of the backup for the set of keys we're adding to
room_keys(iterable[dict]): the keys to add
room_keys(iterable[(str, str, dict)]): the keys to add, in the form
(roomID, sessionID, keyData)
"""
values = []

View file

@ -39,8 +39,8 @@ class E2eRoomKeysHandlerTestCase(unittest.HomeserverTestCase):
)
self.get_success(
self.store.set_e2e_room_key(
"user_id", version1, "room", "session", room_key
self.store.add_e2e_room_keys(
"user_id", version1, [("room", "session", room_key)]
)
)
@ -51,8 +51,8 @@ class E2eRoomKeysHandlerTestCase(unittest.HomeserverTestCase):
)
self.get_success(
self.store.set_e2e_room_key(
"user_id", version2, "room", "session", room_key
self.store.add_e2e_room_keys(
"user_id", version2, [("room", "session", room_key)]
)
)