From b7302580d840fc3d7e3d127df49d79ef12d79d9c Mon Sep 17 00:00:00 2001 From: Paul Buetow Date: Thu, 2 Apr 2026 07:56:31 +0300 Subject: Align GUI and CLI provider wiring --- internal/gui/generator_test.go | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) (limited to 'internal/gui/generator_test.go') diff --git a/internal/gui/generator_test.go b/internal/gui/generator_test.go index f9cb2a0..31b4aa2 100644 --- a/internal/gui/generator_test.go +++ b/internal/gui/generator_test.go @@ -107,9 +107,11 @@ func TestGenerateImagesWithPromptUsesNanoBananaProvider(t *testing.T) { tempDir := t.TempDir() app := &Application{ config: &Config{ - ImageProvider: imageProviderNanoBanana, - GoogleAPIKey: "google-key", - OutputDir: tempDir, + ImageProvider: imageProviderNanoBanana, + GoogleAPIKey: "google-key", + NanoBananaModel: "custom-image-model", + NanoBananaTextModel: "custom-text-model", + OutputDir: tempDir, }, currentWord: "друго", } @@ -125,6 +127,12 @@ func TestGenerateImagesWithPromptUsesNanoBananaProvider(t *testing.T) { if capturedConfig.APIKey != "google-key" { t.Fatalf("Nano Banana API key = %q, want %q", capturedConfig.APIKey, "google-key") } + if capturedConfig.Model != "custom-image-model" { + t.Fatalf("Nano Banana model = %q, want %q", capturedConfig.Model, "custom-image-model") + } + if capturedConfig.TextModel != "custom-text-model" { + t.Fatalf("Nano Banana text model = %q, want %q", capturedConfig.TextModel, "custom-text-model") + } if fakeClient.searchOpts == nil { t.Fatal("expected search options to be captured") } @@ -319,6 +327,9 @@ func TestGenerateAudioUsesGeminiModelDefaultVoiceAndAttribution(t *testing.T) { if !strings.Contains(metadata, "audio_file=audio.wav") { t.Fatalf("gemini metadata missing fresh audio file reference: %q", metadata) } + if !strings.Contains(metadata, "voice=model-default") { + t.Fatalf("gemini metadata missing model-default voice marker: %q", metadata) + } if !strings.Contains(metadata, "format=wav") { t.Fatalf("gemini metadata missing format: %q", metadata) } -- cgit v1.2.3