fork and join sessions (#1239)

* introduce fork and join session, fix inference manager race condition

* destroy stream on DELETE
This commit is contained in:
Henry Ruhs
2026-09-12 15:45:30 +02:00
committed by GitHub
parent b5d396f4b1
commit 106dc7a348
9 changed files with 172 additions and 28 deletions
+22 -2
View File
@@ -1,3 +1,4 @@
import ctypes
import os
import tempfile
from datetime import timedelta
@@ -7,11 +8,12 @@ from unittest.mock import patch
import pytest
from starlette.testclient import TestClient
from facefusion import metadata, process_manager, session_manager, state_manager
from facefusion import metadata, process_manager, rtc, rtc_store, session_manager, state_manager
from facefusion.apis import asset_store
from facefusion.apis.core import create_api
from facefusion.download import conditional_download
from facefusion.types import Session
from facefusion.libraries import datachannel as datachannel_module
from facefusion.types import RtcPeer, Session
from .assert_helper import get_test_example_file, get_test_examples_directory
@@ -19,6 +21,8 @@ from .assert_helper import get_test_example_file, get_test_examples_directory
def before_all() -> None:
state_manager.init()
datachannel_module.pre_check()
process_manager.start()
conditional_download(get_test_examples_directory(),
[
@@ -238,6 +242,21 @@ def test_destroy_session(test_client : TestClient) -> None:
assert session_manager.find_session_id(access_token) == session_id
assert delete_session_response.status_code == 404
peer_connection = rtc.create_peer_connection()
rtc_peer : RtcPeer =\
{
'peer_connection': peer_connection,
'video':
{
'sender_track': rtc.add_video_track(peer_connection, 'sendonly', 'vp8', 96),
'receiver_track': 0,
'codec': 'vp8'
},
'sender_bitrate': ctypes.c_uint(0),
'receiver_bitrate': ctypes.c_uint(0)
}
rtc_store.set_peer(session_id, rtc_peer)
delete_session_response = test_client.delete('/session', headers =
{
'Authorization': 'Bearer ' + access_token
@@ -245,6 +264,7 @@ def test_destroy_session(test_client : TestClient) -> None:
assert session_manager.find_session_id(access_token) is None
assert asset_store.get_assets(session_id) is None
assert rtc_store.has_peer(session_id) is False
assert delete_session_response.status_code == 200
for asset_path in asset_paths: