From ee9a76889a2deb1925c6e4ccdbc4b435fc9ae154 Mon Sep 17 00:00:00 2001
From: afjcjsbx <afjcjsbx@gmail.com>
Date: Tue, 7 Apr 2026 23:21:30 +0200
Subject: [PATCH] fix test

---
 pkg/config/config_test.go               | 32 ++++---------------------
 pkg/config/security_integration_test.go |  7 +++---
 2 files changed, 8 insertions(+), 31 deletions(-)

diff --git a/pkg/config/config_test.go b/pkg/config/config_test.go
index 01ef76578..3a4f69be0 100644
--- a/pkg/config/config_test.go
+++ b/pkg/config/config_test.go
@@ -1084,27 +1084,18 @@ func TestLoadConfig_TelegramPlaceholderTextAcceptsSingleString(t *testing.T) {
 	dir := t.TempDir()
 	cfgPath := filepath.Join(dir, "config.json")
 	data := `{
-		"version": 1,
-		"agents": { "defaults": { "workspace": "", "model": "", "max_tokens": 0, "max_tool_iterations": 0 } },
-		"bindings": [],
-		"session": {},
+		"version": 2,
 		"channels": {
 			"telegram": {
 				"enabled": true,
-				"bot_token": "",
+				"token": "",
 				"allow_from": [],
 				"placeholder": {
 					"enabled": true,
 					"text": "Thinking..."
 				}
 			}
-		},
-		"model_list": [],
-		"gateway": {},
-		"tools": {},
-		"heartbeat": {},
-		"devices": {},
-		"voice": {}
+		}
 	}`
 	if err := os.WriteFile(cfgPath, []byte(data), 0o600); err != nil {
 		t.Fatalf("setup: %v", err)
@@ -1120,25 +1111,12 @@ func TestLoadConfig_TelegramPlaceholderTextAcceptsSingleString(t *testing.T) {
 }
 
 // TestLoadConfig_WarnsForPlaintextAPIKey verifies that LoadConfig resolves a plaintext
-// api_key into memory but does NOT rewrite the config file. File writes are the sole
+// api_keys entry into memory but does NOT rewrite the config file. File writes are the sole
 // responsibility of SaveConfig.
 func TestLoadConfig_WarnsForPlaintextAPIKey(t *testing.T) {
 	dir := t.TempDir()
 	cfgPath := filepath.Join(dir, "config.json")
-	const original = `{"version":1,"model_list":[{"model_name":"test","model":"openai/gpt-4","api_key":"sk-plaintext"}]}`
-	if err := os.WriteFile(cfgPath, []byte(original), 0o600); err != nil {
-		t.Fatalf("setup: %v", err)
-	}
-	secPath := filepath.Join(dir, SecurityConfigFile)
-	const securityConfig = `
-model_list:
-  test:0:
-    api_keys:
-      - "sk-plaintext"
-`
-	if err := os.WriteFile(secPath, []byte(securityConfig), 0o600); err != nil {
-		t.Fatalf("setup: %v", err)
-	}
+	const original = `{"version":2,"model_list":[{"model_name":"test","model":"openai/gpt-4","api_keys":["sk-plaintext"]}]}`
 	if err := os.WriteFile(cfgPath, []byte(original), 0o600); err != nil {
 		t.Fatalf("setup: %v", err)
 	}
diff --git a/pkg/config/security_integration_test.go b/pkg/config/security_integration_test.go
index 6ca8637f4..83ec2b24d 100644
--- a/pkg/config/security_integration_test.go
+++ b/pkg/config/security_integration_test.go
@@ -43,17 +43,16 @@ func TestSecurityConfigIntegration(t *testing.T) {
 	t.Run("Full workflow with security references", func(t *testing.T) {
 		tmpDir := t.TempDir()
 
-		// Create config.json with direct security values (not ref: references)
-		// These values should take precedence over .security.yml
+		// Create config.json with direct security values using the current schema.
 		configPath := filepath.Join(tmpDir, "config.json")
 		configContent := `{
-  "version": 1,
+  "version": 2,
   "model_list": [
     {
       "model_name": "test-model",
       "model": "openai/test-model",
       "api_base": "https://api.openai.com/v1",
-      "api_key": "sk-from-config-json-direct"
+      "api_keys": ["sk-from-config-json-direct"]
     }
   ],
   "channels": {