diff options
| author | Paul Buetow <paul@buetow.org> | 2026-04-01 13:00:40 +0300 |
|---|---|---|
| committer | Paul Buetow <paul@buetow.org> | 2026-04-01 13:00:40 +0300 |
| commit | b006582e4c7323821eb836080a9bc9ec24243a47 (patch) | |
| tree | a8e96ecdc74cd2a5bb19148d8a6a7cf90f6cbbb4 /internal/processor/processor_test.go | |
| parent | a574e5290261f9ea2cc39fd0a9a2ba756fc1f67a (diff) | |
zr: add Gemini-backed translation provider
Diffstat (limited to 'internal/processor/processor_test.go')
| -rw-r--r-- | internal/processor/processor_test.go | 21 |
1 files changed, 7 insertions, 14 deletions
diff --git a/internal/processor/processor_test.go b/internal/processor/processor_test.go index e835681..a9c9ab1 100644 --- a/internal/processor/processor_test.go +++ b/internal/processor/processor_test.go @@ -9,15 +9,8 @@ import ( ) func TestNewProcessor(t *testing.T) { - // Set up test environment - if err := os.Setenv("OPENAI_API_KEY", "test-key"); err != nil { - t.Fatalf("Failed to set OPENAI_API_KEY: %v", err) - } - defer func() { - if err := os.Unsetenv("OPENAI_API_KEY"); err != nil { - t.Errorf("Failed to unset OPENAI_API_KEY: %v", err) - } - }() + t.Setenv("OPENAI_API_KEY", "test-openai-key") + t.Setenv("GOOGLE_API_KEY", "test-google-key") flags := cli.NewFlags() p := NewProcessor(flags) @@ -63,8 +56,8 @@ func TestProcessSingleWord_InvalidWord(t *testing.T) { func TestProcessSingleWord_ValidWord(t *testing.T) { // Skip if no API key - if os.Getenv("OPENAI_API_KEY") == "" { - t.Skip("Skipping test: OPENAI_API_KEY not set") + if os.Getenv("OPENAI_API_KEY") == "" || os.Getenv("GOOGLE_API_KEY") == "" { + t.Skip("Skipping test: OPENAI_API_KEY and GOOGLE_API_KEY must be set") } flags := cli.NewFlags() @@ -114,9 +107,9 @@ func TestProcessBatch_ValidFile(t *testing.T) { flags.SkipImages = true p := NewProcessor(flags) - // Skip if no API key - if os.Getenv("OPENAI_API_KEY") == "" { - t.Skip("Skipping test: OPENAI_API_KEY not set") + // Skip if no API keys + if os.Getenv("OPENAI_API_KEY") == "" || os.Getenv("GOOGLE_API_KEY") == "" { + t.Skip("Skipping test: OPENAI_API_KEY and GOOGLE_API_KEY must be set") } err = p.ProcessBatch() |