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
|
package audio
import (
"context"
"strings"
"testing"
)
func TestNewOpenAIProvider(t *testing.T) {
tests := []struct {
name string
config OpenAIAudioConfig
wantErr bool
errMsg string
}{
{
name: "missing API key",
config: OpenAIAudioConfig{Key: ""},
wantErr: true,
errMsg: "OpenAI API key is required",
},
{
name: "valid config",
config: OpenAIAudioConfig{Key: "test-key"},
wantErr: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
provider, err := NewOpenAIProvider(tt.config, "mp3")
if (err != nil) != tt.wantErr {
t.Errorf("NewOpenAIProvider() error = %v, wantErr %v", err, tt.wantErr)
}
if tt.wantErr && err != nil && err.Error() != tt.errMsg {
t.Errorf("NewOpenAIProvider() error = %v, want %v", err.Error(), tt.errMsg)
}
// Check provider properties
if !tt.wantErr && provider != nil {
if provider.Name() != "openai" {
t.Errorf("Name() = %v, want %v", provider.Name(), "openai")
}
}
})
}
}
func TestOpenAIProviderIsAvailable(t *testing.T) {
tests := []struct {
name string
config OpenAIAudioConfig
wantErr bool
}{
{
name: "with API key",
config: OpenAIAudioConfig{Key: "test-key"},
wantErr: false,
},
{
name: "without API key",
config: OpenAIAudioConfig{Key: ""},
wantErr: true,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
provider := &OpenAIProvider{config: tt.config}
err := provider.IsAvailable()
if (err != nil) != tt.wantErr {
t.Errorf("IsAvailable() error = %v, wantErr %v", err, tt.wantErr)
}
})
}
}
func TestPreprocessBulgarianText(t *testing.T) {
provider := &OpenAIProvider{}
tests := []struct {
name string
input string
expected string
}{
{
name: "simple word",
input: "ябълка",
expected: "ябълка",
},
{
name: "word with punctuation",
input: "ябълка!",
expected: "ябълка",
},
{
name: "word with multiple punctuation",
input: "\"ябълка?\"",
expected: "ябълка",
},
{
name: "word with spaces",
input: " ябълка ",
expected: "ябълка",
},
{
name: "word with dashes",
input: "ябълка-круша",
expected: "ябълкакруша",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
result := provider.preprocessBulgarianText(tt.input)
if result != tt.expected {
t.Errorf("preprocessBulgarianText(%q) = %q, want %q", tt.input, result, tt.expected)
}
})
}
}
func TestGenerateAudioValidation(t *testing.T) {
provider := &OpenAIProvider{config: OpenAIAudioConfig{Key: "test-key"}}
ctx := context.Background()
// Test with non-Bulgarian text
err := provider.GenerateAudio(ctx, "hello", "output.mp3")
if err == nil {
t.Error("Expected error for non-Bulgarian text")
}
if !strings.Contains(err.Error(), "must contain Cyrillic characters") {
t.Errorf("Expected Bulgarian validation error, got: %v", err)
}
// Test with empty text
err = provider.GenerateAudio(ctx, "", "output.mp3")
if err == nil {
t.Error("Expected error for empty text")
}
}
|
