mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
Dan Davison
1c1bd30c4b
Adds a state machine defining valid transitions for a CHASM activity. The state machine does not include transitions related to cancel and pause states: we can add those when we implement those features. Part of foundations for CHASM activity implementation / Standalone Activity. - Built (a unit test was written but not committed)
113 lines
4.0 KiB
Go
113 lines
4.0 KiB
Go
package callback
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"go.temporal.io/server/chasm"
|
|
callbackspb "go.temporal.io/server/chasm/lib/callback/gen/callbackpb/v1"
|
|
"go.temporal.io/server/common/backoff"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
func TestValidTransitions(t *testing.T) {
|
|
// Setup
|
|
currentTime := time.Now().UTC()
|
|
callback := &Callback{
|
|
CallbackState: &callbackspb.CallbackState{
|
|
Callback: &callbackspb.Callback{
|
|
Variant: &callbackspb.Callback_Nexus_{
|
|
Nexus: &callbackspb.Callback_Nexus{
|
|
Url: "http://address:666/path/to/callback?query=string",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
callback.SetStateMachineState(callbackspb.CALLBACK_STATUS_SCHEDULED)
|
|
|
|
// AttemptFailed
|
|
mctx := &chasm.MockMutableContext{}
|
|
err := TransitionAttemptFailed.Apply(callback, mctx, EventAttemptFailed{
|
|
Time: currentTime,
|
|
Err: errors.New("test"),
|
|
RetryPolicy: backoff.NewExponentialRetryPolicy(time.Second),
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
// Assert info object is updated
|
|
require.Equal(t, callbackspb.CALLBACK_STATUS_BACKING_OFF, callback.StateMachineState())
|
|
require.Equal(t, int32(1), callback.Attempt)
|
|
require.Equal(t, "test", callback.LastAttemptFailure.Message)
|
|
require.False(t, callback.LastAttemptFailure.GetApplicationFailureInfo().NonRetryable)
|
|
require.Equal(t, currentTime, callback.LastAttemptCompleteTime.AsTime())
|
|
dt := currentTime.Add(time.Second).Sub(callback.NextAttemptScheduleTime.AsTime())
|
|
require.Less(t, dt, time.Millisecond*200)
|
|
|
|
// Assert backoff task is generated
|
|
require.Len(t, mctx.Tasks, 1)
|
|
require.IsType(t, &callbackspb.BackoffTask{}, mctx.Tasks[0].Payload)
|
|
|
|
// Rescheduled
|
|
mctx = &chasm.MockMutableContext{}
|
|
err = TransitionRescheduled.Apply(callback, mctx, EventRescheduled{})
|
|
require.NoError(t, err)
|
|
|
|
// Assert info object is updated only where needed
|
|
require.Equal(t, callbackspb.CALLBACK_STATUS_SCHEDULED, callback.StateMachineState())
|
|
require.Equal(t, int32(1), callback.Attempt)
|
|
require.Equal(t, "test", callback.LastAttemptFailure.Message)
|
|
// Remains unmodified
|
|
require.Equal(t, currentTime, callback.LastAttemptCompleteTime.AsTime())
|
|
require.Nil(t, callback.NextAttemptScheduleTime)
|
|
|
|
// Assert callback task is generated
|
|
require.Len(t, mctx.Tasks, 1)
|
|
require.IsType(t, &callbackspb.InvocationTask{}, mctx.Tasks[0].Payload)
|
|
|
|
// Store the pre-succeeded state to test Failed later
|
|
dup := &Callback{
|
|
CallbackState: proto.Clone(callback.CallbackState).(*callbackspb.CallbackState),
|
|
}
|
|
dup.Status = callback.StateMachineState()
|
|
|
|
// Succeeded
|
|
currentTime = currentTime.Add(time.Second)
|
|
mctx = &chasm.MockMutableContext{}
|
|
err = TransitionSucceeded.Apply(callback, mctx, EventSucceeded{Time: currentTime})
|
|
require.NoError(t, err)
|
|
|
|
// Assert info object is updated only where needed
|
|
require.Equal(t, callbackspb.CALLBACK_STATUS_SUCCEEDED, callback.StateMachineState())
|
|
require.Equal(t, int32(2), callback.Attempt)
|
|
require.Nil(t, callback.LastAttemptFailure)
|
|
require.Equal(t, currentTime, callback.LastAttemptCompleteTime.AsTime())
|
|
require.Nil(t, callback.NextAttemptScheduleTime)
|
|
|
|
// Assert no task is generated on success transition
|
|
require.Empty(t, mctx.Tasks)
|
|
|
|
// Reset back to scheduled
|
|
callback = dup
|
|
// Increment the time to ensure it's updated in the transition
|
|
currentTime = currentTime.Add(time.Second)
|
|
|
|
// failed
|
|
mctx = &chasm.MockMutableContext{}
|
|
err = TransitionFailed.Apply(callback, mctx, EventFailed{Time: currentTime, Err: errors.New("failed")})
|
|
require.NoError(t, err)
|
|
|
|
// Assert info object is updated only where needed
|
|
require.Equal(t, callbackspb.CALLBACK_STATUS_FAILED, callback.StateMachineState())
|
|
require.Equal(t, int32(2), callback.Attempt)
|
|
require.Equal(t, "failed", callback.LastAttemptFailure.Message)
|
|
require.True(t, callback.LastAttemptFailure.GetApplicationFailureInfo().NonRetryable)
|
|
require.Equal(t, currentTime, callback.LastAttemptCompleteTime.AsTime())
|
|
require.Nil(t, callback.NextAttemptScheduleTime)
|
|
|
|
// Assert task is not generated, failed is terminal
|
|
require.Empty(t, mctx.Tasks)
|
|
}
|