From 5b88a045b29970a3cc057326b85eae58ab41fe02 Mon Sep 17 00:00:00 2001 From: Paul Buetow Date: Wed, 1 Apr 2026 13:07:41 +0300 Subject: zr: restore OpenAI default translation provider --- internal/processor/processor_test.go | 55 ++++++++++++++++++++++++++++++++++++ 1 file changed, 55 insertions(+) (limited to 'internal/processor/processor_test.go') diff --git a/internal/processor/processor_test.go b/internal/processor/processor_test.go index a9c9ab1..9e6b5fc 100644 --- a/internal/processor/processor_test.go +++ b/internal/processor/processor_test.go @@ -6,12 +6,20 @@ import ( "testing" "codeberg.org/snonux/totalrecall/internal/cli" + "github.com/spf13/viper" ) func TestNewProcessor(t *testing.T) { t.Setenv("OPENAI_API_KEY", "test-openai-key") t.Setenv("GOOGLE_API_KEY", "test-google-key") + originalConfig := viper.New() + *originalConfig = *viper.GetViper() + defer func() { + *viper.GetViper() = *originalConfig + }() + viper.Reset() + flags := cli.NewFlags() p := NewProcessor(flags) @@ -36,6 +44,53 @@ func TestNewProcessor(t *testing.T) { } } +func TestNewProcessor_DefaultTranslationProviderUsesOpenAI(t *testing.T) { + t.Setenv("OPENAI_API_KEY", "") + t.Setenv("GOOGLE_API_KEY", "") + + originalConfig := viper.New() + *originalConfig = *viper.GetViper() + defer func() { + *viper.GetViper() = *originalConfig + }() + viper.Reset() + + flags := cli.NewFlags() + p := NewProcessor(flags) + + _, err := p.translator.TranslateWord("ябълка") + if err == nil { + t.Fatal("Expected error for missing OpenAI API key") + } + if err.Error() != "OpenAI API key not found" { + t.Fatalf("Expected OpenAI default provider error, got: %v", err) + } +} + +func TestNewProcessor_ExplicitGeminiTranslationProvider(t *testing.T) { + t.Setenv("OPENAI_API_KEY", "") + t.Setenv("GOOGLE_API_KEY", "") + + originalConfig := viper.New() + *originalConfig = *viper.GetViper() + defer func() { + *viper.GetViper() = *originalConfig + }() + viper.Reset() + viper.Set("translation.provider", "gemini") + + flags := cli.NewFlags() + p := NewProcessor(flags) + + _, err := p.translator.TranslateWord("ябълка") + if err == nil { + t.Fatal("Expected error for missing Google API key") + } + if err.Error() != "Google API key not found" { + t.Fatalf("Expected Gemini provider error, got: %v", err) + } +} + func TestProcessSingleWord_InvalidWord(t *testing.T) { flags := cli.NewFlags() flags.OutputDir = t.TempDir() -- cgit v1.2.3