Add an assert to check that limit has been enforced after calling get…

…_devices_by_remote Signed-off-by: Quentin Dufour <[email protected]>
matrix-org · May 9, 2019 · 47b8871 · 47b8871
1 parent a363b8c
commit 47b8871
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/synapse/federation/sender/per_destination_queue.py b/synapse/federation/sender/per_destination_queue.py
@@ -362,6 +362,8 @@ def _get_new_device_messages(self, limit):
             for content in results
         ]
 
+        assert len(edus) <= limit, "get_devices_by_remote returned too many EDUs"
+
         last_device_stream_id = self._last_device_stream_id
         to_device_stream_id = self._store.get_to_device_stream_token()
         contents, stream_id = yield self._store.get_new_device_msgs_for_remote(