Files
knowledge-governance/cmd/worker/main_test.go
T
zheromiku dcdc8d1ddd feat: 知识治理平台(含已验证发布/回滚契约)
全栈知识治理平台首次入库(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 干净
2026-09-30 10:54:06 +08:00

93 lines
3.1 KiB
Go

package main
import (
"context"
"os"
"path/filepath"
"strings"
"testing"
"time"
"knowledge-governance/internal/appconfig"
postgresstore "knowledge-governance/internal/platform/postgres"
)
func TestRunRequiresDatabaseURL(t *testing.T) {
err := run(context.Background(), func(string) string { return "" })
if err == nil || !strings.Contains(err.Error(), "DATABASE_URL") {
t.Fatalf("run() error = %v, want DATABASE_URL error", err)
}
}
func TestWorkerHandlersExcludeReleaseAndTestJobsByDefault(t *testing.T) {
handlers, err := workerHandlers(nil, appconfig.Config{})
if err != nil {
t.Fatalf("workerHandlers() error = %v", err)
}
if handlers["demo"] == nil || handlers["demo.sleep"] == nil {
t.Fatalf("worker handlers = %#v, want demo and demo.sleep", handlers)
}
if handlers["release.probe"] != nil {
t.Fatal("ordinary worker must not register release.probe")
}
if handlers["test.upload"] != nil || handlers["test.poll"] != nil || handlers["test.validate"] != nil || handlers["test.remove"] != nil {
t.Fatal("ordinary worker must not register test WeKnora jobs when disabled")
}
if handlers["source.scan.pp"] != nil {
t.Fatal("ordinary worker must not register PP scan jobs when disabled")
}
}
func TestWorkerHandlersRegisterPPScanWhenExplicitlyEnabled(t *testing.T) {
root := t.TempDir()
config := appconfig.Config{
PPScanEnabled: true,
PPScanAllowedRoot: root,
PPScanLedgerPath: filepath.Join(root, "state.jsonl"),
PPScanManifestPath: filepath.Join(root, "manifest.json"),
}
handlers, err := workerHandlers(postgresstore.NewStore(nil), config)
if err != nil {
t.Fatalf("workerHandlers() error = %v", err)
}
if handlers["source.scan.pp"] == nil {
t.Fatal("source.scan.pp handler is not registered")
}
}
func TestWorkerHandlersRequireAPIKeyWhenTestWeKnoraEnabled(t *testing.T) {
config := appconfig.Config{TestWeKnoraEnabled: true, TestWeKnoraAPIKeyFile: "/nonexistent/api-key"}
_, err := workerHandlers(nil, config)
if err == nil || !strings.Contains(err.Error(), "read test WeKnora API key") {
t.Fatalf("workerHandlers() error = %v", err)
}
}
func TestWorkerHandlersRegisterTestJobsForExplicitDockerHTTP(t *testing.T) {
keyFile := filepath.Join(t.TempDir(), "api-key")
if err := os.WriteFile(keyFile, []byte("local-test-key\n"), 0o600); err != nil {
t.Fatal(err)
}
config := appconfig.Config{
TestWeKnoraEnabled: true,
TestWeKnoraAllowDockerHTTP: true,
TestWeKnoraBaseURL: "http://app:8080",
TestWeKnoraAPIKeyFile: keyFile,
GovernanceSpoolRoot: t.TempDir(),
TestWeKnoraHTTPTimeout: time.Second,
TestWeKnoraPollInitial: time.Second,
TestWeKnoraStalledAfter: time.Minute,
TestWeKnoraMaxUploadBytes: 1024,
TestAcceptancePolicyVersion: "structural-v1",
}
handlers, err := workerHandlers(postgresstore.NewStore(nil), config)
if err != nil {
t.Fatalf("workerHandlers() error = %v", err)
}
for _, jobType := range []string{"test.upload", "test.poll", "test.validate", "test.remove"} {
if handlers[jobType] == nil {
t.Fatalf("handler %q is not registered", jobType)
}
}
}