Skip to content

Commit

Permalink
Completed the tests for nosql_execution_store_util.go
Browse files Browse the repository at this point in the history
  • Loading branch information
@agautam478
agautam478 committed Apr 5, 2024
1 parent 9740c19 commit db18f6f
Showing 1 changed file with 275 additions and 15 deletions.
290 changes: 275 additions & 15 deletions common/persistence/nosql/nosql_execution_store_util_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -672,53 +672,267 @@ func TestPrepareTransferTasksForWorkflowTxn(t *testing.T) {
testCases := []struct {
name string
tasks []persistence.Task
expectFunc func(*nosqlplugin.MockDB)
domainID string
workflowID string
runID string
validate func(*testing.T, []*nosqlplugin.TransferTask, error)
}{
{
name: "Success - Prepare Transfer Tasks",
name: "CancelExecutionTask - Success",
domainID: "domainID-cancel",
workflowID: "workflowID-cancel",
runID: "runID-cancel",
tasks: []persistence.Task{
&persistence.CancelExecutionTask{
TaskData: persistence.TaskData{
VisibilityTimestamp: time.Now(),
TaskID: 1002,
Version: 1,
},
TargetDomainID: "targetDomainID-cancel",
TargetWorkflowID: "targetWorkflowID-cancel",
TargetRunID: "targetRunID-cancel",
TargetChildWorkflowOnly: true,
InitiatedID: 1002,
},
},
validate: func(t *testing.T, tasks []*nosqlplugin.TransferTask, err error) {
assert.NoError(t, err)
assert.Len(t, tasks, 1)
task := tasks[0]
assert.Equal(t, "targetDomainID-cancel", task.TargetDomainID)
assert.Equal(t, true, task.TargetChildWorkflowOnly)
assert.Equal(t, int64(1002), task.TaskID)
assert.Equal(t, int64(1), task.Version)
},
},
{
name: "ActivityTask - Success",
domainID: "domainID-activity",
workflowID: "workflowID-activity",
runID: "runID-activity",
tasks: []persistence.Task{
&persistence.ActivityTask{
TaskData: persistence.TaskData{
Version: 1,
VisibilityTimestamp: time.Now(),
TaskID: 1001,
Version: 1,
},
DomainID: "domainID",
DomainID: "targetDomainID-activity",
TaskList: "taskList-activity",
ScheduleID: 1001,
},
},
expectFunc: func(mockDB *nosqlplugin.MockDB) {},
validate: func(t *testing.T, tasks []*nosqlplugin.TransferTask, err error) {
assert.NoError(t, err)
assert.NotEmpty(t, tasks)
assert.Len(t, tasks, 1)
task := tasks[0]
assert.Equal(t, persistence.TransferTaskTypeActivityTask, task.TaskType)
assert.Equal(t, "targetDomainID-activity", task.TargetDomainID)
assert.Equal(t, "taskList-activity", task.TaskList)
assert.Equal(t, int64(1001), task.ScheduleID)
assert.Equal(t, int64(1), task.Version)
},
},
{
name: "Failure - Unsupported Task Type",
name: "DefaultTargetRunID - When Empty",
domainID: "domainID-default-runid",
workflowID: "workflowID-default-runid",
runID: "runID-default-runid",
tasks: []persistence.Task{
&persistence.CancelExecutionTask{
TaskData: persistence.TaskData{
VisibilityTimestamp: time.Now(),
TaskID: 2001,
Version: 1,
},
TargetDomainID: "targetDomainID-cancel",
TargetWorkflowID: "targetWorkflowID-cancel",
TargetRunID: "", // Intentionally left empty to trigger the defaulting logic
TargetChildWorkflowOnly: true,
InitiatedID: 2001,
},
&persistence.SignalExecutionTask{
TaskData: persistence.TaskData{
VisibilityTimestamp: time.Now(),
TaskID: 2002,
Version: 1,
},
TargetDomainID: "targetDomainID-signal",
TargetWorkflowID: "targetWorkflowID-signal",
TargetRunID: "", // Intentionally left empty to trigger the defaulting logic
TargetChildWorkflowOnly: false,
InitiatedID: 2002,
},
},
validate: func(t *testing.T, tasks []*nosqlplugin.TransferTask, err error) {
assert.NoError(t, err)
for _, task := range tasks {
assert.Equal(t, persistence.TransferTaskTransferTargetRunID, task.TargetRunID, "TargetRunID should default to TransferTaskTransferTargetRunID")

}
},
},
{
name: "SignalExecutionTask - Success",
domainID: "domainID-signal",
workflowID: "workflowID-signal",
runID: "runID-signal",
tasks: []persistence.Task{
&persistence.SignalExecutionTask{
TaskData: persistence.TaskData{
VisibilityTimestamp: time.Now(),
TaskID: 1003,
Version: 1,
},
TargetDomainID: "targetDomainID-signal",
TargetWorkflowID: "targetWorkflowID-signal",
TargetRunID: "targetRunID-signal",
TargetChildWorkflowOnly: true,
InitiatedID: 1003,
},
},
validate: func(t *testing.T, tasks []*nosqlplugin.TransferTask, err error) {
assert.NoError(t, err)
assert.Len(t, tasks, 1)
task := tasks[0]
assert.Equal(t, "targetDomainID-signal", task.TargetDomainID)
assert.Equal(t, true, task.TargetChildWorkflowOnly)
assert.Equal(t, int64(1003), task.TaskID)
assert.Equal(t, int64(1), task.Version)
},
},
{
name: "StartChildExecutionTask - Success",
domainID: "domainID-start-child",
workflowID: "workflowID-start-child",
runID: "runID-start-child",
tasks: []persistence.Task{
&persistence.StartChildExecutionTask{
TaskData: persistence.TaskData{
VisibilityTimestamp: time.Now(),
TaskID: 1004,
Version: 1,
},
TargetDomainID: "child-execution-domain-id",
TargetWorkflowID: "child-workflow-id",
InitiatedID: 1004,
},
},
validate: func(t *testing.T, tasks []*nosqlplugin.TransferTask, err error) {
assert.NoError(t, err)
assert.Len(t, tasks, 1)
task := tasks[0]
assert.Equal(t, "child-execution-domain-id", task.TargetDomainID)
assert.Equal(t, "child-workflow-id", task.TargetWorkflowID)
assert.Equal(t, int64(1004), task.TaskID)
assert.Equal(t, int64(1), task.Version)
},
},
{
name: "RecordChildExecutionCompletedTask - Success",
domainID: "domainID-record-child",
workflowID: "workflowID-record-child",
runID: "runID-record-child",
tasks: []persistence.Task{
&persistence.RecordChildExecutionCompletedTask{
TaskData: persistence.TaskData{
VisibilityTimestamp: time.Now(),
TaskID: 1005,
Version: 1,
},
TargetDomainID: "completed-child-domain-id",
TargetWorkflowID: "completed-child-workflow-id",
TargetRunID: "completed-child-run-id",
},
},
validate: func(t *testing.T, tasks []*nosqlplugin.TransferTask, err error) {
assert.NoError(t, err)
assert.Len(t, tasks, 1)
task := tasks[0]
assert.Equal(t, "completed-child-domain-id", task.TargetDomainID)
assert.Equal(t, "completed-child-workflow-id", task.TargetWorkflowID)
assert.Equal(t, "completed-child-run-id", task.TargetRunID)
assert.Equal(t, int64(1005), task.TaskID)
assert.Equal(t, int64(1), task.Version)
},
},
{
name: "ApplyParentClosePolicyTask - Success",
domainID: "domainID-apply-parent",
workflowID: "workflowID-apply-parent",
runID: "runID-apply-parent",
tasks: []persistence.Task{
&persistence.ApplyParentClosePolicyTask{
TaskData: persistence.TaskData{
VisibilityTimestamp: time.Now(),
TaskID: 1006,
Version: 1,
},
TargetDomainIDs: map[string]struct{}{"target-domain-id-1": {}, "target-domain-id-2": {}},
},
},
validate: func(t *testing.T, tasks []*nosqlplugin.TransferTask, err error) {
assert.NoError(t, err)
assert.Len(t, tasks, 1)
task := tasks[0]
assert.Equal(t, map[string]struct{}{"target-domain-id-1": {}, "target-domain-id-2": {}}, task.TargetDomainIDs)
assert.Equal(t, int64(1006), task.TaskID)
assert.Equal(t, int64(1), task.Version)
},
},
{
name: "DecisionTask - Success",
domainID: "domainID-decision",
workflowID: "workflowID-decision",
runID: "runID-decision",
tasks: []persistence.Task{
&persistence.DecisionTask{
TaskData: persistence.TaskData{
VisibilityTimestamp: time.Now(),
TaskID: 1001,
Version: 1,
},
DomainID: "targetDomainID-decision",
TaskList: "taskList-decision",
ScheduleID: 1001,
RecordVisibility: true,
},
},
validate: func(t *testing.T, tasks []*nosqlplugin.TransferTask, err error) {
assert.NoError(t, err)
assert.Len(t, tasks, 1)
task := tasks[0]
assert.Equal(t, int64(1001), task.TaskID)
assert.Equal(t, "targetDomainID-decision", task.TargetDomainID)
assert.Equal(t, true, task.RecordVisibility)
},
},
{
name: "Unsupported Task Type",
domainID: "domainID-unsupported",
workflowID: "workflowID-unsupported",
runID: "runID-unsupported",
tasks: []persistence.Task{
&dummyTaskType{
VisibilityTimestamp: time.Now(),
TaskID: -1,
TaskID: 9999,
},
},
expectFunc: func(mockDB *nosqlplugin.MockDB) {},
validate: func(t *testing.T, tasks []*nosqlplugin.TransferTask, err error) {
assert.Error(t, err)
assert.Nil(t, tasks)
},
},
}

for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
mockCtrl := gomock.NewController(t)

mockDB := nosqlplugin.NewMockDB(mockCtrl)
store := newTestNosqlExecutionStore(mockDB, log.NewNoop())

if tc.expectFunc != nil {
tc.expectFunc(mockDB)
}

tasks, err := store.prepareTransferTasksForWorkflowTxn("domainID", "workflowID", "runID", tc.tasks)
tasks, err := store.prepareTransferTasksForWorkflowTxn(tc.domainID, tc.workflowID, tc.runID, tc.tasks)
tc.validate(t, tasks, err)
})
}
Expand Down Expand Up @@ -1089,6 +1303,52 @@ func TestNosqlExecutionStoreUtilsExtended(t *testing.T) {
assert.False(t, ok)
},
},
{
name: "CurrentWorkflowRequestForCreateWorkflowTxn - Zombie mode",
setupStore: func(store *nosqlExecutionStore) (interface{}, error) {
executionInfo := &persistence.InternalWorkflowExecutionInfo{
State: persistence.WorkflowStateCreated,
CloseStatus: persistence.WorkflowCloseStatusNone,
CreateRequestID: "create-request-id-zombie",
}
request := &persistence.InternalCreateWorkflowExecutionRequest{
Mode: persistence.CreateWorkflowModeZombie,
PreviousRunID: "previous-run-id-zombie",
}
return store.prepareCurrentWorkflowRequestForCreateWorkflowTxn(
"domain-id-zombie", "workflow-id-zombie", "run-id-zombie", executionInfo, 123, request)
},
validate: func(t *testing.T, result interface{}, err error) {
assert.NoError(t, err)
currentWorkflowReq := result.(*nosqlplugin.CurrentWorkflowWriteRequest)
assert.Equal(t, nosqlplugin.CurrentWorkflowWriteModeNoop, currentWorkflowReq.WriteMode)
assert.Equal(t, "create-request-id-zombie", currentWorkflowReq.Row.CreateRequestID)
},
},
{
name: "CurrentWorkflowRequestForCreateWorkflowTxn - ContinueAsNew mode",
setupStore: func(store *nosqlExecutionStore) (interface{}, error) {
executionInfo := &persistence.InternalWorkflowExecutionInfo{
State: persistence.WorkflowStateRunning,
CloseStatus: persistence.WorkflowCloseStatusNone,
CreateRequestID: "create-request-id-continueasnew",
}
request := &persistence.InternalCreateWorkflowExecutionRequest{
Mode: persistence.CreateWorkflowModeContinueAsNew,
PreviousRunID: "previous-run-id-continueasnew",
}
return store.prepareCurrentWorkflowRequestForCreateWorkflowTxn(
"domain-id-continueasnew", "workflow-id-continueasnew", "run-id-continueasnew", executionInfo, 123, request)
},
validate: func(t *testing.T, result interface{}, err error) {
assert.NoError(t, err)
currentWorkflowReq := result.(*nosqlplugin.CurrentWorkflowWriteRequest)
assert.Equal(t, nosqlplugin.CurrentWorkflowWriteModeUpdate, currentWorkflowReq.WriteMode)
assert.Equal(t, "create-request-id-continueasnew", currentWorkflowReq.Row.CreateRequestID)
assert.NotNil(t, currentWorkflowReq.Condition)
assert.Equal(t, "previous-run-id-continueasnew", *currentWorkflowReq.Condition.CurrentRunID)
},
},
{
name: "assertNotCurrentExecution - Success with different RunID",
setupStore: func(store *nosqlExecutionStore) (interface{}, error) {
Expand Down

0 comments on commit db18f6f

Please sign in to comment.