Improve tests (#1047)

* improve tests

* does it hurt someone to have crazy numbers?

---------

Co-authored-by: henryruhs <info@henryruhs.com>
This commit is contained in:
Harisreedhar
2026-03-17 14:02:16 +01:00
committed by henryruhs
co-authored by henryruhs
parent d90b796b51
commit cd3023b9e1
2 changed files with 63 additions and 0 deletions
+19
View File
@@ -16,6 +16,7 @@ def before_all() -> None:
conditional_download(get_test_examples_directory(),
[
'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.jpg',
'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/source.mp3',
'https://github.com/facefusion/facefusion-assets/releases/download/examples-3.0.0/target-240p.mp4'
])
@@ -78,6 +79,24 @@ def test_upload_asset(test_client : TestClient) -> None:
assert upload_response.status_code == 201
audio_path = get_test_example_file('source.mp3')
with open(audio_path, 'rb') as audio_file:
audio_content = audio_file.read()
upload_response = test_client.post('/assets?type=source', headers =
{
'Authorization': 'Bearer ' + access_token
}, files =
[
('file', ('source.mp3', audio_content, 'audio/mpeg'))
])
upload_body = upload_response.json()
asset = asset_store.get_asset(session_id, upload_body.get('asset_ids')[0])
assert asset.get('media') == 'audio'
assert asset.get('type') == 'source'
assert upload_response.status_code == 201
upload_response = test_client.post('/assets?type=invalid', headers =
{
'Authorization': 'Bearer ' + access_token