🚧 Pass Config to generator

This commit is contained in:
Dan Jones 2025-03-13 15:36:30 -05:00
commit 921020d9fd
6 changed files with 31 additions and 31 deletions

View file

@ -10,10 +10,10 @@ import (
)
func TestWithGenerator(t *testing.T) {
g := func() (string, error) { return "abc", nil }
g := func(*Config) (string, error) { return "abc", nil }
var c Config
WithGenerator(g)(&c)
st, err := c.generator()
st, err := c.generator(&c)
assert.NoError(t, err)
assert.Equal(t, "abc", st)
}
@ -22,49 +22,49 @@ func TestMultiGeneratorInOrder(t *testing.T) {
st1 := "abc"
st2 := "def"
er1 := errors.New("oops")
g1 := func() (string, error) { return st1, nil }
g2 := func() (string, error) { return st2, nil }
g3 := func() (string, error) { return "", er1 }
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()
st, err := g(nil)
assert.NoError(t, err)
assert.Equal(t, st1, st)
st, err = g()
st, err = g(nil)
assert.NoError(t, err)
assert.Equal(t, st2, st)
st, err = g()
st, err = g(nil)
assert.Zero(t, st)
assert.ErrorIs(t, err, er1)
st, err = g()
st, err = g(nil)
assert.NoError(t, err)
assert.Equal(t, st1, st)
}
func TestMultiGeneratorInOrderOne(t *testing.T) {
st1 := "abc"
g1 := func() (string, error) { return st1, nil }
g1 := func(*Config) (string, error) { return st1, nil }
g := MultiGeneratorInOrder(g1)
st, err := g()
st, err := g(nil)
assert.NoError(t, err)
assert.Equal(t, st1, st)
st, err = g()
st, err = g(nil)
assert.NoError(t, err)
assert.Equal(t, st1, st)
}
func TestMultiGeneratorInOrderMissing(t *testing.T) {
g := MultiGeneratorInOrder()
st, err := g()
st, err := g(nil)
assert.Zero(t, st)
assert.ErrorIs(t, err, ErrMissingGenerators)
st, err = g()
st, err = g(nil)
assert.Zero(t, st)
assert.ErrorIs(t, err, ErrMissingGenerators)
}
func TestUUID(t *testing.T) {
st, err := UUID()()
st, err := UUID()(nil)
assert.NoError(t, err)
_, parseErr := uuid.Parse(st)
assert.NoError(t, parseErr)
@ -80,13 +80,13 @@ func TestUUIDFail(t *testing.T) {
uuid.SetRand(badRead{})
defer uuid.SetRand(nil)
_, err := UUID()()
_, err := UUID()(nil)
assert.Equal(t, errors.New("sorry"), err)
}
func TestTimestamp(t *testing.T) {
n := time.Now()
st, err := Timestamp()()
st, err := Timestamp()(nil)
assert.NoError(t, err)
assert.Equal(t, n.Format(FileTimestamp), st)
}
@ -94,14 +94,14 @@ func TestTimestamp(t *testing.T) {
func TestTime(t *testing.T) {
d := time.Date(1986, time.March, 28, 12, 0, 0, 0, time.UTC)
st, err := Time(d)()
st, err := Time(d)(nil)
assert.NoError(t, err)
assert.Equal(t, d.Format(FileTimestamp), st)
}
func TestTimestampUTC(t *testing.T) {
n := time.Now()
st, err := TimestampUTC()()
st, err := TimestampUTC()(nil)
assert.NoError(t, err)
assert.Equal(t, n.UTC().Format(FileTimestampNoTZ), st)
}