Fix the unit test when feature is disabled (#2386)

uber · Jul 16, 2022 · b7bfaea · b7bfaea
1 parent e35874a
commit b7bfaea
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/service/history/task/transfer_standby_task_executor_test.go b/service/history/task/transfer_standby_task_executor_test.go
@@ -787,11 +787,13 @@ func (s *transferStandbyTaskExecutorSuite) TestProcessRecordWorkflowStartedTask(
 	persistenceMutableState, err := test.CreatePersistenceMutableState(mutableState, startEvent.ID, startEvent.Version)
 	s.NoError(err)
 	s.mockExecutionMgr.On("GetWorkflowExecution", mock.Anything, mock.Anything).Return(&persistence.GetWorkflowExecutionResponse{State: persistenceMutableState}, nil)
-	s.mockVisibilityMgr.On(
-		"RecordWorkflowExecutionUninitialized",
-		mock.Anything,
-		createRecordWorkflowExecutionUninitializedRequest(transferTask, mutableState),
-	).Once().Return(nil)
+	if s.mockShard.GetConfig().EnableRecordWorkflowExecutionUninitialized() {
+		s.mockVisibilityMgr.On(
+			"RecordWorkflowExecutionUninitialized",
+			mock.Anything,
+			createRecordWorkflowExecutionUninitializedRequest(transferTask, mutableState),
+		).Once().Return(nil)
+	}
 	s.mockVisibilityMgr.On(
 		"RecordWorkflowExecutionStarted",
 		mock.Anything,