test(agent): 异步部署的确定性测试(不阻塞 + Pending→running / →failed)

- 慢 AM mock 验证 startTemplateAgentAsync 立即返回(不阻塞 30s)。
- Pending 记录在 AM 响应后自动回填 running + runtime_id + subdomain。
- AM 失败时记录标 failed + failure_reason,不卡在 Pending。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
2026-06-08 16:48:29 +08:00
co-authored by Claude Opus 4.8
parent a1a5b6037b
commit da854277d3
@@ -0,0 +1,78 @@
package controller
import (
"net/http"
"net/http/httptest"
"testing"
"time"
"github.com/heicode/manager/model"
"github.com/stretchr/testify/require"
)
// startTemplateAgentAsync must NOT block the caller on the (slow) AM start, and
// must flip the pre-persisted Pending record to running once AM responds — this
// is the core of the 504 fix (deploy returns immediately; status lands later).
func TestStartTemplateAgentAsync_PendingToRunning(t *testing.T) {
setupResourceControllerTestDB(t)
require.NoError(t, model.DB.AutoMigrate(&model.AgentDeployment{}))
// Slow AM mock — simulates the ~30s start the handler must not block on.
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
time.Sleep(150 * time.Millisecond)
w.Header().Set("Content-Type", "application/json")
_, _ = w.Write([]byte(`{"success":true,"data":{"namespace":"rt-async-1","access_info":{"domain":"https://async.agents.example"},"status":"running"}}`))
}))
defer srv.Close()
t.Setenv("AGENT_RUNTIME_BASE_URL", srv.URL)
// What HeicodeDeployAgent persists before returning.
require.NoError(t, model.DB.Create(&model.AgentDeployment{
DeploymentID: "dep_async_run",
UserID: "22",
TemplateID: "architect",
Status: "Pending",
}).Error)
start := time.Now()
startTemplateAgentAsync("dep_async_run", 0, amStartArgs{ManagerDeploymentID: "dep_async_run", UserID: "22", TemplateKey: "architect"})
require.Less(t, time.Since(start), 50*time.Millisecond, "must not block on the AM start call")
var got model.AgentDeployment
require.Eventually(t, func() bool {
model.DB.Where("deployment_id = ?", "dep_async_run").First(&got)
return got.Status == "running"
}, 3*time.Second, 20*time.Millisecond, "Pending should flip to running after async AM start")
require.Equal(t, "rt-async-1", got.RuntimeDeploymentID)
require.Equal(t, "https://async.agents.example", got.Subdomain)
}
// On AM failure the record must be marked failed (with a reason), not left
// stuck on Pending — so the client surfaces a failure instead of a hang.
func TestStartTemplateAgentAsync_FailureMarksFailed(t *testing.T) {
setupResourceControllerTestDB(t)
require.NoError(t, model.DB.AutoMigrate(&model.AgentDeployment{}))
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusInternalServerError)
_, _ = w.Write([]byte(`{"success":false,"message":"AM boom"}`))
}))
defer srv.Close()
t.Setenv("AGENT_RUNTIME_BASE_URL", srv.URL)
require.NoError(t, model.DB.Create(&model.AgentDeployment{
DeploymentID: "dep_async_fail",
UserID: "22",
TemplateID: "architect",
Status: "Pending",
}).Error)
startTemplateAgentAsync("dep_async_fail", 0, amStartArgs{ManagerDeploymentID: "dep_async_fail", UserID: "22", TemplateKey: "architect"})
var got model.AgentDeployment
require.Eventually(t, func() bool {
model.DB.Where("deployment_id = ?", "dep_async_fail").First(&got)
return got.Status == "failed"
}, 3*time.Second, 20*time.Millisecond, "AM failure should mark the record failed")
require.NotEmpty(t, got.FailureReason)
}