zheromiku
dcdc8d1ddd
全栈知识治理平台首次入库（P1-P10 + 发布链路模拟验证闭环）： - 治理核心：源目录/变更集/人工审核/发布批次/发布执行（fail-closed 闸口） - 测试通道：WeKnora 测试 KB 上传/轮询/验收；poll 完成时写入 parser_generation/index_generation，修复 manual/seed 路径发布闸口永久拦截 - 发布适配器 v071（本地 source=测试角色 / 云端 target=生产角色 模拟验证通过）： * ES 导出/导入 mapping 自适应（.keyword 子字段、可选 tenant_id） * OSS locator 支持嵌套 scheme；chunks.seq_id 由目标序列分配 * bundle 归档同身份重试原子替换；verifier 按 bundle chunk_type 计数 * 回滚 job payload 支持 prior_execution_id（strict 解码） - 三通道发布+回滚演练实测：publish 4/4 verified，检索命中， rollback 4/4 rolled_back，目标端 PG is_enabled=false、ES is_enabled=false - 前端 A·Data-console 风格全站统一，120 用例通过 - 门禁：go vet / go test 全包通过，gofmt 干净
101 lines
3.9 KiB
Go
101 lines
3.9 KiB
Go
package testweknora
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"knowledge-governance/internal/governance"
|
|
"knowledge-governance/internal/jobs"
|
|
postgresstore "knowledge-governance/internal/platform/postgres"
|
|
"knowledge-governance/internal/platform/weknora"
|
|
)
|
|
|
|
type removalStoreStub struct {
|
|
candidate postgresstore.PublicationForRemoval
|
|
completed string
|
|
failed string
|
|
}
|
|
|
|
func (store *removalStoreStub) LoadPublicationForRemoval(context.Context, string) (postgresstore.PublicationForRemoval, error) {
|
|
return store.candidate, nil
|
|
}
|
|
|
|
func (store *removalStoreStub) RecordRemovalCompleted(_ context.Context, publicationID string, _ time.Time) error {
|
|
store.completed = publicationID
|
|
return nil
|
|
}
|
|
|
|
func (store *removalStoreStub) RecordRemovalFailure(_ context.Context, publicationID, code string, _ time.Time) error {
|
|
store.failed = publicationID + ":" + code
|
|
return nil
|
|
}
|
|
|
|
type removalClientStub struct {
|
|
knowledge weknora.Knowledge
|
|
getCalls int
|
|
deleted string
|
|
keepAfterDelete bool
|
|
}
|
|
|
|
func (client *removalClientStub) GetKnowledge(context.Context, string) (weknora.Knowledge, error) {
|
|
client.getCalls++
|
|
if client.getCalls > 1 && !client.keepAfterDelete {
|
|
return weknora.Knowledge{}, &weknora.HTTPError{StatusCode: 404}
|
|
}
|
|
return client.knowledge, nil
|
|
}
|
|
|
|
func (client *removalClientStub) DeleteKnowledge(_ context.Context, knowledgeID string) error {
|
|
client.deleted = knowledgeID
|
|
return nil
|
|
}
|
|
|
|
func TestRemoveHandlerVerifiesIdentityAndReadBackBeforeCompletion(t *testing.T) {
|
|
store := &removalStoreStub{candidate: postgresstore.PublicationForRemoval{
|
|
ID: "publication-1", TestKBID: "kb-1", KnowledgeID: "knowledge-1", RetrievalState: governance.RetrievalDisabling,
|
|
}}
|
|
client := &removalClientStub{knowledge: weknora.Knowledge{ID: "knowledge-1", KnowledgeBaseID: "kb-1"}}
|
|
handler, err := NewRemoveHandler(RemoveHandlerConfig{Store: store, Client: client, Now: func() time.Time { return time.Date(2026, 9, 9, 3, 0, 0, 0, time.UTC) }})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := handler.Handle(context.Background(), jobs.Job{Payload: []byte(`{"publication_id":"publication-1"}`)}); err != nil {
|
|
t.Fatalf("Handle() error = %v", err)
|
|
}
|
|
if client.deleted != "knowledge-1" || store.completed != "publication-1" || store.failed != "" {
|
|
t.Fatalf("deleted=%q completed=%q failed=%q", client.deleted, store.completed, store.failed)
|
|
}
|
|
}
|
|
|
|
func TestRemoveHandlerRejectsMismatchedKnowledgeBaseWithoutDeleting(t *testing.T) {
|
|
store := &removalStoreStub{candidate: postgresstore.PublicationForRemoval{
|
|
ID: "publication-1", TestKBID: "kb-1", KnowledgeID: "knowledge-1", RetrievalState: governance.RetrievalDisabling,
|
|
}}
|
|
client := &removalClientStub{knowledge: weknora.Knowledge{ID: "knowledge-1", KnowledgeBaseID: "other-kb"}}
|
|
handler, err := NewRemoveHandler(RemoveHandlerConfig{Store: store, Client: client})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = handler.Handle(context.Background(), jobs.Job{Payload: []byte(`{"publication_id":"publication-1"}`)})
|
|
if err != nil || client.deleted != "" || store.failed != "publication-1:removal_identity_invalid" {
|
|
t.Fatalf("error=%v deleted=%q failed=%q", err, client.deleted, store.failed)
|
|
}
|
|
}
|
|
|
|
func TestRemoveHandlerRetriesWhileAsynchronousDeleteIsStillVisible(t *testing.T) {
|
|
store := &removalStoreStub{candidate: postgresstore.PublicationForRemoval{
|
|
ID: "publication-1", TestKBID: "kb-1", KnowledgeID: "knowledge-1", RetrievalState: governance.RetrievalDisabling,
|
|
}}
|
|
client := &removalClientStub{knowledge: weknora.Knowledge{ID: "knowledge-1", KnowledgeBaseID: "kb-1"}, keepAfterDelete: true}
|
|
handler, err := NewRemoveHandler(RemoveHandlerConfig{Store: store, Client: client})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
err = handler.Handle(context.Background(), jobs.Job{Payload: []byte(`{"publication_id":"publication-1"}`)})
|
|
if !errors.Is(err, ErrRemovePending) || store.completed != "" {
|
|
t.Fatalf("error=%v completed=%q", err, store.completed)
|
|
}
|
|
}
|