Files
weichen 77b8d07cd7 fix(governance): 统一文档访问判定与上下文快照排除计数
补齐 DocumentAccessDecider HTTP 409、projectcontext 私有对话排除统计,并更新整改 gap matrix 与测试报告。
2026-09-13 02:50:23 +00:00

81 lines
2.2 KiB
Go

package projectinstance
import (
"context"
"errors"
"sync"
"testing"
)
func TestActivationToken_OneTimeUse(t *testing.T) {
ctx := context.Background()
store := newMemStore()
_ = SeedProjectMeta(store, 200, LifecyclePlanning, "", 1, 0)
svc := NewService(store, testResolver(), nil, nil)
_, _ = svc.SaveWorkflowPlan(ctx, 200, 1, true, SavePlanInput{
Plan: validPlan(), ExpectedPlanRevision: 0, ExpectedSelectionVersion: 1,
})
preview, err := svc.ActivationPreview(ctx, 200, 1, true)
if err != nil {
t.Fatalf("ActivationPreview: %v", err)
}
_, err = svc.Activate(ctx, 200, 1, true, ActivateInput{
ConfirmationToken: preview.ConfirmationToken,
ExpectedProjectVersion: preview.ExpectedProjectVersion,
ExpectedPlanRevision: preview.ExpectedPlanRevision,
})
if err != nil {
t.Fatalf("first activate: %v", err)
}
_, err = svc.Activate(ctx, 200, 1, true, ActivateInput{
ConfirmationToken: preview.ConfirmationToken,
ExpectedProjectVersion: preview.ExpectedProjectVersion,
ExpectedPlanRevision: preview.ExpectedPlanRevision,
})
if !errors.Is(err, ErrTokenInvalid) && !errors.Is(err, ErrPlanFrozen) {
t.Fatalf("expected ErrTokenInvalid or ErrPlanFrozen on reuse, got %v", err)
}
}
func TestActivationToken_ConcurrentActivateOnlyOneSucceeds(t *testing.T) {
ctx := context.Background()
store := newMemStore()
_ = SeedProjectMeta(store, 201, LifecyclePlanning, "", 1, 0)
svc := NewService(store, testResolver(), nil, nil)
_, _ = svc.SaveWorkflowPlan(ctx, 201, 1, true, SavePlanInput{
Plan: validPlan(), ExpectedPlanRevision: 0, ExpectedSelectionVersion: 1,
})
preview, err := svc.ActivationPreview(ctx, 201, 1, true)
if err != nil {
t.Fatalf("ActivationPreview: %v", err)
}
var wg sync.WaitGroup
successes := 0
var mu sync.Mutex
for i := 0; i < 3; i++ {
wg.Add(1)
go func() {
defer wg.Done()
_, actErr := svc.Activate(ctx, 201, 1, true, ActivateInput{
ConfirmationToken: preview.ConfirmationToken,
ExpectedProjectVersion: preview.ExpectedProjectVersion,
ExpectedPlanRevision: preview.ExpectedPlanRevision,
})
if actErr == nil {
mu.Lock()
successes++
mu.Unlock()
}
}()
}
wg.Wait()
if successes != 1 {
t.Fatalf("expected exactly 1 successful activation, got %d", successes)
}
}