[matter_yamltests] Increase the maximum message size that can be rece…

…ived by the websocket (#27005)
project-chip · Aug 10, 2023 · 3248193 · 3248193
1 parent d22546c
commit 3248193
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/py_matter_yamltests/matter_yamltests/websocket_runner.py b/scripts/py_matter_yamltests/matter_yamltests/websocket_runner.py
@@ -23,6 +23,7 @@
 from .runner import TestRunner
 
 _KEEP_ALIVE_TIMEOUT_IN_SECONDS = 40
+_MAX_MESSAGE_SIZE_IN_BYTES = 10485760  # 10 MB
 
 
 @dataclass
@@ -67,7 +68,7 @@ async def _start_client(self, url, max_retries=4, interval_between_retries=1):
             start = time.time()
             try:
                 self._hooks.connecting(url)
-                connection = await websockets.connect(url, ping_timeout=_KEEP_ALIVE_TIMEOUT_IN_SECONDS)
+                connection = await websockets.connect(url, ping_timeout=_KEEP_ALIVE_TIMEOUT_IN_SECONDS, max_size=_MAX_MESSAGE_SIZE_IN_BYTES)
                 duration = round((time.time() - start) * 1000, 0)
                 self._hooks.success(duration)
                 return connection