mirror of
https://github.com/facefusion/facefusion.git
synced 2026-04-23 01:46:09 +02:00
Henry Ruhs
6180ccf6e6
* Add API scopes * Add API scopes * Add API scopes * Add API scopes * Add API scopes * Add API scopes * Add API scopes * Add API scopes * Remove system memory limit (#986) # Conflicts: # facefusion/program.py * Add session_id, make token size more reasonable (#983) * Add session_id, make token size more reasonable * Use more direct approach * Fix more stuff * Fix ignore comments * Fix naming * Fix lint
82 lines
2.2 KiB
Python
82 lines
2.2 KiB
Python
from typing import Iterator
|
|
|
|
import pytest
|
|
from starlette.testclient import TestClient
|
|
|
|
from facefusion import args_store, metadata, session_manager, state_manager
|
|
from facefusion.apis.core import create_api
|
|
|
|
|
|
@pytest.fixture(scope = 'module')
|
|
def test_client() -> Iterator[TestClient]:
|
|
args_store.register_args([ 'execution_providers' ], scopes = [ 'api' ])
|
|
state_manager.init_item('execution_providers', [ 'cpu' ])
|
|
|
|
with TestClient(create_api()) as test_client:
|
|
yield test_client
|
|
|
|
|
|
@pytest.fixture(scope = 'function', autouse = True)
|
|
def before_each() -> None:
|
|
session_manager.SESSIONS.clear()
|
|
|
|
|
|
def test_get_state(test_client : TestClient) -> None:
|
|
get_state_response = test_client.get('/state')
|
|
|
|
assert get_state_response.status_code == 401
|
|
|
|
create_session_response = test_client.post('/session', json =
|
|
{
|
|
'client_version': metadata.get('version')
|
|
})
|
|
create_session_body = create_session_response.json()
|
|
|
|
get_state_response = test_client.get('/state', headers =
|
|
{
|
|
'Authorization': 'Bearer ' + create_session_body.get('access_token')
|
|
})
|
|
get_state_body = get_state_response.json()
|
|
|
|
assert get_state_body.get('execution_providers') == [ 'cpu' ]
|
|
assert get_state_response.status_code == 200
|
|
|
|
|
|
def test_set_state(test_client : TestClient) -> None:
|
|
set_state_response = test_client.put('/state', json =
|
|
{
|
|
'execution_providers': [ 'cuda' ]
|
|
})
|
|
|
|
assert set_state_response.status_code == 401
|
|
|
|
create_session_response = test_client.post('/session', json =
|
|
{
|
|
'client_version': metadata.get('version')
|
|
})
|
|
create_session_body = create_session_response.json()
|
|
|
|
set_state_response = test_client.put('/state', json =
|
|
{
|
|
'execution_providers': [ 'cuda' ]
|
|
}, headers =
|
|
{
|
|
'Authorization': 'Bearer ' + create_session_body.get('access_token')
|
|
})
|
|
set_state_body = set_state_response.json()
|
|
|
|
assert set_state_body.get('execution_providers') == [ 'cuda' ]
|
|
assert set_state_response.status_code == 200
|
|
|
|
set_state_response = test_client.put('/state', json =
|
|
{
|
|
'invalid': 'invalid'
|
|
}, headers =
|
|
{
|
|
'Authorization': 'Bearer ' + create_session_body.get('access_token')
|
|
})
|
|
set_state_body = set_state_response.json()
|
|
|
|
assert set_state_body.get('invalid') is None
|
|
assert set_state_response.status_code == 200
|