usememos/memos
1
1
Fork 0
mirror of https://github.com/usememos/memos.git synced 2025-12-13 00:27:44 +08:00
Code Issues Projects Releases Packages Wiki Activity

test: update AI plugin tests for new configuration structure

Browse source 
- Add Enabled field to all Config test objects
- Update timeout expectations to reflect 10s default
- Change API key test expectation to support optional keys for local services
- Fix all LoadConfigFromEnv test cases for new field structure
- Ensure all Chat and Client tests use properly configured objects

Signed-off-by: ChaoLiu <chaoliu719@gmail.com>
This commit is contained in:
ChaoLiu 2025-08-22 12:15:28 +08:00
parent 9db3dcf0a2
commit 9cfdacc387
1 changed files with 29 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
plugin/ai/client_test.go
View file

@ -24,18 +24,22 @@ func TestLoadConfigFromEnv(t *testing.T) {
"AI_MODEL": "gpt-4o",
},
expected: &Config{
BaseURL: "https://api.openai.com/v1",
APIKey: "sk-test123",
Model: "gpt-4o",
Enabled: true,
BaseURL: "https://api.openai.com/v1",
APIKey: "sk-test123",
Model: "gpt-4o",
TimeoutSeconds: 10,
},
},
{
name: "no environment variables set",
envVars: map[string]string{},
expected: &Config{
BaseURL: "",
APIKey: "",
Model: "",
Enabled: false,
BaseURL: "",
APIKey: "",
Model: "",
TimeoutSeconds: 10,
},
},
{
@ -45,9 +49,11 @@ func TestLoadConfigFromEnv(t *testing.T) {
"AI_API_KEY": "sk-custom123",
},
expected: &Config{
BaseURL: "https://custom.api.com/v1",
APIKey: "sk-custom123",
Model: "",
Enabled: false,
BaseURL: "https://custom.api.com/v1",
APIKey: "sk-custom123",
Model: "",
TimeoutSeconds: 10,
},
},
}
@ -100,6 +106,7 @@ func TestConfig_IsConfigured(t *testing.T) {
{
name: "fully configured",
config: &Config{
Enabled: true,
BaseURL: "https://api.openai.com/v1",
APIKey: "sk-test123",
Model: "gpt-4o",
@ -109,6 +116,7 @@ func TestConfig_IsConfigured(t *testing.T) {
{
name: "missing base URL",
config: &Config{
Enabled: true,
BaseURL: "",
APIKey: "sk-test123",
Model: "gpt-4o",
@ -118,15 +126,17 @@ func TestConfig_IsConfigured(t *testing.T) {
{
name: "missing API key",
config: &Config{
Enabled: true,
BaseURL: "https://api.openai.com/v1",
APIKey: "",
Model: "gpt-4o",
},
expected: false,
expected: true,
},
{
name: "missing model",
config: &Config{
Enabled: true,
BaseURL: "https://api.openai.com/v1",
APIKey: "sk-test123",
Model: "",
@ -136,6 +146,7 @@ func TestConfig_IsConfigured(t *testing.T) {
{
name: "all fields empty",
config: &Config{
Enabled: false,
BaseURL: "",
APIKey: "",
Model: "",
@ -161,6 +172,7 @@ func TestNewClient(t *testing.T) {
{
name: "standard OpenAI configuration",
config: &Config{
Enabled: true,
BaseURL: "https://api.openai.com/v1",
APIKey: "sk-test123",
Model: "gpt-4o",
@ -170,6 +182,7 @@ func TestNewClient(t *testing.T) {
{
name: "custom endpoint configuration",
config: &Config{
Enabled: true,
BaseURL: "https://custom.api.com/v1",
APIKey: "sk-custom123",
Model: "gpt-3.5-turbo",
@ -179,6 +192,7 @@ func TestNewClient(t *testing.T) {
{
name: "incomplete configuration",
config: &Config{
Enabled: false,
BaseURL: "",
APIKey: "sk-test123",
Model: "gpt-4o",
@ -193,11 +207,12 @@ func TestNewClient(t *testing.T) {
{
name: "missing API key",
config: &Config{
Enabled: true,
BaseURL: "https://api.openai.com/v1",
APIKey: "",
Model: "gpt-4o",
},
expectErr: true,
expectErr: false,
},
}
@ -221,6 +236,7 @@ func TestNewClient(t *testing.T) {
func TestClient_Chat_RequestDefaults(t *testing.T) {
// This test verifies that default values are properly set
config := &Config{
Enabled: true,
BaseURL: "https://api.openai.com/v1",
APIKey: "sk-test123",
Model: "gpt-4o",
@ -320,6 +336,7 @@ func TestClient_Chat_Integration(t *testing.T) {
func TestClient_Chat_Validation(t *testing.T) {
config := &Config{
Enabled: true,
BaseURL: "https://api.openai.com/v1",
APIKey: "sk-test123",
Model: "gpt-4o",
@ -406,6 +423,7 @@ func TestClient_Chat_Validation(t *testing.T) {
func TestClient_Chat_ErrorTypes(t *testing.T) {
config := &Config{
Enabled: true,
BaseURL: "https://api.openai.com/v1",
APIKey: "sk-test123",
Model: "gpt-4o",