CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-05-31 01:49:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

move stream mode to query parameter (#1089)

Browse Source
This commit is contained in:
Henry Ruhs
2026-05-08 19:16:40 +02:00
committed by GitHub
parent 5a81f16142
commit 35e68d17e1
4 changed files with 16 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_api_helper.py
+11
View File
@@ -0,0 +1,11 @@
from facefusion.apis.api_helper import get_sec_websocket_protocol
def test_get_sec_websocket_protocol() -> None:
scope =\
{
'type': 'websocket',
'headers': [ (b'sec-websocket-protocol', b'access_token.abc') ]
}
assert get_sec_websocket_protocol(scope) == 'access_token.abc'
tests/test_stream_helper.py
+1 -19
View File
@@ -1,6 +1,6 @@
import os
from facefusion.apis.stream_helper import calculate_bitrate, calculate_buffer_size, detect_websocket_stream_mode, read_pipe_buffer
from facefusion.apis.stream_helper import calculate_bitrate, calculate_buffer_size, read_pipe_buffer
def test_calculate_bitrate() -> None:
@@ -19,24 +19,6 @@ def test_calculate_buffer_size() -> None:
assert calculate_buffer_size((3840, 2160)) == 14000
def test_detect_websocket_stream_mode() -> None:
scope =\
{
'type': 'websocket',
'headers': [ (b'sec-websocket-protocol', b'image') ]
}
assert detect_websocket_stream_mode(scope) == 'image'
scope =\
{
'type': 'websocket',
'headers': [ (b'sec-websocket-protocol', b'video') ]
}
assert detect_websocket_stream_mode(scope) == 'video'
def test_read_pipe_buffer() -> None:
read_pipe, write_pipe = os.pipe()
os.write(write_pipe, b'123456')