[Pause] Reject queries when workflow is paused

Author: gow

service/history/api/queryworkflow/api.go

@@ -96,6 +96,16 @@ func Invoke(
 			}, nil
 		}
 	}
+	// If workflow is paused, return query rejected with PAUSED status.
+	if mutableStateStatus == enumspb.WORKFLOW_EXECUTION_STATUS_PAUSED {
+		return &historyservice.QueryWorkflowResponse{
+			Response: &workflowservice.QueryWorkflowResponse{
+				QueryRejected: &querypb.QueryRejected{
+					Status: enumspb.WORKFLOW_EXECUTION_STATUS_PAUSED,
+				},
+			},
+		}, nil
+	}
 
 	mutableState := workflowLease.GetMutableState()
 	if !mutableState.IsWorkflowExecutionRunning() && !mutableState.HasCompletedAnyWorkflowTask() {

tests/pause_workflow_execution_test.go

@@ -165,7 +165,7 @@ func (s *PauseWorkflowExecutionSuite) TestQueryWorkflowWhenPaused() {
 		}
 	}, 5*time.Second, 200*time.Millisecond)
 
-	// Issue a query with reject condition so that paused workflows return QueryRejected with PAUSED status.
+	// Issue a query to the paused workflow. It should return QueryRejected with WORKFLOW_EXECUTION_STATUS_PAUSED status.
 	queryReq := &workflowservice.QueryWorkflowRequest{
 		Namespace: s.Namespace().String(),
 		Execution: &commonpb.WorkflowExecution{
@@ -175,7 +175,6 @@ func (s *PauseWorkflowExecutionSuite) TestQueryWorkflowWhenPaused() {
 		Query: &querypb.WorkflowQuery{
 			QueryType: "__stack_trace",
 		},
-		QueryRejectCondition: enumspb.QUERY_REJECT_CONDITION_NOT_OPEN,
 	}
 	queryResp, err := s.FrontendClient().QueryWorkflow(ctx, queryReq)
 	s.NoError(err)