46 lines
1.5 KiB
Go
46 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// The wiring of the per-engine deadline: a gateway whose primary model never answers
|
|
// must still reach the fallback model inside the job's own deadline (adversarial review
|
|
// F4). The fake gateway answers per model from the multipart `model` field.
|
|
|
|
func TestBuildTranscriber_AWedgedPrimaryReachesTheFallbackBeforeTheJobDeadline(t *testing.T) {
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
_ = r.ParseMultipartForm(1 << 20)
|
|
if r.FormValue("model") == "iguana/kb-whisper" {
|
|
<-r.Context().Done() // wedged: never answers
|
|
return
|
|
}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_, _ = w.Write([]byte(`{"text":"hej","duration":1,"segments":[{"start":0,"end":1,"text":"hej"}]}`))
|
|
}))
|
|
defer srv.Close()
|
|
cfg := uploadCfg(t, true)
|
|
cfg.GatewayURL = srv.URL
|
|
cfg.TranscribeModel = "iguana/kb-whisper"
|
|
cfg.TranscribeFallbackModel = "berget/kb-whisper"
|
|
cfg.TranscribeEngineTimeout = 150 * time.Millisecond
|
|
wav := filepath.Join(t.TempDir(), "a.wav")
|
|
require.NoError(t, os.WriteFile(wav, []byte("RIFF....WAVEfmt "), 0o600))
|
|
job, cancel := context.WithTimeout(context.Background(), 20*time.Second)
|
|
defer cancel()
|
|
|
|
got, err := buildTranscriber(cfg).Transcribe(job, wav)
|
|
|
|
require.NoError(t, err, "the wedged primary used the whole job and the fallback never ran")
|
|
assert.Equal(t, "berget/kb-whisper", got.Endpoint)
|
|
}
|