knowledge-governance/cmd/worker/library_role_test.go

package main

import (
	"context"
	"strings"
	"testing"

	"knowledge-governance/internal/appconfig"
	"knowledge-governance/internal/jobs"
	postgresstore "knowledge-governance/internal/platform/postgres"
	"knowledge-governance/internal/versionhistory"
)

// 发布执行器必须只挂四类 release 作业：多挂一类就会抢走本机队列里别的执行者的作业。
func TestRegisterLibraryReleaseHandlersCoversOnlyReleaseJobs(t *testing.T) {
	handlers := map[string]jobs.Handler{}
	registerLibraryReleaseHandlers(handlers, nil, appconfig.Config{}, nil)
	want := []string{librarySyncJobType, libraryRestoreJobType, libraryRecoverJobType, libraryBaselineJobType}
	if len(handlers) != len(want) {
		t.Fatalf("registered %d handlers, want %d: %#v", len(handlers), len(want), handlers)
	}
	for _, jobType := range want {
		if handlers[jobType] == nil {
			t.Fatalf("handler %q is not registered", jobType)
		}
	}
}

// 准入未通过时 handler 必须"明确失败且不做任何动作"：不能靠 nil store 或进程内状态蒙过去。
func TestRegisterLibraryReleaseHandlersFailsClosedWhenNotAdmitted(t *testing.T) {
	handlers := map[string]jobs.Handler{}
	registerLibraryReleaseHandlers(handlers, nil, appconfig.Config{}, errNotAdmitted{})
	for _, jobType := range []string{librarySyncJobType, libraryRestoreJobType, libraryRecoverJobType, libraryBaselineJobType} {
		err := handlers[jobType](context.Background(), jobs.Job{})
		if err == nil || !strings.Contains(err.Error(), "准入未通过") {
			t.Fatalf("handler %q error = %v, want admission refusal", jobType, err)
		}
	}
}

type errNotAdmitted struct{}

func (errNotAdmitted) Error() string { return "fixture: target instance unknown" }

// capture 角色不能注册任何 source 之外的队列；release-executor 反之。
func TestWorkerHandlersFollowWorkerRole(t *testing.T) {
	release := appconfig.Config{
		LibraryWorkerRole:            appconfig.LibraryWorkerRoleReleaseExecutor,
		LibraryAgentSocket:           "/run/library-agent/agent.sock",
		TestWeKnoraEnabled:           true,
		TestWeKnoraAPIKeyFile:        "/nonexistent/api-key",
		PPScanEnabled:                true,
		VersionHistoryGiteaBaseURL:   "https://gitea.invalid/Weknora/knowledge-governance",
		VersionHistoryAllowedOrigins: []string{"https://gitea.invalid"},
	}
	handlers, err := workerHandlers(postgresstore.NewStore(nil), release)
	if err != nil {
		t.Fatalf("release-executor workerHandlers() error = %v", err)
	}
	for _, forbidden := range []string{libraryCaptureJobType, "test.upload", "test.poll", "test.validate", "test.remove", "source.scan.pp", "source.catalog.refresh", "artifact.fetch", versionhistory.JobType} {
		if handlers[forbidden] != nil {
			t.Fatalf("release-executor registered %q", forbidden)
		}
	}
	if handlers["demo"] == nil {
		t.Fatal("demo handler must stay registered")
	}

	capture := appconfig.Config{
		LibraryWorkerRole:     appconfig.LibraryWorkerRoleCapture,
		LibraryAgentSocket:    "/run/library-agent/agent.sock",
		TestWeKnoraEnabled:    true,
		TestWeKnoraAPIKeyFile: "/nonexistent/api-key",
	}
	if _, err := workerHandlers(nil, capture); err == nil || !strings.Contains(err.Error(), "test WeKnora API key") {
		t.Fatalf("capture role must reach the test WeKnora wiring, got %v", err)
	}
	capture.TestWeKnoraEnabled = false
	capture.PPScanEnabled = false
	handlers, err = workerHandlers(postgresstore.NewStore(nil), capture)
	if err != nil {
		t.Fatalf("capture workerHandlers() error = %v", err)
	}
	if handlers[libraryCaptureJobType] == nil {
		t.Fatal("capture role must register library.capture")
	}
}