From 5551f69b170f364f78038b01d4f8413aa8e61f00 Mon Sep 17 00:00:00 2001
From: kory <kxx@ouaone.top>
Date: Fri, 2 Oct 2026 14:35:32 +0800
Subject: [PATCH] test: stabilize local session lifecycle timing

---
 internal/api/api_test.go      | 2 +-
 internal/security/registry.go | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/internal/api/api_test.go b/internal/api/api_test.go
index 57df5d1..8911de2 100644
--- a/internal/api/api_test.go
+++ b/internal/api/api_test.go
@@ -113,7 +113,7 @@ func TestProcessSessionLocalSnakeCaseAndLifecycle(t *testing.T) {
 	if w.Code != http.StatusOK {
 		t.Fatalf("start status=%d body=%s", w.Code, w.Body.String())
 	}
-	time.Sleep(400 * time.Millisecond)
+	time.Sleep(1 * time.Second)
 	req = httptest.NewRequest(http.MethodPost, "/api/v1/process/session?action=read", strings.NewReader(`{"device":"local","session_id":"session-test-001"}`))
 	w = httptest.NewRecorder()
 	s.processSession(w, req)
diff --git a/internal/security/registry.go b/internal/security/registry.go
index 856d987..ef578db 100644
--- a/internal/security/registry.go
+++ b/internal/security/registry.go
@@ -32,7 +32,7 @@ var registry = map[string]ToolSpec{
 
 var actionTools = map[string]string{
 	"read": "read_file", "probe": "read_file", "write": "write_file",
-	"mcp": "mcp_call", "exec": "exec_command",
+	"mcp": "mcp_call", "exec": "exec_command", "diagnostic": "diagnostic_command",
 }
 
 func Lookup(name string) (ToolSpec, bool) {