diff options
Diffstat (limited to 'internal/processor/processor_test.go')
| -rw-r--r-- | internal/processor/processor_test.go | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/internal/processor/processor_test.go b/internal/processor/processor_test.go index d0a0070..ad71263 100644 --- a/internal/processor/processor_test.go +++ b/internal/processor/processor_test.go @@ -6,6 +6,7 @@ import ( "testing" "codeberg.org/snonux/totalrecall/internal/cli" + "codeberg.org/snonux/totalrecall/internal/phonetic" "github.com/spf13/viper" ) @@ -44,6 +45,45 @@ func TestNewProcessor(t *testing.T) { } } +func TestNewProcessor_DefaultPhoneticProviderUsesOpenAI(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) + + if got := p.phoneticFetcher.Provider(); got != phonetic.ProviderOpenAI { + t.Fatalf("expected default phonetic provider %q, got %q", phonetic.ProviderOpenAI, got) + } +} + +func TestNewProcessor_ExplicitGeminiPhoneticProvider(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("phonetic.provider", "gemini") + + flags := cli.NewFlags() + p := NewProcessor(flags) + + if got := p.phoneticFetcher.Provider(); got != phonetic.ProviderGemini { + t.Fatalf("expected gemini phonetic provider %q, got %q", phonetic.ProviderGemini, got) + } +} + func TestNewProcessor_DefaultTranslationProviderUsesOpenAI(t *testing.T) { t.Setenv("OPENAI_API_KEY", "") t.Setenv("GOOGLE_API_KEY", "") |