Gan, Jimmy
89 lines
3.1 KiB
Python
89 lines
3.1 KiB
Python
import io
|
|
import sys
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
# Mock whisper before importing server
|
|
mock_whisper = MagicMock()
|
|
mock_whisper.load_model.return_value = MagicMock()
|
|
sys.modules['whisper'] = mock_whisper
|
|
|
|
from server import app
|
|
|
|
@pytest.fixture
|
|
def client():
|
|
app.config['TESTING'] = True
|
|
with app.test_client() as c:
|
|
yield c
|
|
|
|
# --- Static file serving ---
|
|
|
|
def test_index(client):
|
|
resp = client.get('/')
|
|
assert resp.status_code == 200
|
|
|
|
def test_allowed_static_file(client):
|
|
resp = client.get('/app.js')
|
|
assert resp.status_code == 200
|
|
|
|
def test_blocked_static_file(client):
|
|
resp = client.get('/server.py')
|
|
assert resp.status_code == 404
|
|
|
|
def test_path_traversal_blocked(client):
|
|
resp = client.get('/../requirements.txt')
|
|
assert resp.status_code == 404
|
|
|
|
# --- COOP/COEP headers ---
|
|
|
|
def test_coop_coep_headers(client):
|
|
resp = client.get('/')
|
|
assert resp.headers['Cross-Origin-Opener-Policy'] == 'same-origin'
|
|
assert resp.headers['Cross-Origin-Embedder-Policy'] == 'require-corp'
|
|
|
|
# --- /transcribe endpoint ---
|
|
|
|
def test_transcribe_no_file(client):
|
|
resp = client.post('/transcribe')
|
|
assert resp.status_code == 400
|
|
assert 'No file provided' in resp.get_json()['error']
|
|
|
|
def test_transcribe_empty_filename(client):
|
|
data = {'file': (io.BytesIO(b'data'), '')}
|
|
resp = client.post('/transcribe', data=data, content_type='multipart/form-data')
|
|
assert resp.status_code == 400
|
|
assert 'No file selected' in resp.get_json()['error']
|
|
|
|
def test_transcribe_video_rejected(client):
|
|
for ext in ['.mp4', '.avi', '.mov', '.mkv', '.flv', '.wmv', '.webm']:
|
|
data = {'file': (io.BytesIO(b'data'), f'video{ext}')}
|
|
resp = client.post('/transcribe', data=data, content_type='multipart/form-data')
|
|
assert resp.status_code == 400
|
|
assert 'Video files not allowed' in resp.get_json()['error']
|
|
|
|
@patch('server.model')
|
|
def test_transcribe_success(mock_model, client):
|
|
mock_model.transcribe.return_value = {'text': 'hello world'}
|
|
data = {'file': (io.BytesIO(b'audio data'), 'test.mp3')}
|
|
resp = client.post('/transcribe', data=data, content_type='multipart/form-data')
|
|
assert resp.status_code == 200
|
|
assert resp.get_json()['text'] == 'hello world'
|
|
|
|
@patch('server.model')
|
|
def test_transcribe_with_language(mock_model, client):
|
|
mock_model.transcribe.return_value = {'text': 'hola'}
|
|
data = {'file': (io.BytesIO(b'audio'), 'test.mp3'), 'language': 'es-ES'}
|
|
resp = client.post('/transcribe', data=data, content_type='multipart/form-data')
|
|
assert resp.status_code == 200
|
|
mock_model.transcribe.assert_called_once()
|
|
assert mock_model.transcribe.call_args[1]['language'] == 'es'
|
|
|
|
@patch('server.model')
|
|
def test_transcribe_cleans_up_temp_file_on_error(mock_model, client):
|
|
mock_model.transcribe.side_effect = RuntimeError('model failed')
|
|
data = {'file': (io.BytesIO(b'audio'), 'test.mp3')}
|
|
resp = client.post('/transcribe', data=data, content_type='multipart/form-data')
|
|
assert resp.status_code == 500
|
|
assert 'model failed' in resp.get_json()['error']
|