79 lines
2.5 KiB
Go
79 lines
2.5 KiB
Go
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)
|
||
}
|
||
}
|