multi_lookup_test.go

package tempura_test

import (
	"context"
	"fmt"
	"os"
	"testing"

	"github.com/ebi-yade/go-tempura"
	"github.com/stretchr/testify/assert"
)

func TestMultiLookup_Validate(t *testing.T) {
	t.Parallel()

	keyAsValue := func(val string) (string, bool) {
		return val, true
	}
	fetchSecret := func(ctx context.Context, key string) (string, bool, error) {
		return "XXXXXXXX", true, nil
	}
	always := func(val string) (string, bool, error) {
		return "", false, fmt.Errorf("this function always returns an error")
	}

	tests := []struct {
		name     string
		receiver *tempura.MultiLookup
		checkErr func(t *testing.T, err error)
	}{
		// ==================== VALID CASES ====================
		{
			name: "single valid function",
			receiver: &tempura.MultiLookup{
				tempura.DotPrefix("env"): tempura.Func(os.LookupEnv),
			},
			checkErr: func(t *testing.T, err error) {
				assert.NoError(t, err)
			},
		},
		{
			name: "multiple valid functions",
			receiver: &tempura.MultiLookup{
				tempura.DotPrefix("env"):     tempura.Func(os.LookupEnv),
				tempura.DotPrefix("default"): tempura.Func(keyAsValue),
				tempura.DotPrefix("oops"):    tempura.FuncWithError(always),
			},
			checkErr: func(t *testing.T, err error) {
				assert.NoError(t, err)
			},
		},
		// ==================== INVALID CASES ====================
		{
			name:     "no functions registered",
			receiver: &tempura.MultiLookup{},
			checkErr: func(t *testing.T, err error) {
				assert.ErrorIs(t, err, tempura.ErrNoFunctionRegistered)
			},
		},
		{
			name: "contains invalid function that receives context",
			receiver: &tempura.MultiLookup{
				tempura.DotPrefix("env"):     tempura.Func(os.LookupEnv),
				tempura.DotPrefix("default"): tempura.Func(keyAsValue),
				tempura.DotPrefix("secret"):  tempura.FuncWithContextError(fetchSecret),
			},
			checkErr: func(t *testing.T, err error) {
				expected := tempura.InvalidFunctionError{}
				assert.ErrorAs(t, err, &expected)
				assert.Equal(t, "MultiLookup", expected.Type, "Type mismatch")
				assert.Equal(t, tempura.DotPrefix("secret"), expected.Prefix, "Prefix mismatch")
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := tt.receiver.Validate()
			tt.checkErr(t, err)
		})
	}
}

func TestMultiLookupContext_Validate(t *testing.T) {
	t.Parallel()

	keyAsValue := func(val string) (string, bool) {
		return val, true
	}
	fetchSecret := func(ctx context.Context, key string) (string, bool, error) {
		return "XXXXXXXX", true, nil
	}
	always := func(val string) (string, bool, error) {
		return "", false, fmt.Errorf("this function always returns an error")
	}

	ctx := context.TODO()

	tests := []struct {
		name     string
		receiver *tempura.MultiLookupContext
		checkErr func(t *testing.T, err error)
	}{
		// ==================== VALID CASES ====================
		{
			name: "single valid function that receives context",
			receiver: tempura.MultiLookup{
				tempura.DotPrefix("secret"): tempura.FuncWithContextError(fetchSecret),
			}.BindContext(ctx),
			checkErr: func(t *testing.T, err error) {
				assert.NoError(t, err)
			},
		},
		{
			name: "multiple valid functions",
			receiver: tempura.MultiLookup{
				tempura.DotPrefix("env"):     tempura.Func(os.LookupEnv),
				tempura.DotPrefix("default"): tempura.Func(keyAsValue),
				tempura.DotPrefix("oops"):    tempura.FuncWithError(always),
				tempura.DotPrefix("secret"):  tempura.FuncWithContextError(fetchSecret),
			}.BindContext(ctx),
			checkErr: func(t *testing.T, err error) {
				assert.NoError(t, err)
			},
		},
		// ==================== INVALID CASES ====================
		{
			name:     "no functions registered",
			receiver: tempura.MultiLookup{}.BindContext(ctx),
			checkErr: func(t *testing.T, err error) {
				assert.ErrorIs(t, err, tempura.ErrNoFunctionRegistered)
			},
		},
		{
			name: "forget to call BindContext",
			receiver: &tempura.MultiLookupContext{
				MultiLookup: tempura.MultiLookup{
					tempura.DotPrefix("secret"): tempura.FuncWithContextError(fetchSecret),
				},
				//Ctx is nil
			},
			checkErr: func(t *testing.T, err error) {
				assert.ErrorIs(t, err, tempura.ErrContextUntypedNil)
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			err := tt.receiver.Validate()
			tt.checkErr(t, err)
		})
	}
}