Skip to content
This repository has been archived by the owner on Apr 26, 2024. It is now read-only.

Commit

Permalink
Fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
erikjohnston committed Mar 29, 2022
1 parent f8af30f commit 0af131a
Showing 1 changed file with 10 additions and 4 deletions.
14 changes: 10 additions & 4 deletions tests/storage/test_devices.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,9 @@ def test_get_device_updates_by_remote(self):

# Add two device updates with sequential `stream_id`s
self.get_success(
self.store.add_device_change_to_streams("user_id", device_ids, ["somehost"])
self.store.add_device_change_to_streams(
"user_id", device_ids, ["somehost"], []
)
)

# Get all device updates ever meant for this remote
Expand All @@ -122,7 +124,9 @@ def test_get_device_updates_by_remote_can_limit_properly(self):
"device_id5",
]
self.get_success(
self.store.add_device_change_to_streams("user_id", device_ids, ["somehost"])
self.store.add_device_change_to_streams(
"user_id", device_ids, ["somehost"], []
)
)

# Get device updates meant for this remote
Expand All @@ -144,7 +148,9 @@ def test_get_device_updates_by_remote_can_limit_properly(self):
# Add some more device updates to ensure it still resumes properly
device_ids = ["device_id6", "device_id7"]
self.get_success(
self.store.add_device_change_to_streams("user_id", device_ids, ["somehost"])
self.store.add_device_change_to_streams(
"user_id", device_ids, ["somehost"], []
)
)

# Get the next batch of device updates
Expand Down Expand Up @@ -220,7 +226,7 @@ def test_get_device_updates_by_remote_cross_signing_key_updates(

self.get_success(
self.store.add_device_change_to_streams(
"@user_id:test", device_ids, ["somehost"]
"@user_id:test", device_ids, ["somehost"], []
)
)

Expand Down

0 comments on commit 0af131a

Please sign in to comment.