功能：扩展云管资源运维、账号授权及监控管理

platform/internal/server/server_test.go

@@ -24,7 +24,8 @@ type fixture struct {
 	t      *testing.T
 }
 
-func newFixture(t *testing.T) *fixture {
+func newFixture(t *testing.T) *fixture { return newFixtureWorker(t, true) }
+func newFixtureWorker(t *testing.T, worker bool) *fixture {
 	t.Helper()
 	url := os.Getenv("VDI_TEST_DATABASE_URL")
 	if url == "" {
@@ -35,7 +36,7 @@ func newFixture(t *testing.T) *fixture {
 	if err != nil {
 		t.Fatal(err)
 	}
-	if _, err = db.Exec(`TRUNCATE connection_tickets,audit_logs,tasks,bindings,users,terminals,desktops,resources,admin_sessions,administrators CASCADE`); err != nil {
+	if _, err = db.Exec(`TRUNCATE terminal_metric_samples,terminal_metrics_current,power_batch_items,power_batches,snapshot_jobs,hardware_jobs,pve_connection,login_limits,alert_states,alerts,alert_rules,resource_samples,session_history,import_batches,terminal_policy_deliveries,terminal_policy_deployments,terminal_policy_templates,provisioning_jobs,desktop_pools,image_templates,connection_tickets,audit_logs,tasks,bindings,users,terminals,desktops,resources,admin_sessions,administrators CASCADE`); err != nil {
 		t.Fatal(err)
 	}
 	if err = store.Bootstrap(ctx, db, "demo", "", ""); err != nil {
@@ -43,8 +44,14 @@ func newFixture(t *testing.T) *fixture {
 	}
 	s := New(config.Config{Mode: "demo", JWTSecret: strings.Repeat("test-", 10)}, db, nil)
 	ctx, cancel := context.WithCancel(ctx)
-	go s.RunWorker(ctx)
-	t.Cleanup(func() { cancel(); time.Sleep(30 * time.Millisecond); db.Close() })
+	workerDone := make(chan struct{})
+	go func() {
+		defer close(workerDone)
+		if worker {
+			s.RunWorker(ctx)
+		}
+	}()
+	t.Cleanup(func() { cancel(); <-workerDone; db.Close() })
 	return &fixture{s, db, t}
 }
 func (f *fixture) request(method, path, token string, body any, key string) (int, map[string]any) {
@@ -166,6 +173,10 @@ func TestDeviceIdentityBindingTicketAndAcknowledgement(t *testing.T) {
 		t.Fatal("missing allowed desktops")
 	}
 	f.db.Exec("UPDATE desktops SET endpoint='192.0.2.100' WHERE id='desk_demo_windows'")
+	f.db.Exec("UPDATE desktops SET setup_required=true WHERE id='desk_demo_windows'")
+	status, out = f.request("POST", "/terminal/connect-ticket", device, map[string]any{"desktopId": "desk_demo_windows"}, "")
+	expect(t, 409, status, out)
+	f.db.Exec("UPDATE desktops SET setup_required=false WHERE id='desk_demo_windows'")
 	status, out = f.request("POST", "/terminal/connect-ticket", device, map[string]any{"desktopId": "desk_demo_windows"}, "")
 	expect(t, 201, status, out)
 	ticket := out["ticket"].(string)