CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-04-22 17:36:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

Simplify testing

Browse Source
This commit is contained in:
henryruhs
2026-01-17 11:13:13 +01:00
parent 4ba6f7643e
commit 1fe6e2d6de
1 changed files with 38 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_api_assets.py
+38 -87
View File
@@ -31,119 +31,70 @@ def before_each() -> None:
asset_store.clear()
def test_upload_asset_without_auth(test_client : TestClient) -> None:
def test_upload_asset(test_client : TestClient) -> None:
upload_response = test_client.post('/assets?type=source')
assert upload_response.status_code == 401
def test_upload_asset_invalid_type(test_client : TestClient) -> None:
create_session_response = test_client.post('/session', json =
{
'client_version': metadata.get('version')
})
create_session_body = create_session_response.json()
access_token = create_session_body.get('access_token')
session_id = session_manager.find_session_id(access_token)
source_path = get_test_example_file('source.jpg')
target_path = get_test_example_file('target-240p.mp4')
with open(source_path, 'rb') as source_file, open(target_path, 'rb') as target_file:
source_content = source_file.read()
target_content = target_file.read()
upload_response = test_client.post('/assets?type=source', headers =
{
'Authorization': 'Bearer ' + access_token
}, files =
[
('file', ('source.jpg', source_content, 'image/jpeg')),
('file', ('target.mp4', target_content, 'video/mp4'))
])
upload_body = upload_response.json()
asset_ids = upload_body.get('asset_ids')
asset = asset_store.get_asset(session_id, asset_ids[0])
assert asset.get('media') == 'image'
assert asset.get('type') == 'source'
assert asset.get('format') == 'jpeg'
asset = asset_store.get_asset(session_id, asset_ids[1])
assert asset.get('media') == 'video'
assert asset.get('type') == 'source'
assert asset.get('format') == 'mp4'
assert upload_response.status_code == 201
upload_response = test_client.post('/assets?type=invalid', headers =
{
'Authorization': 'Bearer ' + create_session_body.get('access_token')
'Authorization': 'Bearer ' + access_token
})
assert upload_response.status_code == 400
def test_upload_asset_no_file(test_client : TestClient) -> None:
create_session_response = test_client.post('/session', json =
{
'client_version': metadata.get('version')
})
create_session_body = create_session_response.json()
upload_response = test_client.post('/assets?type=source', headers =
{
'Authorization': 'Bearer ' + create_session_body.get('access_token')
'Authorization': 'Bearer ' + access_token
})
assert upload_response.status_code == 400
def test_upload_source_asset(test_client : TestClient) -> None:
create_session_response = test_client.post('/session', json =
{
'client_version': metadata.get('version')
})
create_session_body = create_session_response.json()
with open(get_test_example_file('source.jpg'), 'rb') as source_file:
upload_response = test_client.post('/assets?type=source', headers =
{
'Authorization': 'Bearer ' + create_session_body.get('access_token')
}, files =
{
'file': ('source.jpg', source_file, 'image/jpeg')
})
assert upload_response.status_code == 201
assert upload_response.json().get('asset_ids')
assert len(upload_response.json().get('asset_ids')) == 1
def test_upload_multiple_source_assets(test_client : TestClient) -> None:
create_session_response = test_client.post('/session', json =
{
'client_version': metadata.get('version')
})
create_session_body = create_session_response.json()
with open(get_test_example_file('source.jpg'), 'rb') as source_file:
source_content = source_file.read()
upload_response = test_client.post('/assets?type=source', headers =
{
'Authorization': 'Bearer ' + create_session_body.get('access_token')
}, files =
[
('file', ('source1.jpg', source_content, 'image/jpeg')),
('file', ('source2.jpg', source_content, 'image/jpeg'))
])
assert upload_response.status_code == 201
assert upload_response.json().get('asset_ids')
assert len(upload_response.json().get('asset_ids')) == 2
def test_upload_target_asset(test_client : TestClient) -> None:
create_session_response = test_client.post('/session', json =
{
'client_version': metadata.get('version')
})
create_session_body = create_session_response.json()
with open(get_test_example_file('target-240p.mp4'), 'rb') as target_file:
upload_response = test_client.post('/assets?type=target', headers =
{
'Authorization': 'Bearer ' + create_session_body.get('access_token')
}, files =
{
'file': ('target.mp4', target_file, 'video/mp4')
})
assert upload_response.status_code == 201
assert upload_response.json().get('asset_ids')
def test_upload_unsupported_format(test_client : TestClient) -> None:
create_session_response = test_client.post('/session', json =
{
'client_version': metadata.get('version')
})
create_session_body = create_session_response.json()
upload_response = test_client.post('/assets?type=source', headers =
{
'Authorization': 'Bearer ' + create_session_body.get('access_token')
'Authorization': 'Bearer ' + access_token
}, files =
{
'file': ('test.txt', b'invalid', 'text/plain')
'file': ('invalid.txt', 'invalid'.encode(), 'text/plain')
})
assert upload_response.status_code == 400