magic-agent 2.4.3 → 2.6.0

package/internal/client/restart_test.go

@@ -0,0 +1,151 @@
+package client
+
+import (
+	"encoding/json"
+	"io"
+	"log"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+)
+
+/*
+重启链路（2026-10-25 用户：「点击重启服务时标志一下 然后定时器触发一下」）：
+
+	先立标志（重复点击不再起第二个助手）→ 先答 200 → 定时器到点才拉助手。
+	助手本身用假动作替身，不碰真进程。
+*/
+func TestHandleRestartFlagsThenTimer(t *testing.T) {
+	dir := t.TempDir()
+	statePath := filepath.Join(dir, "client.json")
+	t.Setenv(EnvState, statePath)
+	exe, err := os.Executable()
+	if err != nil {
+		t.Fatal(err)
+	}
+	// 状态文件指向**本进程**：过 handleRestart 那道「不许重启别人」的闸。
+	if err := writeDaemonState(statePath, &DaemonState{
+		PID: os.Getpid(), Port: 12345, Token: "tok", Exe: exe, Lan: true,
+	}); err != nil {
+		t.Fatal(err)
+	}
+
+	// 假助手：只记录被拉起过几次、带的是什么可执行文件。
+	spawned := make(chan string, 4)
+	oldSpawner := restartSpawner
+	restartSpawner = func(gotExe string, _ *log.Logger) { spawned <- gotExe }
+	defer func() { restartSpawner = oldSpawner }()
+	// 标志是进程级的：用例前后都重置，别把别的用例立过的 pending 带进来。
+	resetRestartPendingForTest()
+	defer resetRestartPendingForTest()
+	// 定时器等短一点：用例不真等一秒（仍要足够长，长过「返回即查」那一瞬）。
+	oldDelay := restartDelay
+	restartDelay = 100 * time.Millisecond
+	defer func() { restartDelay = oldDelay }()
+
+	s := &Service{log: log.New(io.Discard, "", 0)}
+
+	// 第一次点击：200 + restarting，且**定时器到点前**助手不许已被拉起
+	// （响应落地必须严格早于助手起身）。
+	rec := httptest.NewRecorder()
+	s.handleRestart(rec, httptest.NewRequest(http.MethodPost, "/desk/restart", nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("POST /desk/restart = %d，want 200（body: %s）", rec.Code, rec.Body.String())
+	}
+	var body map[string]any
+	if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
+		t.Fatalf("解析响应失败: %v（%s）", err, rec.Body.String())
+	}
+	if body["restarting"] != true {
+		t.Errorf("响应应带 restarting=true，实际 %v", body["restarting"])
+	}
+	select {
+	case <-spawned:
+		t.Fatal("定时器到点前就拉了助手：响应可能还没落地")
+	default:
+	}
+
+	// 重复点击：认账（already_pending）但不再排第二个定时器。
+	rec = httptest.NewRecorder()
+	s.handleRestart(rec, httptest.NewRequest(http.MethodPost, "/desk/restart", nil))
+	if rec.Code != http.StatusOK {
+		t.Fatalf("重复点击 = %d，want 200", rec.Code)
+	}
+	body = nil
+	if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
+		t.Fatalf("解析重复点击响应失败: %v", err)
+	}
+	if body["already_pending"] != true {
+		t.Errorf("重复点击应答 already_pending=true，实际 %v", body["already_pending"])
+	}
+
+	// 定时器到点：助手被拉起一次，带的是本进程的可执行文件。
+	select {
+	case got := <-spawned:
+		if got != exe {
+			t.Errorf("助手可执行文件 = %q，want %q", got, exe)
+		}
+	case <-time.After(2 * time.Second):
+		t.Fatal("定时器没有拉起助手")
+	}
+	// 只有一次：重复点击没有多起助手（两个助手会互相停掉对方刚拉起的实例）。
+	select {
+	case <-spawned:
+		t.Fatal("重复点击多拉起了一个助手")
+	default:
+	}
+}
+
+/*
+/desk/health 要带出重启标志：手机据此把「旧实例认账」当成「它马上要下线」，
+
+	不必等连接真的断；再探到不带标志的 200 才是新一代。
+*/
+func TestHandleHealthReportsRestarting(t *testing.T) {
+	// 直接搭一个最小 Service（不走 New()：那个要 exec core 契约，Windows 上跑不了）。
+	// Token 留空 → guard 放行、publicURL 的探测直接短路；Storage.Dir 指临时目录，
+	// instanceID 不会去写用户目录。
+	prof := DefaultProfile()
+	prof.Storage.Dir = t.TempDir()
+	svc := &Service{
+		log:      log.New(io.Discard, "", 0),
+		contract: Contract{ContractVersion: 2, Version: "1.0.103"},
+		opts:     Options{Profile: prof},
+	}
+	restarting := func() bool {
+		rec := httptest.NewRecorder()
+		svc.Handler().ServeHTTP(rec, httptest.NewRequest(http.MethodGet, "/desk/health", nil))
+		if rec.Code != http.StatusOK {
+			t.Fatalf("health = %d，want 200（body: %s）", rec.Code, rec.Body.String())
+		}
+		var got struct {
+			Restarting bool `json:"restarting"`
+		}
+		if err := json.Unmarshal(rec.Body.Bytes(), &got); err != nil {
+			t.Fatalf("解析 health 失败: %v", err)
+		}
+		return got.Restarting
+	}
+
+	resetRestartPendingForTest()
+	defer resetRestartPendingForTest()
+	if restarting() {
+		t.Fatal("没受理重启时 health.restarting 应为 false")
+	}
+	restartMu.Lock()
+	restartPending = true
+	restartMu.Unlock()
+	if !restarting() {
+		t.Error("受理重启后 health.restarting 应为 true（旧实例认账，即将下线）")
+	}
+}
+
+// resetRestartPendingForTest 清掉重启标志（仅供测试：标志是进程级的，用例之间要重置）。
+func resetRestartPendingForTest() {
+	restartMu.Lock()
+	restartPending = false
+	restartMu.Unlock()
+}