summaryrefslogtreecommitdiff
path: root/internal/image/nanobanana_test.go
diff options
context:
space:
mode:
authorPaul Buetow <paul@buetow.org>2026-04-01 15:23:35 +0300
committerPaul Buetow <paul@buetow.org>2026-04-01 15:23:35 +0300
commitba04917670aac3fb4dff524fcdf614f13bbbfebe (patch)
treeb6f178c5404b29687aad00383aec3ccd6d976bff /internal/image/nanobanana_test.go
parent0cdc2fb5e732323847d683282e39e4d3cf494daf (diff)
z6: ignore whitespace translation input
Diffstat (limited to 'internal/image/nanobanana_test.go')
-rw-r--r--internal/image/nanobanana_test.go33
1 files changed, 33 insertions, 0 deletions
diff --git a/internal/image/nanobanana_test.go b/internal/image/nanobanana_test.go
index 84bb57b..bf0de8c 100644
--- a/internal/image/nanobanana_test.go
+++ b/internal/image/nanobanana_test.go
@@ -359,6 +359,39 @@ func TestNanoBananaClient_Search_CustomPromptPreservesTranslationMetadata(t *tes
}
}
+func TestNanoBananaClient_Search_CustomPromptIgnoresWhitespaceTranslation(t *testing.T) {
+ originalText := nanoBananaGenerateText
+ originalImage := nanoBananaGenerateImage
+ t.Cleanup(func() {
+ nanoBananaGenerateText = originalText
+ nanoBananaGenerateImage = originalImage
+ })
+
+ nanoBananaGenerateText = func(_ context.Context, _ *NanoBananaClient, _, _, _ string, _ float32, _ int32) (string, error) {
+ t.Fatal("unexpected text generation for custom prompt")
+ return "", nil
+ }
+ nanoBananaGenerateImage = func(_ context.Context, _ *NanoBananaClient, _ string) ([]byte, string, error) {
+ return mustPNGBytes(t), "image/png", nil
+ }
+
+ client := NewNanoBananaClient(&NanoBananaConfig{APIKey: "test-key"})
+ results, err := client.Search(context.Background(), &SearchOptions{
+ Query: "ябълка",
+ Translation: " ",
+ CustomPrompt: "flashcard prompt",
+ })
+ if err != nil {
+ t.Fatalf("Search() unexpected error: %v", err)
+ }
+ if len(results) != 1 {
+ t.Fatalf("expected 1 result, got %d", len(results))
+ }
+ if strings.Contains(results[0].Description, "(") || strings.Contains(results[0].Description, ")") {
+ t.Fatalf("result description = %q, want no translation metadata for whitespace-only input", results[0].Description)
+ }
+}
+
func TestNanoBananaClient_Search_InvalidOptions(t *testing.T) {
t.Parallel()