mirror of
https://github.com/facefusion/facefusion.git
synced 2026-09-18 13:32:16 +02:00
fork and join sessions (#1239)
* introduce fork and join session, fix inference manager race condition * destroy stream on DELETE
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user