call_flow_with_commit.go

package testcases

import (
	"testing"
	"time"

	"github.com/makasim/flowstate"
	"github.com/stretchr/testify/require"
)

func CallFlowWithCommit(t *testing.T, e flowstate.Engine, fr flowstate.FlowRegistry, d flowstate.Driver) {
	endedCh := make(chan struct{})
	trkr := &Tracker{}

	mustSetFlow(fr, "call", flowstate.FlowFunc(func(stateCtx *flowstate.StateCtx, e flowstate.Engine) (flowstate.Command, error) {
		Track(stateCtx, trkr)

		if stateCtx.Current.Annotation(`state`) == `resumed` {
			return flowstate.Transit(stateCtx, `callEnd`), nil
		}

		nextStateCtx := &flowstate.StateCtx{
			Current: flowstate.State{
				ID: "aNextTID",
			},
		}

		if err := e.Do(
			flowstate.Commit(
				flowstate.Park(stateCtx),
				flowstate.Stack(nextStateCtx, stateCtx, `caller_state`),
				flowstate.Transit(nextStateCtx, `called`),
			),
			flowstate.Execute(nextStateCtx),
		); err != nil {
			return nil, err
		}

		return flowstate.Noop(), nil
	}))
	mustSetFlow(fr, "called", flowstate.FlowFunc(func(stateCtx *flowstate.StateCtx, e flowstate.Engine) (flowstate.Command, error) {
		Track(stateCtx, trkr)

		if err := e.Do(
			flowstate.Transit(stateCtx, `calledEnd`),
			flowstate.Execute(stateCtx),
		); err != nil {
			return nil, err
		}

		return flowstate.Noop(), nil
	}))
	mustSetFlow(fr, "calledEnd", flowstate.FlowFunc(func(stateCtx *flowstate.StateCtx, e flowstate.Engine) (flowstate.Command, error) {
		Track(stateCtx, trkr)
		if stateCtx.Current.Annotations[`caller_state`] != "" {
			callStateCtx := &flowstate.StateCtx{}

			if err := e.Do(
				flowstate.Commit(
					flowstate.Unstack(stateCtx, callStateCtx, `caller_state`),
					flowstate.Transit(callStateCtx, `call`).WithAnnotation(`state`, `resumed`),
					flowstate.Park(stateCtx),
				),
				flowstate.Execute(callStateCtx),
			); err != nil {
				return nil, err
			}

			return flowstate.Noop(), nil
		}

		return flowstate.Park(stateCtx), nil
	}))
	mustSetFlow(fr, "callEnd", flowstate.FlowFunc(func(stateCtx *flowstate.StateCtx, e flowstate.Engine) (flowstate.Command, error) {
		Track(stateCtx, trkr)

		close(endedCh)

		return flowstate.Park(stateCtx), nil
	}))

	stateCtx := &flowstate.StateCtx{
		Current: flowstate.State{
			ID: "aTID",
		},
	}
	err := e.Do(flowstate.Commit(
		flowstate.Transit(stateCtx, `call`),
	))
	require.NoError(t, err)

	err = e.Execute(stateCtx)
	require.NoError(t, err)

	require.Eventually(t, func() bool {
		select {
		case <-endedCh:
			return true
		default:
			return false
		}
	}, time.Second*5, time.Millisecond*50)

	require.Equal(t, []string{
		`call`,
		`called`,
		`calledEnd`,
		`call`,
		`callEnd`,
	}, trkr.Visited())
}