chore: directory migration - gateway→server, web→client/electron

server/internal/repo/platform_owner_test.go

@@ -0,0 +1,78 @@
+package repo
+
+import (
+	"context"
+	"testing"
+)
+
+// platform_session_id 必须只发给归属方 —— 生产上 pi 的会话 id 被推给了抄送方 dsh，
+// DSH 在自己磁盘上找不到那个文件，按 N-8 抛错，邮件静默消失。
+func TestPlatformSessionFor_ReturnsOwnerFromMirror(t *testing.T) {
+	setupTestDB(t)
+	ctx := context.Background()
+	seedPlatformAgent(t, "pi")
+	seedPlatformAgent(t, "dsh")
+
+	// pi 上报一条平台会话
+	if err := ReplacePlatformSessions(ctx, "pi", []PlatformSession{
+		{PlatformID: "pid-pi-1", Workspace: "/w", Slug: "项目定位", Title: "项目定位"},
+	}); err != nil {
+		t.Fatalf("ReplacePlatformSessions: %v", err)
+	}
+
+	id, err := AdoptPlatformSession(ctx, "pi", "pid-pi-1", "项目定位", "/w", "项目定位")
+	if err != nil {
+		t.Fatalf("AdoptPlatformSession: %v", err)
+	}
+
+	pid, owner := PlatformSessionFor(ctx, id)
+	if pid != "pid-pi-1" {
+		t.Errorf("platformID = %q, want pid-pi-1", pid)
+	}
+	if owner != "pi" {
+		t.Errorf("owner = %q, want pi（镜像里 agent_name=pi）", owner)
+	}
+}
+
+// 未接管的普通会话不该返回任何 platform id。
+func TestPlatformSessionFor_PlainSession(t *testing.T) {
+	setupTestDB(t)
+	ctx := context.Background()
+	seedAgent(t, "dsh", 20)
+	id, err := CreateSession(ctx, nil, "dsh", "普通会话", "/w")
+	if err != nil {
+		t.Fatalf("CreateSession: %v", err)
+	}
+	pid, owner := PlatformSessionFor(ctx, id)
+	if pid != "" || owner != "" {
+		t.Errorf("got (%q,%q), want ('','')", pid, owner)
+	}
+}
+
+// 镜像那行被整表替换掉（平台侧删了会话）时退回 sessions.from_agent，
+// 而不是让 owner 变空 —— 变空会让归属方也收不到 platform_session_id。
+func TestPlatformSessionFor_MirrorGoneFallsBackToFromAgent(t *testing.T) {
+	setupTestDB(t)
+	ctx := context.Background()
+	seedPlatformAgent(t, "pi")
+	if err := ReplacePlatformSessions(ctx, "pi", []PlatformSession{
+		{PlatformID: "pid-pi-2", Workspace: "/w", Slug: "s2", Title: "t2"},
+	}); err != nil {
+		t.Fatalf("ReplacePlatformSessions: %v", err)
+	}
+	id, err := AdoptPlatformSession(ctx, "pi", "pid-pi-2", "s2", "/w", "t2")
+	if err != nil {
+		t.Fatalf("AdoptPlatformSession: %v", err)
+	}
+	// 平台侧删了这条会话 → 心跳整表替换成空
+	if err := ReplacePlatformSessions(ctx, "pi", []PlatformSession{}); err != nil {
+		t.Fatalf("ReplacePlatformSessions(empty): %v", err)
+	}
+	pid, owner := PlatformSessionFor(ctx, id)
+	if pid != "pid-pi-2" {
+		t.Errorf("platformID = %q, want pid-pi-2", pid)
+	}
+	if owner != "pi" {
+		t.Errorf("owner = %q, want pi（退回 sessions.from_agent）", owner)
+	}
+}