Add pipeline module registries

internal/framework/pipeline/input_registry_test.go

@@ -43,6 +43,81 @@ func TestInputAdapterRegistryRegisterAndBuildTrimKeys(t *testing.T) {
 	}
 }
 
+func TestInputAdapterRegistryRegisterWithSpecStoresMetadata(t *testing.T) {
+	registry := NewInputAdapterRegistry()
+	spec := ModuleSpec{
+		Key:      " generic-input ",
+		Stage:    StageInput,
+		Provides: []string{" parsed-source ", "source-document", "parsed-source", ""},
+		Requires: []string{" raw-bytes ", "raw-bytes", ""},
+	}
+
+	if err := registry.RegisterWithSpec(spec, fakeInputAdapterConstructor("generic-input")); err != nil {
+		t.Fatalf("RegisterWithSpec() error = %v, want nil", err)
+	}
+
+	got, ok := registry.Spec("\tgeneric-input\n")
+	if !ok {
+		t.Fatal("Spec() ok = false, want true")
+	}
+	want := ModuleSpec{
+		Key:      "generic-input",
+		Stage:    StageInput,
+		Provides: []string{"parsed-source", "source-document"},
+		Requires: []string{"raw-bytes"},
+	}
+	if !reflect.DeepEqual(got, want) {
+		t.Fatalf("Spec() = %#v, want %#v", got, want)
+	}
+
+	got.Provides[0] = "changed"
+	again, ok := registry.Spec("generic-input")
+	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 TestInputAdapterRegistryRegisterStoresDefaultSpec(t *testing.T) {
+	registry := NewInputAdapterRegistry()
+
+	if err := registry.Register(" generic-input ", fakeInputAdapterConstructor("generic-input")); err != nil {
+		t.Fatalf("Register() error = %v, want nil", err)
+	}
+
+	got, ok := registry.Spec("generic-input")
+	if !ok {
+		t.Fatal("Spec() ok = false, want true")
+	}
+	want := ModuleSpec{Key: "generic-input", Stage: StageInput}
+	if !reflect.DeepEqual(got, want) {
+		t.Fatalf("Spec() = %#v, want %#v", got, want)
+	}
+}
+
+func TestInputAdapterRegistryRegisterWithSpecRejectsWrongStage(t *testing.T) {
+	registry := NewInputAdapterRegistry()
+
+	err := registry.RegisterWithSpec(ModuleSpec{Key: "generic-input", Stage: StageExtract}, fakeInputAdapterConstructor("generic-input"))
+
+	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 TestInputAdapterRegistrySpecRejectsUnknownKey(t *testing.T) {
+	registry := NewInputAdapterRegistry()
+
+	if _, ok := registry.Spec("missing-input"); ok {
+		t.Fatal("Spec() ok = true, want false")
+	}
+}
+
 func TestInputAdapterRegistryRegisterRejectsEmptyKey(t *testing.T) {
 	registry := NewInputAdapterRegistry()
 
@@ -184,6 +259,9 @@ func TestInputAdapterRegistryNilRegistryBehavior(t *testing.T) {
 	if _, err := registry.Build("generic-input"); err == nil {
 		t.Fatal("Build() error = nil, want error")
 	}
+	if _, ok := registry.Spec("generic-input"); ok {
+		t.Fatal("Spec() ok = true, want false")
+	}
 	if keys := registry.RegisteredKeys(); keys != nil {
 		t.Fatalf("RegisteredKeys() = %#v, want nil", keys)
 	}