mirror of
https://github.com/hanzoai/tasks.git
synced 2026-11-03 03:56:04 +00:00
zeekay
5a601d081b
Full module path rebrand from Temporal to Hanzo Tasks: - go.mod module: github.com/hanzoai/tasks - All internal Go imports updated (1906 files) - cmd/server renamed to cmd/tasksd - Binary name: tasksd (was temporal-server) - Docker images: hanzoai/tasks (was temporalio/server) - README rewritten with Hanzo Tasks branding - LLM.md created with integration notes - Makefile, goreleaser, CI workflows updated - go.temporal.io/sdk and go.temporal.io/api imports preserved (separate repos) Fork of Temporal.io (MIT license).
127 lines
3.8 KiB
Go
127 lines
3.8 KiB
Go
package isactivitytaskvalid
|
|
|
|
import (
|
|
"math/rand"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
"go.temporal.io/api/serviceerror"
|
|
persistencespb "github.com/hanzoai/tasks/api/persistence/v1"
|
|
"github.com/hanzoai/tasks/common"
|
|
"github.com/hanzoai/tasks/service/history/api"
|
|
historyi "github.com/hanzoai/tasks/service/history/interfaces"
|
|
"go.uber.org/mock/gomock"
|
|
)
|
|
|
|
type (
|
|
apiSuite struct {
|
|
suite.Suite
|
|
*require.Assertions
|
|
|
|
controller *gomock.Controller
|
|
workflowLease api.WorkflowLease
|
|
workflowContext *historyi.MockWorkflowContext
|
|
mutableState *historyi.MockMutableState
|
|
}
|
|
)
|
|
|
|
func TestAPISuite(t *testing.T) {
|
|
s := new(apiSuite)
|
|
suite.Run(t, s)
|
|
}
|
|
|
|
func (s *apiSuite) SetupTest() {
|
|
s.Assertions = require.New(s.T())
|
|
|
|
s.controller = gomock.NewController(s.T())
|
|
s.workflowContext = historyi.NewMockWorkflowContext(s.controller)
|
|
s.mutableState = historyi.NewMockMutableState(s.controller)
|
|
s.workflowLease = api.NewWorkflowLease(
|
|
s.workflowContext,
|
|
func(err error) {},
|
|
s.mutableState,
|
|
)
|
|
}
|
|
|
|
func (s *apiSuite) TeardownTest() {
|
|
s.controller.Finish()
|
|
}
|
|
|
|
func (s *apiSuite) TestWorkflowCompleted() {
|
|
s.mutableState.EXPECT().IsWorkflowExecutionRunning().Return(false)
|
|
|
|
_, err := isActivityTaskValid(s.workflowLease, rand.Int63(), rand.Int31())
|
|
s.Error(err)
|
|
s.IsType(&serviceerror.NotFound{}, err)
|
|
}
|
|
|
|
func (s *apiSuite) TestWorkflowRunning_ActivityTaskNotStarted() {
|
|
s.mutableState.EXPECT().IsWorkflowExecutionRunning().Return(true)
|
|
activityScheduleEventID := rand.Int63()
|
|
stamp := rand.Int31()
|
|
s.mutableState.EXPECT().GetActivityInfo(activityScheduleEventID).Return(&persistencespb.ActivityInfo{
|
|
ScheduledEventId: activityScheduleEventID,
|
|
StartedEventId: common.EmptyEventID,
|
|
Stamp: stamp,
|
|
}, true)
|
|
|
|
valid, err := isActivityTaskValid(s.workflowLease, activityScheduleEventID, stamp)
|
|
s.NoError(err)
|
|
s.True(valid)
|
|
}
|
|
|
|
func (s *apiSuite) TestWorkflowRunning_ActivityTaskStarted() {
|
|
s.mutableState.EXPECT().IsWorkflowExecutionRunning().Return(true)
|
|
activityScheduleEventID := rand.Int63()
|
|
stamp := rand.Int31()
|
|
s.mutableState.EXPECT().GetActivityInfo(activityScheduleEventID).Return(&persistencespb.ActivityInfo{
|
|
ScheduledEventId: activityScheduleEventID,
|
|
StartedEventId: activityScheduleEventID + 1,
|
|
Stamp: stamp,
|
|
}, true)
|
|
|
|
valid, err := isActivityTaskValid(s.workflowLease, activityScheduleEventID, stamp)
|
|
s.NoError(err)
|
|
s.False(valid)
|
|
}
|
|
|
|
func (s *apiSuite) TestWorkflowRunning_ActivityTaskStampMismatch() {
|
|
s.mutableState.EXPECT().IsWorkflowExecutionRunning().Return(true)
|
|
activityScheduleEventID := rand.Int63()
|
|
const storedStamp = int32(456)
|
|
s.mutableState.EXPECT().GetActivityInfo(activityScheduleEventID).Return(&persistencespb.ActivityInfo{
|
|
ScheduledEventId: activityScheduleEventID,
|
|
StartedEventId: common.EmptyEventID,
|
|
Stamp: storedStamp,
|
|
}, true)
|
|
|
|
valid, err := isActivityTaskValid(s.workflowLease, activityScheduleEventID, storedStamp+1)
|
|
s.NoError(err)
|
|
s.False(valid)
|
|
}
|
|
|
|
func (s *apiSuite) TestWorkflowRunning_ActivityTaskStampLegacy() {
|
|
s.mutableState.EXPECT().IsWorkflowExecutionRunning().Return(true)
|
|
activityScheduleEventID := rand.Int63()
|
|
s.mutableState.EXPECT().GetActivityInfo(activityScheduleEventID).Return(&persistencespb.ActivityInfo{
|
|
ScheduledEventId: activityScheduleEventID,
|
|
StartedEventId: common.EmptyEventID,
|
|
Stamp: 0,
|
|
}, true)
|
|
|
|
valid, err := isActivityTaskValid(s.workflowLease, activityScheduleEventID, 0)
|
|
s.NoError(err)
|
|
s.True(valid)
|
|
}
|
|
|
|
func (s *apiSuite) TestWorkflowRunning_ActivityTaskMissing() {
|
|
s.mutableState.EXPECT().IsWorkflowExecutionRunning().Return(true)
|
|
activityScheduleEventID := rand.Int63()
|
|
s.mutableState.EXPECT().GetActivityInfo(activityScheduleEventID).Return(nil, false)
|
|
|
|
valid, err := isActivityTaskValid(s.workflowLease, activityScheduleEventID, rand.Int31())
|
|
s.NoError(err)
|
|
s.False(valid)
|
|
}
|