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
|
package image
import (
"context"
"io"
"strings"
"testing"
)
// mockSearcher implements ImageSearcher for testing
type mockSearcher struct {
name string
searchResults []SearchResult
searchErr error
downloadErr error
}
func (m *mockSearcher) Search(ctx context.Context, opts *SearchOptions) ([]SearchResult, error) {
if m.searchErr != nil {
return nil, m.searchErr
}
return m.searchResults, nil
}
func (m *mockSearcher) Download(ctx context.Context, url string) (io.ReadCloser, error) {
if m.downloadErr != nil {
return nil, m.downloadErr
}
return io.NopCloser(strings.NewReader("mock image data")), nil
}
func (m *mockSearcher) GetAttribution(result *SearchResult) string {
return result.Attribution
}
func (m *mockSearcher) Name() string {
return m.name
}
func TestDefaultSearchOptions(t *testing.T) {
opts := DefaultSearchOptions("ябълка")
if opts.Query != "ябълка" {
t.Errorf("Expected query 'ябълка', got '%s'", opts.Query)
}
if opts.Language != "bg" {
t.Errorf("Expected language 'bg', got '%s'", opts.Language)
}
if !opts.SafeSearch {
t.Error("Expected SafeSearch to be true")
}
if opts.PerPage != 10 {
t.Errorf("Expected PerPage 10, got %d", opts.PerPage)
}
if opts.Page != 1 {
t.Errorf("Expected Page 1, got %d", opts.Page)
}
if opts.ImageType != "photo" {
t.Errorf("Expected ImageType 'photo', got '%s'", opts.ImageType)
}
}
func TestSearchError(t *testing.T) {
err := &SearchError{
Provider: "test",
Code: "404",
Message: "Not found",
}
expected := "test: Not found"
if err.Error() != expected {
t.Errorf("Expected error '%s', got '%s'", expected, err.Error())
}
}
func TestRateLimitError(t *testing.T) {
err := &RateLimitError{
Provider: "test",
RetryAfter: 60,
LimitPerHour: 100,
}
expected := "test: rate limit exceeded"
if err.Error() != expected {
t.Errorf("Expected error '%s', got '%s'", expected, err.Error())
}
}
func TestMockSearcher(t *testing.T) {
mockResults := []SearchResult{
{
ID: "1",
URL: "https://example.com/image1.jpg",
Width: 800,
Height: 600,
Description: "Test image",
Source: "mock",
},
}
searcher := &mockSearcher{
name: "mock",
searchResults: mockResults,
}
ctx := context.Background()
opts := DefaultSearchOptions("test")
results, err := searcher.Search(ctx, opts)
if err != nil {
t.Fatalf("Search() failed: %v", err)
}
if len(results) != 1 {
t.Fatalf("Expected 1 result, got %d", len(results))
}
if results[0].ID != "1" {
t.Errorf("Expected ID '1', got '%s'", results[0].ID)
}
}
func TestDownloadOptions(t *testing.T) {
opts := DefaultDownloadOptions()
if opts.OutputDir != "./images" {
t.Errorf("Expected output dir './images', got '%s'", opts.OutputDir)
}
if opts.OverwriteExisting {
t.Error("Expected OverwriteExisting to be false")
}
if !opts.CreateDir {
t.Error("Expected CreateDir to be true")
}
if opts.MaxSizeBytes != 10*1024*1024 {
t.Errorf("Expected MaxSizeBytes 10MB, got %d", opts.MaxSizeBytes)
}
}
|
