Add pipeline module registries

internal/framework/pipeline/extractor_registry_test.go

@@ -42,6 +42,81 @@ func TestExtractorRegistryRegisterAndBuildTrimKeys(t *testing.T) {
 	}
 }
 
+func TestExtractorRegistryRegisterWithSpecStoresMetadata(t *testing.T) {
+	registry := NewExtractorRegistry()
+	spec := ModuleSpec{
+		Key:      " generic-extractor ",
+		Stage:    StageExtract,
+		Provides: []string{" generic-artifact ", "source-citations", "generic-artifact", ""},
+		Requires: []string{" source-document ", "source-document", ""},
+	}
+
+	if err := registry.RegisterWithSpec(spec, fakeExtractorConstructor("generic-extractor")); err != nil {
+		t.Fatalf("RegisterWithSpec() error = %v, want nil", err)
+	}
+
+	got, ok := registry.Spec("\tgeneric-extractor\n")
+	if !ok {
+		t.Fatal("Spec() ok = false, want true")
+	}
+	want := ModuleSpec{
+		Key:      "generic-extractor",
+		Stage:    StageExtract,
+		Provides: []string{"generic-artifact", "source-citations"},
+		Requires: []string{"source-document"},
+	}
+	if !reflect.DeepEqual(got, want) {
+		t.Fatalf("Spec() = %#v, want %#v", got, want)
+	}
+
+	got.Provides[0] = "changed"
+	again, ok := registry.Spec("generic-extractor")
+	if !ok {
+		t.Fatal("Spec() after caller mutation ok = false, want true")
+	}
+	if !reflect.DeepEqual(again, want) {
+		t.Fatalf("Spec() after caller mutation = %#v, want %#v", again, want)
+	}
+}
+
+func TestExtractorRegistryRegisterStoresDefaultSpec(t *testing.T) {
+	registry := NewExtractorRegistry()
+
+	if err := registry.Register(" generic-extractor ", fakeExtractorConstructor("generic-extractor")); err != nil {
+		t.Fatalf("Register() error = %v, want nil", err)
+	}
+
+	got, ok := registry.Spec("generic-extractor")
+	if !ok {
+		t.Fatal("Spec() ok = false, want true")
+	}
+	want := ModuleSpec{Key: "generic-extractor", Stage: StageExtract}
+	if !reflect.DeepEqual(got, want) {
+		t.Fatalf("Spec() = %#v, want %#v", got, want)
+	}
+}
+
+func TestExtractorRegistryRegisterWithSpecRejectsWrongStage(t *testing.T) {
+	registry := NewExtractorRegistry()
+
+	err := registry.RegisterWithSpec(ModuleSpec{Key: "generic-extractor", Stage: StageInput}, fakeExtractorConstructor("generic-extractor"))
+
+	if err == nil {
+		t.Fatal("RegisterWithSpec() error = nil, want error")
+	}
+	if !strings.Contains(err.Error(), "stage") {
+		t.Fatalf("RegisterWithSpec() error = %q, want stage error", err.Error())
+	}
+}
+
+func TestExtractorRegistrySpecRejectsUnknownKey(t *testing.T) {
+	registry := NewExtractorRegistry()
+
+	if _, ok := registry.Spec("missing-extractor"); ok {
+		t.Fatal("Spec() ok = true, want false")
+	}
+}
+
 func TestExtractorRegistryRegisterRejectsEmptyKey(t *testing.T) {
 	registry := NewExtractorRegistry()
 
@@ -183,6 +258,9 @@ func TestExtractorRegistryNilRegistryBehavior(t *testing.T) {
 	if _, err := registry.Build("generic-extractor"); err == nil {
 		t.Fatal("Build() error = nil, want error")
 	}
+	if _, ok := registry.Spec("generic-extractor"); ok {
+		t.Fatal("Spec() ok = true, want false")
+	}
 	if keys := registry.RegisteredKeys(); keys != nil {
 		t.Fatalf("RegisteredKeys() = %#v, want nil", keys)
 	}