package pipeline

import (
	"testing"

	"gitea.maximumdirect.net/eric/notarius/internal/framework/contracts"
)

func TestOutputEncoderRegistryBehavior(t *testing.T) {
	runRegistryBehaviorTests(t, registryBehaviorCase[contracts.OutputEncoder]{
		name:       "OutputEncoderRegistry",
		key:        "generic-output",
		stage:      StageOutput,
		wrongStage: StageExtract,
		newRegistry: func() any {
			return NewOutputEncoderRegistry()
		},
		register: func(registry any, key string, constructor func() (contracts.OutputEncoder, error)) error {
			return registry.(*OutputEncoderRegistry).Register(key, constructor)
		},
		registerWithSpec: func(registry any, spec ModuleSpec, constructor func() (contracts.OutputEncoder, error)) error {
			return registry.(*OutputEncoderRegistry).RegisterWithSpec(spec, constructor)
		},
		build: func(registry any, key string) (contracts.OutputEncoder, error) {
			return registry.(*OutputEncoderRegistry).Build(key)
		},
		spec: func(registry any, key string) (ModuleSpec, bool) {
			return registry.(*OutputEncoderRegistry).Spec(key)
		},
		registeredKeys: func(registry any) []string {
			return registry.(*OutputEncoderRegistry).RegisteredKeys()
		},
		nilRegister: func(key string, constructor func() (contracts.OutputEncoder, error)) error {
			var registry *OutputEncoderRegistry
			return registry.Register(key, constructor)
		},
		nilBuild: func(key string) (contracts.OutputEncoder, error) {
			var registry *OutputEncoderRegistry
			return registry.Build(key)
		},
		nilSpec: func(key string) (ModuleSpec, bool) {
			var registry *OutputEncoderRegistry
			return registry.Spec(key)
		},
		nilRegisteredKey: func() []string {
			var registry *OutputEncoderRegistry
			return registry.RegisteredKeys()
		},
		constructor: func(key string) func() (contracts.OutputEncoder, error) {
			return func() (contracts.OutputEncoder, error) {
				return registryOutputEncoder{key: key}, nil
			}
		},
		moduleKey: func(module contracts.OutputEncoder) string {
			return module.Key()
		},
	})
}