import os
import pytest
from src.whisper_audio_extractor import record_audio, transcribe_audio, AUDIO_PATH

def test_record_audio():
    """
    Verifica que la función de grabación crea un archivo de audio con un tamaño válido.
    """
    # Ejecuta la grabación con una duración de prueba corta
    record_audio(duration=2)  # Graba por 2 segundos para el test
    
    # Comprueba si el archivo de audio existe
    assert os.path.exists(AUDIO_PATH), "El archivo de audio no fue creado."

    # Comprueba que el archivo no esté vacío
    assert os.path.getsize(AUDIO_PATH) > 0, "El archivo de audio está vacío."

def test_transcribe_audio():
    """
    Verifica que la función de transcripción devuelve texto.
    """
    # Ejecuta la transcripción del audio grabado
    transcription = transcribe_audio()
    
    # Asegura que se obtuvo texto
    assert isinstance(transcription, str) and len(transcription) > 0, "La transcripción está vacía o no es texto."

if __name__ == "__main__":
    pytest.main()