@@ -114,7 +114,7 @@ func (s *historyBuilderSuite) TestHistoryBuilderDynamicSuccess() {
114114 id := "dynamic-historybuilder-success-test-workflow-id"
115115 rid := "dynamic-historybuilder-success-test-run-id"
116116 wt := "dynamic-historybuilder-success-type"
117- tl := "dynamic-historybuilder-success-taskqueue"
117+ tl := & taskqueuepb. TaskQueue { Kind : enumspb . TASK_QUEUE_KIND_NORMAL , Name : "dynamic-historybuilder-success-taskqueue" }
118118 identity := "dynamic-historybuilder-success-worker"
119119 input := payloads .EncodeString ("dynamic-historybuilder-success-input" )
120120 execTimeout := int32 (70 )
@@ -318,7 +318,7 @@ func (s *historyBuilderSuite) TestHistoryBuilderWorkflowStartFailures() {
318318 id := "historybuilder-workflowstart-failures-test-workflow-id"
319319 rid := "historybuilder-workflowstart-failures-test-run-id"
320320 wt := "historybuilder-workflowstart-failures-type"
321- tl := "historybuilder-workflowstart-failures-taskqueue"
321+ tl := & taskqueuepb. TaskQueue { Kind : enumspb . TASK_QUEUE_KIND_NORMAL , Name : "historybuilder-workflowstart-failures-taskqueue" }
322322 identity := "historybuilder-workflowstart-failures-worker"
323323 input := payloads .EncodeString ("historybuilder-workflowstart-failures-input" )
324324 execTimeout := int32 (70 )
@@ -349,7 +349,7 @@ func (s *historyBuilderSuite) TestHistoryBuilderWorkflowStartFailures() {
349349 StartRequest : & workflowservice.StartWorkflowExecutionRequest {
350350 WorkflowId : we .WorkflowId ,
351351 WorkflowType : & commonpb.WorkflowType {Name : wt },
352- TaskQueue : & taskqueuepb. TaskQueue { Name : tl } ,
352+ TaskQueue : tl ,
353353 Input : input ,
354354 WorkflowExecutionTimeoutSeconds : execTimeout ,
355355 WorkflowRunTimeoutSeconds : runTimeout ,
@@ -370,7 +370,7 @@ func (s *historyBuilderSuite) TestHistoryBuilderWorkflowTaskScheduledFailures()
370370 id := "historybuilder-workflow-task-scheduled-failures-test-workflow-id"
371371 rid := "historybuilder-workflow-task-scheduled-failures-test-run-id"
372372 wt := "historybuilder-workflow-task-scheduled-failures-type"
373- tl := "historybuilder-workflow-task-scheduled-failures-taskqueue"
373+ tl := & taskqueuepb. TaskQueue { Kind : enumspb . TASK_QUEUE_KIND_NORMAL , Name : "historybuilder-workflow-task-scheduled-failures-taskqueue" }
374374 identity := "historybuilder-workflow-task-scheduled-failures-worker"
375375 input := payloads .EncodeString ("historybuilder-workflow-task-scheduled-failures-input" )
376376 execTimeout := int32 (70 )
@@ -406,7 +406,7 @@ func (s *historyBuilderSuite) TestHistoryBuilderWorkflowTaskStartedFailures() {
406406 id := "historybuilder-workflow-task-started-failures-test-workflow-id"
407407 rid := "historybuilder-workflow-task-started-failures-test-run-id"
408408 wt := "historybuilder-workflow-task-started-failures-type"
409- tl := "historybuilder-workflow-task-started-failures-taskqueue"
409+ tl := & taskqueuepb. TaskQueue { Kind : enumspb . TASK_QUEUE_KIND_NORMAL , Name : "historybuilder-workflow-task-started-failures-taskqueue" }
410410 identity := "historybuilder-workflow-task-started-failures-worker"
411411 input := payloads .EncodeString ("historybuilder-workflow-task-started-failures-input" )
412412 execTimeout := int32 (70 )
@@ -422,7 +422,7 @@ func (s *historyBuilderSuite) TestHistoryBuilderWorkflowTaskStartedFailures() {
422422 s .Equal (int64 (2 ), s .getNextEventID ())
423423
424424 _ , _ , err := s .msBuilder .AddWorkflowTaskStartedEvent (2 , uuid .New (), & workflowservice.PollWorkflowTaskQueueRequest {
425- TaskQueue : & taskqueuepb. TaskQueue { Name : tl } ,
425+ TaskQueue : tl ,
426426 Identity : identity ,
427427 })
428428 s .NotNil (err )
@@ -440,7 +440,7 @@ func (s *historyBuilderSuite) TestHistoryBuilderWorkflowTaskStartedFailures() {
440440 s .Equal (common .EmptyEventID , s .getPreviousWorkflowTaskStartedEventID ())
441441
442442 _ , _ , err = s .msBuilder .AddWorkflowTaskStartedEvent (100 , uuid .New (), & workflowservice.PollWorkflowTaskQueueRequest {
443- TaskQueue : & taskqueuepb. TaskQueue { Name : tl } ,
443+ TaskQueue : tl ,
444444 Identity : identity ,
445445 })
446446 s .NotNil (err )
@@ -463,7 +463,7 @@ func (s *historyBuilderSuite) TestHistoryBuilderFlushBufferedEvents() {
463463 id := "flush-buffered-events-test-workflow-id"
464464 rid := "flush-buffered-events-test-run-id"
465465 wt := "flush-buffered-events-type"
466- tl := "flush-buffered-events-taskqueue"
466+ tl := & taskqueuepb. TaskQueue { Kind : enumspb . TASK_QUEUE_KIND_NORMAL , Name : "flush-buffered-events-taskqueue" }
467467 identity := "flush-buffered-events-worker"
468468 input := payloads .EncodeString ("flush-buffered-events-input" )
469469 execTimeout := int32 (70 )
@@ -627,7 +627,7 @@ func (s *historyBuilderSuite) TestHistoryBuilderFlushBufferedEvents() {
627627
628628func (s * historyBuilderSuite ) TestHistoryBuilderWorkflowCancellationRequested () {
629629 workflowType := "some random workflow type"
630- taskqueue := "some random taskqueue"
630+ taskqueue := & taskqueuepb. TaskQueue { Kind : enumspb . TASK_QUEUE_KIND_NORMAL , Name : "some random taskqueue" }
631631 identity := "some random identity"
632632 input := payloads .EncodeString ("some random workflow input" )
633633 execTimeout := int32 (70 )
@@ -699,7 +699,7 @@ func (s *historyBuilderSuite) TestHistoryBuilderWorkflowCancellationRequested()
699699
700700func (s * historyBuilderSuite ) TestHistoryBuilderWorkflowCancellationFailed () {
701701 workflowType := "some random workflow type"
702- taskqueue := "some random taskqueue"
702+ taskqueue := & taskqueuepb. TaskQueue { Kind : enumspb . TASK_QUEUE_KIND_NORMAL , Name : "some random taskqueue" }
703703 identity := "some random identity"
704704 input := payloads .EncodeString ("some random workflow input" )
705705 execTimeout := int32 (70 )
@@ -782,13 +782,13 @@ func (s *historyBuilderSuite) getPreviousWorkflowTaskStartedEventID() int64 {
782782 return s .msBuilder .GetExecutionInfo ().LastProcessedEvent
783783}
784784
785- func (s * historyBuilderSuite ) addWorkflowExecutionStartedEvent (we commonpb.WorkflowExecution , workflowType ,
786- taskQueue string , input * commonpb.Payloads , executionTimeout , runTimeout , taskTimeout int32 , identity string ) * historypb.HistoryEvent {
785+ func (s * historyBuilderSuite ) addWorkflowExecutionStartedEvent (we commonpb.WorkflowExecution , workflowType string ,
786+ taskQueue * taskqueuepb. TaskQueue , input * commonpb.Payloads , executionTimeout , runTimeout , taskTimeout int32 , identity string ) * historypb.HistoryEvent {
787787
788788 request := & workflowservice.StartWorkflowExecutionRequest {
789789 WorkflowId : we .WorkflowId ,
790790 WorkflowType : & commonpb.WorkflowType {Name : workflowType },
791- TaskQueue : & taskqueuepb. TaskQueue { Name : taskQueue } ,
791+ TaskQueue : taskQueue ,
792792 Input : input ,
793793 WorkflowExecutionTimeoutSeconds : executionTimeout ,
794794 WorkflowRunTimeoutSeconds : runTimeout ,
@@ -816,9 +816,9 @@ func (s *historyBuilderSuite) addWorkflowTaskScheduledEvent() *workflowTaskInfo
816816}
817817
818818func (s * historyBuilderSuite ) addWorkflowTaskStartedEvent (scheduleID int64 ,
819- taskQueue , identity string ) * historypb.HistoryEvent {
819+ taskQueue * taskqueuepb. TaskQueue , identity string ) * historypb.HistoryEvent {
820820 event , _ , err := s .msBuilder .AddWorkflowTaskStartedEvent (scheduleID , uuid .New (), & workflowservice.PollWorkflowTaskQueueRequest {
821- TaskQueue : & taskqueuepb. TaskQueue { Name : taskQueue } ,
821+ TaskQueue : taskQueue ,
822822 Identity : identity ,
823823 })
824824 s .Nil (err )
@@ -934,15 +934,15 @@ func (s *historyBuilderSuite) addRequestCancelExternalWorkflowExecutionFailedEve
934934 return event
935935}
936936
937- func (s * historyBuilderSuite ) validateWorkflowExecutionStartedEvent (event * historypb.HistoryEvent , workflowType ,
938- taskQueue string , input * commonpb.Payloads , executionTimeout , runTimeout , taskTimeout int32 , identity string ) {
937+ func (s * historyBuilderSuite ) validateWorkflowExecutionStartedEvent (event * historypb.HistoryEvent , workflowType string ,
938+ taskQueue * taskqueuepb. TaskQueue , input * commonpb.Payloads , executionTimeout , runTimeout , taskTimeout int32 , identity string ) {
939939 s .NotNil (event )
940940 s .Equal (enumspb .EVENT_TYPE_WORKFLOW_EXECUTION_STARTED , event .EventType )
941941 s .Equal (common .FirstEventID , event .EventId )
942942 attributes := event .GetWorkflowExecutionStartedEventAttributes ()
943943 s .NotNil (attributes )
944944 s .Equal (workflowType , attributes .WorkflowType .Name )
945- s .Equal (taskQueue , attributes .TaskQueue . Name )
945+ s .Equal (taskQueue , attributes .TaskQueue )
946946 s .Equal (input , attributes .Input )
947947 s .Equal (executionTimeout , attributes .WorkflowExecutionTimeoutSeconds )
948948 s .Equal (runTimeout , attributes .WorkflowRunTimeoutSeconds )
@@ -951,7 +951,7 @@ func (s *historyBuilderSuite) validateWorkflowExecutionStartedEvent(event *histo
951951}
952952
953953func (s * historyBuilderSuite ) validateWorkflowTaskScheduledEvent (di * workflowTaskInfo , eventID int64 ,
954- taskQueue string , timeout int32 ) {
954+ taskQueue * taskqueuepb. TaskQueue , timeout int32 ) {
955955 s .NotNil (di )
956956 s .Equal (eventID , di .ScheduleID )
957957 s .Equal (taskQueue , di .TaskQueue )
0 commit comments