mirror of
https://github.com/facefusion/facefusion.git
synced 2026-05-12 18:32:18 +02:00
Local API (#982)
* Introduce API scelleton * Raw impl for session * Simple state endpoint * Apply _body naming * Finalize session testing and comment out tons of useless code * Clean and refactor part1 * Clean and refactor part2 * Clean and refactor part2 * Clean and refactor part2 * Clean and refactor part2 * Refactor middleware * Refactor middleware * Clean and refactor part3 * TDD and 2 beers * TDD and 2 beers * Complete state endpoints * You can only set what is already present * Use only JSON as response * Use default logger * Improve auth extraction * Extend api command with more args * Adjust API messages
This commit is contained in:
@@ -0,0 +1,80 @@
|
||||
from typing import Iterator
|
||||
|
||||
import pytest
|
||||
from starlette.testclient import TestClient
|
||||
|
||||
from facefusion import metadata, session_manager, state_manager
|
||||
from facefusion.apis.core import create_api
|
||||
|
||||
|
||||
@pytest.fixture(scope = 'module')
|
||||
def test_client() -> Iterator[TestClient]:
|
||||
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
|
||||
Reference in New Issue
Block a user