weichenandCursor
79797cab7c
ListWorkflowDocumentsByProject now projects author/reviewer/approver id+name so clients cannot invent the current user as author. Co-authored-by: Cursor <cursoragent@cursor.com>
82 lines
2.4 KiB
Go
82 lines
2.4 KiB
Go
package docgov_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"chat-agent-system/internal/docgov"
|
|
)
|
|
|
|
func TestListWorkflowDocumentsIncludesActiveLease(t *testing.T) {
|
|
engine := docgov.NewGovernanceEngine(nil)
|
|
ctx := context.Background()
|
|
const docID = "doc_project_docs_lease"
|
|
if _, err := engine.CreateOrGetDocument(
|
|
ctx, docID, 143, "intake", "用户初始需求收集表", "V0.1", "",
|
|
9, "编制人", 2, "审核人", 3, "批准人",
|
|
); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := engine.AcquireLease(ctx, docgov.LeaseAcquireRequest{
|
|
DocID: docID, HolderType: docgov.LeaseHolderSquad, HolderID: "run-visible", TTL: time.Minute,
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
docs, err := engine.ListWorkflowDocumentsByProject(ctx, 143)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(docs) != 1 {
|
|
t.Fatalf("documents=%d", len(docs))
|
|
}
|
|
if docs[0].LeaseHolderType != docgov.LeaseHolderSquad || docs[0].LeaseHolderID != "run-visible" {
|
|
t.Fatalf("lease not projected: %+v", docs[0])
|
|
}
|
|
if docs[0].LeaseExpiresAt == nil || docs[0].LeaseHeartbeatAt == nil {
|
|
t.Fatalf("lease timestamps missing: %+v", docs[0])
|
|
}
|
|
}
|
|
|
|
func TestListWorkflowDocumentsIncludesAuthoritativeDuties(t *testing.T) {
|
|
engine := docgov.NewGovernanceEngine(nil)
|
|
ctx := context.Background()
|
|
const (
|
|
docID = "doc_proj143_intake"
|
|
authorID = int64(9)
|
|
reviewerID = int64(2)
|
|
approverID = int64(3)
|
|
)
|
|
if _, err := engine.CreateOrGetDocument(
|
|
ctx, docID, 143, "intake", "用户初始需求收集表", "V0.1", "",
|
|
authorID, "编制人A", reviewerID, "审核人B", approverID, "批准人C",
|
|
); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
docs, err := engine.ListWorkflowDocumentsByProject(ctx, 143)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(docs) != 1 {
|
|
t.Fatalf("documents=%d", len(docs))
|
|
}
|
|
got := docs[0]
|
|
if got.AuthorID != authorID || got.AuthorName != "编制人A" {
|
|
t.Fatalf("author not projected: id=%d name=%q", got.AuthorID, got.AuthorName)
|
|
}
|
|
if got.ReviewerID != reviewerID || got.ReviewerName != "审核人B" {
|
|
t.Fatalf("reviewer not projected: id=%d name=%q", got.ReviewerID, got.ReviewerName)
|
|
}
|
|
if got.ApproverID != approverID || got.ApproverName != "批准人C" {
|
|
t.Fatalf("approver not projected: id=%d name=%q", got.ApproverID, got.ApproverName)
|
|
}
|
|
if got.AuthorID == reviewerID {
|
|
t.Fatalf("reviewer B must not become author: %+v", got)
|
|
}
|
|
if got.AuthorID == 99 {
|
|
t.Fatalf("stranger must not become author: %+v", got)
|
|
}
|
|
}
|