✨ Add MultiGenerator

generators_test.go

@@ -18,6 +18,51 @@ func TestWithGenerator(t *testing.T) {
 	assert.Equal(t, "abc", st)
 }
 
+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 }
+	g := MultiGeneratorInOrder(g1, g2, g3)
+	st, err := g()
+	assert.NoError(t, err)
+	assert.Equal(t, st1, st)
+	st, err = g()
+	assert.NoError(t, err)
+	assert.Equal(t, st2, st)
+	st, err = g()
+	assert.Zero(t, st)
+	assert.ErrorIs(t, err, er1)
+	st, err = g()
+	assert.NoError(t, err)
+	assert.Equal(t, st1, st)
+}
+
+func TestMultiGeneratorInOrderOne(t *testing.T) {
+	st1 := "abc"
+	g1 := func() (string, error) { return st1, nil }
+	g := MultiGeneratorInOrder(g1)
+
+	st, err := g()
+	assert.NoError(t, err)
+	assert.Equal(t, st1, st)
+	st, err = g()
+	assert.NoError(t, err)
+	assert.Equal(t, st1, st)
+}
+
+func TestMultiGeneratorInOrderMissing(t *testing.T) {
+	g := MultiGeneratorInOrder()
+	st, err := g()
+	assert.Zero(t, st)
+	assert.ErrorIs(t, err, ErrMissingGenerators)
+	st, err = g()
+	assert.Zero(t, st)
+	assert.ErrorIs(t, err, ErrMissingGenerators)
+}
+
 func TestUUID(t *testing.T) {
 	st, err := UUID()()
 	assert.NoError(t, err)