import numpy as np
import pytest
import soundfile as sf

from audio_files import export_take, prune_outputs, read_reference
from requests_schema import InputError


def test_silent_reference_rejected(tmp_path):
    p=tmp_path/'silent.wav';sf.write(p,np.zeros(16000),16000)
    with pytest.raises(InputError):read_reference(p)


def test_reference_duration_and_original_bytes(tmp_path):
    p=tmp_path/'voice.wav';sf.write(p,.1*np.sin(np.arange(16000)*.05),16000)
    assert read_reference(p)==p.read_bytes()
    short=tmp_path/'short.wav';sf.write(short,np.ones(20)*.1,16000)
    with pytest.raises(InputError):read_reference(short)


@pytest.mark.parametrize('pcm',[np.array([]),np.array([float('nan')]*4410),np.ones(4410)*1.5])
def test_invalid_output_is_not_saved(tmp_path,pcm):
    with pytest.raises(ValueError):export_take(pcm,tmp_path,seed=1,request_seconds=1,inference_seconds=.5)
    assert not list(tmp_path.iterdir())


def test_pruning_tolerates_another_session_removing_old_output(tmp_path, monkeypatch):
    from pathlib import Path
    old=tmp_path/'zonos-old.wav';old.write_bytes(b'old')
    fresh=tmp_path/'zonos-fresh.wav';fresh.write_bytes(b'fresh')
    original_stat=Path.stat
    def raced_stat(path, *args, **kwargs):
        if path == old:
            old.unlink(missing_ok=True)
            raise FileNotFoundError()
        return original_stat(path, *args, **kwargs)
    monkeypatch.setattr(Path,'stat',raced_stat)
    prune_outputs(tmp_path)
    assert fresh.read_bytes()==b'fresh'