CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-06-07 21:23:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

state api updates

Browse Source
This commit is contained in:
harisreedhar
2026-01-09 18:09:41 +05:30
committed by henryruhs
parent 8624d6178c
commit 151ec22a35
4 changed files with 167 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_api_state.py
+82
View File
@@ -5,10 +5,12 @@ from starlette.testclient import TestClient
from facefusion import args_store, metadata, session_manager, state_manager
from facefusion.apis.core import create_api
from facefusion.apis import asset_store
@pytest.fixture(scope = 'module')
def test_client() -> Iterator[TestClient]:
args_store.register_args([ 'source_paths', 'target_path' ], scopes = [ 'api' ])
args_store.register_args([ 'execution_providers' ], scopes = [ 'api' ])
state_manager.init_item('execution_providers', [ 'cpu' ])
@@ -19,6 +21,7 @@ def test_client() -> Iterator[TestClient]:
@pytest.fixture(scope = 'function', autouse = True)
def before_each() -> None:
session_manager.SESSIONS.clear()
asset_store.clear()
def test_get_state(test_client : TestClient) -> None:
@@ -79,3 +82,82 @@ def test_set_state(test_client : TestClient) -> None:
assert set_state_body.get('invalid') is None
assert set_state_response.status_code == 200
def test_select_source_assets(test_client : TestClient) -> None:
asset_id_1 = asset_store.register_asset('/path/to/source1.jpg')
asset_id_2 = asset_store.register_asset('/path/to/source2.jpg')
select_response = test_client.put('/state?action=select&type=source', json =
{
'asset_ids': [ asset_id_1, asset_id_2 ]
})
assert select_response.status_code == 401
create_session_response = test_client.post('/session', json =
{
'client_version': metadata.get('version')
})
create_session_body = create_session_response.json()
select_response = test_client.put('/state?action=select&type=source', json =
{
'asset_ids': 'invalid_string_not_list'
}, headers =
{
'Authorization': 'Bearer ' + create_session_body.get('access_token')
})
assert select_response.status_code == 404
select_response = test_client.put('/state?action=select&type=source', json =
{
'asset_ids': [ asset_id_1, asset_id_2 ]
}, headers =
{
'Authorization': 'Bearer ' + create_session_body.get('access_token')
})
select_body = select_response.json()
assert select_body.get('source_paths') == [ '/path/to/source1.jpg', '/path/to/source2.jpg' ]
assert select_response.status_code == 200
def test_select_target_assets(test_client : TestClient) -> None:
asset_id_1 = asset_store.register_asset('/path/to/target1.jpg')
select_response = test_client.put('/state?action=select&type=target', json =
{
'asset_id': asset_id_1
})
assert select_response.status_code == 401
create_session_response = test_client.post('/session', json =
{
'client_version': metadata.get('version')
})
create_session_body = create_session_response.json()
select_response = test_client.put('/state?action=select&type=target', json =
{
'asset_id': 'invalid_asset_id'
}, headers =
{
'Authorization': 'Bearer ' + create_session_body.get('access_token')
})
assert select_response.status_code == 404
select_response = test_client.put('/state?action=select&type=target', json =
{
'asset_id': asset_id_1
}, headers =
{
'Authorization': 'Bearer ' + create_session_body.get('access_token')
})
select_body = select_response.json()
assert select_body.get('target_path') == '/path/to/target1.jpg'
assert select_response.status_code == 200