nomino/generators_test.go

package nomino

import (
	"errors"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestWithGenerator(t *testing.T) {
	g := func(*Config) (string, error) { return "abc", nil }
	var c Config
	WithGenerator(g)(&c)
	st, err := c.generator(&c)
	assert.NoError(t, err)
	assert.Equal(t, "abc", st)
}

func TestMultiGeneratorInOrder(t *testing.T) {
	st1 := "abc"
	st2 := "def"
	er1 := errors.New("oops")
	g1 := func(*Config) (string, error) { return st1, nil }
	g2 := func(*Config) (string, error) { return st2, nil }
	g3 := func(*Config) (string, error) { return "", er1 }
	g := MultiGeneratorInOrder(g1, g2, g3)
	st, err := g(nil)
	assert.NoError(t, err)
	assert.Equal(t, st1, st)
	st, err = g(nil)
	assert.NoError(t, err)
	assert.Equal(t, st2, st)
	st, err = g(nil)
	assert.Zero(t, st)
	assert.ErrorIs(t, err, er1)
	st, err = g(nil)
	assert.NoError(t, err)
	assert.Equal(t, st1, st)
}

func TestMultiGeneratorInOrderOne(t *testing.T) {
	st1 := "abc"
	g1 := func(*Config) (string, error) { return st1, nil }
	g := MultiGeneratorInOrder(g1)

	st, err := g(nil)
	assert.NoError(t, err)
	assert.Equal(t, st1, st)
	st, err = g(nil)
	assert.NoError(t, err)
	assert.Equal(t, st1, st)
}

func TestMultiGeneratorInOrderMissing(t *testing.T) {
	g := MultiGeneratorInOrder()
	st, err := g(nil)
	assert.Zero(t, st)
	assert.ErrorIs(t, err, ErrMissingGenerators)
	st, err = g(nil)
	assert.Zero(t, st)
	assert.ErrorIs(t, err, ErrMissingGenerators)
}