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: Args:
user_id(str): the user whose backup we're adding to 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 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 = [] values = []

View file

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