Function TestLoopOnIteration

flow/loop_test.go:51–61

(t *testing.T)

}

func TestLoopOnIteration(t *testing.T) {
	var seen []int
	body := func(ctx context.Context, in State) (State, error) { return in, nil }
	step := Loop(body, LoopMax(3), OnIteration(func(iter int, s State) { seen = append(seen, iter) }))
	if _, err := step(context.Background(), State{}); err != nil {
		t.Fatal(err)
	}
	if len(seen) != 3 || seen[0] != 1 || seen[2] != 3 {
		t.Fatalf("expected iterations [1 2 3], got %v", seen)
	}
}

func TestLoopBodyError(t *testing.T) {
	body := func(ctx context.Context, in State) (State, error) {