1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
|
package image
import (
"context"
"os"
"testing"
)
func TestOpenAIClient_NewClient(t *testing.T) {
tests := []struct {
name string
config *OpenAIConfig
wantNil bool
}{
{
name: "with API key",
config: &OpenAIConfig{
APIKey: "test-key",
Model: "dall-e-2",
Size: "512x512",
},
wantNil: false,
},
{
name: "without API key",
config: &OpenAIConfig{
APIKey: "",
},
wantNil: false, // Client is created but will fail on operations
},
{
name: "with defaults",
config: &OpenAIConfig{
APIKey: "test-key",
},
wantNil: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
client := NewOpenAIClient(tt.config)
if (client == nil) != tt.wantNil {
t.Errorf("NewOpenAIClient() returned nil = %v, want %v", client == nil, tt.wantNil)
}
if client != nil && tt.config.APIKey != "" {
// Check defaults were set
if tt.config.Model == "" && client.model != "dall-e-2" {
t.Errorf("Expected default model dall-e-2, got %s", client.model)
}
if tt.config.Size == "" && client.size != "512x512" {
t.Errorf("Expected default size 512x512, got %s", client.size)
}
}
})
}
}
func TestOpenAIClient_createEducationalPrompt(t *testing.T) {
// Skip this test as it requires a valid OpenAI client
t.Skip("Skipping test that requires OpenAI client")
}
// translateBulgarianToEnglish test removed - now uses OpenAI API
func TestOpenAIClient_getSizeWidthHeight(t *testing.T) {
tests := []struct {
size string
width int
height int
}{
{"256x256", 256, 256},
{"512x512", 512, 512},
{"1024x1024", 1024, 1024},
{"1024x1792", 1024, 1024}, // Non-square sizes default to 1024x1024
{"1792x1024", 1024, 1024}, // Non-square sizes default to 1024x1024
{"unknown", 1024, 1024}, // Default is 1024x1024
}
for _, tt := range tests {
t.Run(tt.size, func(t *testing.T) {
client := &OpenAIClient{size: tt.size}
if w := client.getSizeWidth(); w != tt.width {
t.Errorf("getSizeWidth() = %d, want %d", w, tt.width)
}
if h := client.getSizeHeight(); h != tt.height {
t.Errorf("getSizeHeight() = %d, want %d", h, tt.height)
}
})
}
}
func TestOpenAIClient_Search_NoAPIKey(t *testing.T) {
client := NewOpenAIClient(&OpenAIConfig{})
opts := DefaultSearchOptions("ябълка")
_, err := client.Search(context.Background(), opts)
if err == nil {
t.Error("Expected error for missing API key")
}
if searchErr, ok := err.(*SearchError); ok {
if searchErr.Code != "NO_API_KEY" {
t.Errorf("Expected NO_API_KEY error, got %s", searchErr.Code)
}
} else {
t.Error("Expected SearchError type")
}
}
func TestOpenAIClient_Name(t *testing.T) {
client := &OpenAIClient{}
if name := client.Name(); name != "openai" {
t.Errorf("Name() = %s, want 'openai'", name)
}
}
func TestOpenAIClient_GetAttribution(t *testing.T) {
client := &OpenAIClient{}
result := &SearchResult{}
attr := client.GetAttribution(result)
if !contains(attr, "OpenAI DALL-E") {
t.Errorf("Attribution doesn't mention OpenAI DALL-E: %s", attr)
}
}
// Helper function
func contains(s, substr string) bool {
return len(s) >= len(substr) &&
(s == substr || len(s) > 0 && containsHelper(s, substr))
}
func containsHelper(s, substr string) bool {
for i := 0; i <= len(s)-len(substr); i++ {
if s[i:i+len(substr)] == substr {
return true
}
}
return false
}
// Integration test (skipped by default)
func TestOpenAIClient_Search_Integration(t *testing.T) {
apiKey := os.Getenv("OPENAI_API_KEY")
if apiKey == "" {
t.Skip("OPENAI_API_KEY not set, skipping integration test")
}
client := NewOpenAIClient(&OpenAIConfig{
APIKey: apiKey,
Model: "dall-e-2",
Size: "256x256", // Smallest size to minimize cost
})
opts := DefaultSearchOptions("ябълка")
results, err := client.Search(context.Background(), opts)
if err != nil {
t.Fatalf("Search failed: %v", err)
}
if len(results) != 1 {
t.Fatalf("Expected 1 result, got %d", len(results))
}
result := results[0]
// Check result fields
if result.ID == "" {
t.Error("Result ID is empty")
}
if result.URL == "" {
t.Error("Result URL is empty")
}
if result.Width != 256 || result.Height != 256 {
t.Errorf("Expected 256x256, got %dx%d", result.Width, result.Height)
}
if result.Source != "openai" {
t.Errorf("Expected source 'openai', got '%s'", result.Source)
}
// Test caching - second request should use cache
results2, err := client.Search(context.Background(), opts)
if err != nil {
t.Fatalf("Second search failed: %v", err)
}
if results2[0].URL != results[0].URL {
t.Log("Note: URLs differ, cache might not be working as expected")
}
}
|
