feat(tests): additional tests for uploads

Signed-off-by: Arian Nasr <arian@2ari.ca>

e2e/unit/api/test_api.py

@@ -20,3 +20,47 @@ def test_api_upload_non_audio_file(base_url, upload_folder):
 
     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)