CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-09-15 20:15:28 +02:00
Code Issues Packages Projects Releases Wiki Activity

handle garbage and empty buffer for websocket (#1232)

Browse Source
This commit is contained in:
Henry Ruhs
2026-09-11 11:06:04 +02:00
committed by GitHub
parent 743ebb8622
commit 41ee571fc0
3 changed files with 25 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_api_stream_manager.py
+16 -2
View File
@@ -42,6 +42,9 @@ async def test_process_image() -> None:
{
'type': 'websocket.receive',
'bytes': image_buffer
},
{
'type': 'websocket.disconnect'
}
]
@@ -68,14 +71,25 @@ async def test_receive_vision_frames() -> None:
'type': 'websocket.receive',
'bytes': 'invalid'.encode()
},
{
'type': 'websocket.receive',
'bytes': bytes()
},
{
'type': 'websocket.receive',
'bytes': image_buffer
},
{
'type': 'websocket.disconnect'
}
]
vision_frames = []
vision_frames = receive_vision_frames(websocket_mock)
async for vision_frame in receive_vision_frames(websocket_mock):
vision_frames.append(vision_frame)
assert create_hash((await anext(vision_frames)).tobytes()) == '5ed32ca0'
assert len(vision_frames) == 2
assert create_hash(vision_frames[0].tobytes()) == '5ed32ca0'
@pytest.mark.parametrize('video_codec, session_id', [ ('av1', 'test-process-video-av1'), ('vp8', 'test-process-video-vp8') ])