diff options
| author | Paul Buetow <paul@buetow.org> | 2026-04-01 20:29:50 +0300 |
|---|---|---|
| committer | Paul Buetow <paul@buetow.org> | 2026-04-01 20:29:50 +0300 |
| commit | 5ae64f1658fb8e1daa9400efab96f3fcee5d17c0 (patch) | |
| tree | 7cf31ca579cd010594431af0e37679c2a8a8da93 /internal/processor | |
| parent | 3c82415b3b9bca4b376d3044bb677438b17e491e (diff) | |
zg: add consumer coverage for shared voice lists
Diffstat (limited to 'internal/processor')
| -rw-r--r-- | internal/processor/processor.go | 4 | ||||
| -rw-r--r-- | internal/processor/processor_test.go | 72 |
2 files changed, 75 insertions, 1 deletions
diff --git a/internal/processor/processor.go b/internal/processor/processor.go index d2a380b..3b53737 100644 --- a/internal/processor/processor.go +++ b/internal/processor/processor.go @@ -38,6 +38,8 @@ var newNanoBananaImageClient = func(config *image.NanoBananaConfig) image.ImageS return image.NewNanoBananaClient(config) } +var newAudioProvider = audio.NewProvider + // NewProcessor creates a new word processor func NewProcessor(flags *cli.Flags) *Processor { openAIKey := cli.GetOpenAIKey() @@ -343,7 +345,7 @@ func (p *Processor) generateAudioWithVoiceAndFilenameInDir(word, voice, filename } // Create the audio provider - provider, err := audio.NewProvider(providerConfig) + provider, err := newAudioProvider(providerConfig) if err != nil { return err } diff --git a/internal/processor/processor_test.go b/internal/processor/processor_test.go index 8ab13e7..1b86910 100644 --- a/internal/processor/processor_test.go +++ b/internal/processor/processor_test.go @@ -8,6 +8,7 @@ import ( "strings" "testing" + "codeberg.org/snonux/totalrecall/internal/audio" "codeberg.org/snonux/totalrecall/internal/cli" "codeberg.org/snonux/totalrecall/internal/gui" "codeberg.org/snonux/totalrecall/internal/image" @@ -56,6 +57,27 @@ func (s *stubImageSearcher) GetLastPrompt() string { return s.lastPrompt } +type fakeAudioProvider struct { + generateCalls int + lastText string + lastOutputFile string +} + +func (f *fakeAudioProvider) GenerateAudio(_ context.Context, text, outputFile string) error { + f.generateCalls++ + f.lastText = text + f.lastOutputFile = outputFile + return nil +} + +func (f *fakeAudioProvider) Name() string { + return "fake-audio" +} + +func (f *fakeAudioProvider) IsAvailable() error { + return nil +} + func TestNewProcessor(t *testing.T) { t.Setenv("OPENAI_API_KEY", "test-openai-key") t.Setenv("GOOGLE_API_KEY", "test-google-key") @@ -231,6 +253,56 @@ func TestGUIConfigForRunModeHonorsExplicitImageAPI(t *testing.T) { } } +func TestGenerateAudioUsesSharedOpenAIVoices(t *testing.T) { + originalFactory := newAudioProvider + t.Cleanup(func() { + newAudioProvider = originalFactory + }) + + originalVoices := append([]string(nil), audio.OpenAIVoices...) + t.Cleanup(func() { + audio.OpenAIVoices = originalVoices + }) + + audio.OpenAIVoices = []string{"sentinel-openai-voice"} + + fakeProvider := &fakeAudioProvider{} + var capturedConfig *audio.Config + newAudioProvider = func(config *audio.Config) (audio.Provider, error) { + copyConfig := *config + capturedConfig = ©Config + return fakeProvider, nil + } + + tempDir := t.TempDir() + flags := cli.NewFlags() + flags.OutputDir = tempDir + flags.AudioFormat = "mp3" + flags.AllVoices = true + + p := NewProcessor(flags) + + if err := p.generateAudio("ябълка"); err != nil { + t.Fatalf("generateAudio() unexpected error: %v", err) + } + + if capturedConfig == nil { + t.Fatal("expected audio provider config to be captured") + } + if capturedConfig.OpenAIVoice != "sentinel-openai-voice" { + t.Fatalf("captured OpenAIVoice = %q, want %q", capturedConfig.OpenAIVoice, "sentinel-openai-voice") + } + if fakeProvider.generateCalls != 1 { + t.Fatalf("GenerateAudio() calls = %d, want %d", fakeProvider.generateCalls, 1) + } + if fakeProvider.lastText != "ябълка" { + t.Fatalf("GenerateAudio() text = %q, want %q", fakeProvider.lastText, "ябълка") + } + if !strings.HasSuffix(fakeProvider.lastOutputFile, "audio_sentinel-openai-voice.mp3") { + t.Fatalf("GenerateAudio() output file = %q, want shared voice name in filename", fakeProvider.lastOutputFile) + } +} + func TestDownloadImagesWithTranslationUsesNanoBananaConfigAndSavesPrompt(t *testing.T) { t.Setenv("OPENAI_API_KEY", "test-openai-key") t.Setenv("GOOGLE_API_KEY", "test-google-key") |
