feat(tests): additional tests for uploads
Signed-off-by: Arian Nasr <arian@2ari.ca>
This commit is contained in:
@@ -18,5 +18,49 @@ def test_api_upload_non_audio_file(base_url, upload_folder):
|
||||
assert "not allowed" in response.json().get("error", "")
|
||||
assert not os.path.exists(expected_filename)
|
||||
|
||||
if os.path.exists(expected_filename):
|
||||
os.remove(expected_filename)
|
||||
|
||||
def test_api_upload_empty_file(base_url, upload_folder):
|
||||
files = {'file': ('empty.mp3', io.BytesIO(b''))}
|
||||
expected_filename = os.path.join(upload_folder, secure_filename('empty.mp3'))
|
||||
response = requests.post(f'{base_url}/', files=files)
|
||||
assert response.status_code == 400
|
||||
assert "not allowed" in response.json().get("error", "")
|
||||
assert not os.path.exists(expected_filename)
|
||||
|
||||
if os.path.exists(expected_filename):
|
||||
os.remove(expected_filename)
|
||||
|
||||
def test_api_upload_mp3_file(base_url, upload_folder):
|
||||
files = {'file': ('test.mp3', io.BytesIO(b'fake mp3 content'))}
|
||||
expected_filename = os.path.join(upload_folder, secure_filename('test.mp3'))
|
||||
response = requests.post(f'{base_url}/', files=files)
|
||||
assert response.status_code == 200
|
||||
assert "uploaded successfully" in response.json().get("message", "")
|
||||
assert os.path.exists(expected_filename)
|
||||
|
||||
if os.path.exists(expected_filename):
|
||||
os.remove(expected_filename)
|
||||
|
||||
def test_api_upload_flac_file(base_url, upload_folder):
|
||||
files = {'file': ('test.flac', io.BytesIO(b'fake flac content'))}
|
||||
expected_filename = os.path.join(upload_folder, secure_filename('test.flac'))
|
||||
response = requests.post(f'{base_url}/', files=files)
|
||||
assert response.status_code == 200
|
||||
assert "uploaded successfully" in response.json().get("message", "")
|
||||
assert os.path.exists(expected_filename)
|
||||
|
||||
if os.path.exists(expected_filename):
|
||||
os.remove(expected_filename)
|
||||
|
||||
def test_api_upload_wav_file(base_url, upload_folder):
|
||||
files = {'file': ('test.wav', io.BytesIO(b'fake wav content'))}
|
||||
expected_filename = os.path.join(upload_folder, secure_filename('test.wav'))
|
||||
response = requests.post(f'{base_url}/', files=files)
|
||||
assert response.status_code == 400
|
||||
assert "not allowed" in response.json().get("error", "")
|
||||
assert not os.path.exists(expected_filename)
|
||||
|
||||
if os.path.exists(expected_filename):
|
||||
os.remove(expected_filename)
|
||||
Reference in New Issue
Block a user