Add worker identity to TestUpdateWorkflow tests (#4475)

temporalio · Jun 12, 2023 · 30c9be5 · 30c9be5
1 parent eeec6e0
commit 30c9be5
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/tests/update_workflow_test.go b/tests/update_workflow_test.go
@@ -412,6 +412,7 @@ func (s *integrationSuite) TestUpdateWorkflow_NewNormalWorkflowTask_AcceptComple
 				Engine:              s.engine,
 				Namespace:           s.namespace,
 				TaskQueue:           tv.TaskQueue(),
+				Identity:            tv.WorkerIdentity(),
 				WorkflowTaskHandler: wtHandler,
 				MessageHandler:      msgHandler,
 				Logger:              s.Logger,
@@ -544,6 +545,7 @@ func (s *integrationSuite) TestUpdateWorkflow_FirstNormalStartedWorkflowTask_Acc
 				Engine:              s.engine,
 				Namespace:           s.namespace,
 				TaskQueue:           tv.TaskQueue(),
+				Identity:            tv.WorkerIdentity(),
 				WorkflowTaskHandler: wtHandler,
 				MessageHandler:      msgHandler,
 				Logger:              s.Logger,
@@ -678,6 +680,7 @@ func (s *integrationSuite) TestUpdateWorkflow_NormalScheduledWorkflowTask_Accept
 				Engine:              s.engine,
 				Namespace:           s.namespace,
 				TaskQueue:           tv.TaskQueue(),
+				Identity:            tv.WorkerIdentity(),
 				WorkflowTaskHandler: wtHandler,
 				MessageHandler:      msgHandler,
 				Logger:              s.Logger,
@@ -1036,6 +1039,7 @@ func (s *integrationSuite) TestUpdateWorkflow_ValidateWorkerMessages() {
 				Engine:              s.engine,
 				Namespace:           s.namespace,
 				TaskQueue:           tv.TaskQueue(),
+				Identity:            tv.WorkerIdentity(),
 				WorkflowTaskHandler: wtHandler,
 				MessageHandler:      msgHandler,
 				Logger:              s.Logger,
@@ -1168,6 +1172,7 @@ func (s *integrationSuite) TestUpdateWorkflow_NewStickySpeculativeWorkflowTask_A
 				TaskQueue:                    tv.TaskQueue(),
 				StickyTaskQueue:              tv.StickyTaskQueue(),
 				StickyScheduleToStartTimeout: 3 * time.Second,
+				Identity:                     tv.WorkerIdentity(),
 				WorkflowTaskHandler:          wtHandler,
 				MessageHandler:               msgHandler,
 				Logger:                       s.Logger,
@@ -1290,6 +1295,7 @@ func (s *integrationSuite) TestUpdateWorkflow_NewStickySpeculativeWorkflowTask_A
 		TaskQueue:                    tv.TaskQueue(),
 		StickyTaskQueue:              tv.StickyTaskQueue(),
 		StickyScheduleToStartTimeout: 3 * time.Second,
+		Identity:                     tv.WorkerIdentity(),
 		WorkflowTaskHandler:          wtHandler,
 		MessageHandler:               msgHandler,
 		Logger:                       s.Logger,
@@ -1402,6 +1408,7 @@ func (s *integrationSuite) TestUpdateWorkflow_FirstNormalStartedWorkflowTask_Rej
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -1623,6 +1630,7 @@ func (s *integrationSuite) TestUpdateWorkflow_NewNormalWorkflowTask_Reject() {
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -1762,6 +1770,7 @@ func (s *integrationSuite) TestUpdateWorkflow_1stAccept_2ndAccept_2ndComplete_1s
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -1922,6 +1931,7 @@ func (s *integrationSuite) TestUpdateWorkflow_1stAccept_2ndReject_1stComplete()
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -2092,6 +2102,7 @@ func (s *integrationSuite) TestUpdateWorkflow_FailSpeculativeWorkflowTask() {
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -2232,6 +2243,7 @@ func (s *integrationSuite) TestUpdateWorkflow_ConvertStartedSpeculativeWorkflowT
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -2344,6 +2356,7 @@ func (s *integrationSuite) TestUpdateWorkflow_ConvertScheduledSpeculativeWorkflo
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -2487,6 +2500,7 @@ func (s *integrationSuite) TestUpdateWorkflow_StartToCloseTimeoutSpeculativeWork
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -2600,6 +2614,7 @@ func (s *integrationSuite) TestUpdateWorkflow_ScheduleToStartTimeoutSpeculativeW
 		TaskQueue:                    tv.TaskQueue(),
 		StickyTaskQueue:              tv.StickyTaskQueue(),
 		StickyScheduleToStartTimeout: 1 * time.Second, // Important!
+		Identity:                     tv.WorkerIdentity(),
 		WorkflowTaskHandler:          wtHandler,
 		MessageHandler:               msgHandler,
 		Logger:                       s.Logger,
@@ -2703,6 +2718,7 @@ func (s *integrationSuite) TestUpdateWorkflow_StartedSpeculativeWorkflowTask_Ter
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -2801,6 +2817,7 @@ func (s *integrationSuite) TestUpdateWorkflow_ScheduledSpeculativeWorkflowTask_T
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -2940,6 +2957,7 @@ func (s *integrationSuite) TestUpdateWorkflow_CompleteWorkflow_TerminateUpdate()
 				Engine:              s.engine,
 				Namespace:           s.namespace,
 				TaskQueue:           tv.TaskQueue(),
+				Identity:            tv.WorkerIdentity(),
 				WorkflowTaskHandler: wtHandler,
 				MessageHandler:      msgHandler,
 				Logger:              s.Logger,
@@ -3060,6 +3078,7 @@ func (s *integrationSuite) TestUpdateWorkflow_SpeculativeWorkflowTask_Heartbeat(
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -3165,6 +3184,7 @@ func (s *integrationSuite) TestUpdateWorkflow_NewScheduledSpeculativeWorkflowTas
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -3311,6 +3331,7 @@ func (s *integrationSuite) TestUpdateWorkflow_NewStartedSpeculativeWorkflowTaskL
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -3443,6 +3464,7 @@ func (s *integrationSuite) TestUpdateWorkflow_ScheduledSpeculativeWorkflowTask_D
 		Engine:              s.engine,
 		Namespace:           s.namespace,
 		TaskQueue:           tv.TaskQueue(),
+		Identity:            tv.WorkerIdentity(),
 		WorkflowTaskHandler: wtHandler,
 		MessageHandler:      msgHandler,
 		Logger:              s.Logger,
@@ -3694,6 +3716,7 @@ func (s *integrationSuite) TestUpdateWorkflow_CompletedSpeculativeWorkflowTask_D
 				Engine:              s.engine,
 				Namespace:           s.namespace,
 				TaskQueue:           tv.TaskQueue(),
+				Identity:            tv.WorkerIdentity(),
 				WorkflowTaskHandler: wtHandler,
 				MessageHandler:      msgHandler,
 				Logger:              s.Logger,