995
}
996
998
>
testNamespaceUUID := namespace.ID(uuid.NewString())
999
>
closeTime := time.Now().UTC()
1000
>
startTime := closeTime.Add(-5 * time.Second)
1001
>
1002
>
var startRequests []*manager.RecordWorkflowExecutionStartedRequest
1003
>
for range 5 {
1004
>
startRequests = append(
1005
>
startRequests,
1006
>
s.createOpenWorkflowRecord(
1007
>
testNamespaceUUID,
1008
>
"visibility-workflow-test",
1009
>
"visibility-workflow",
1010
>
startTime,
1011
>
startTime,
1012
>
"test-queue",
1013
>
),
1014
>
)
1015
>
}
1016
1018
>
enumspb.WORKFLOW_EXECUTION_STATUS_RUNNING.String(),
1019
>
enumspb.INDEXED_VALUE_TYPE_KEYWORD,
1020
>
)
1021
>
resp, err := s.VisibilityMgr.CountWorkflowExecutions(
1022
>
s.ctx,
1023
>
&manager.CountWorkflowExecutionsRequest{
1024
>
NamespaceID: testNamespaceUUID,
1025
>
Query: "GROUP BY ExecutionStatus",
1026
>
},
1027
>
)
1028
>
s.NoError(err)
1029
>
s.Equal(int64(5), resp.Count)
1030
>
s.Equal(
1031
>
[]*workflowservice.CountWorkflowExecutionsResponse_AggregationGroup{
1032
>
{
1033
>
GroupValues: []*commonpb.Payload{runningStatusPayload},
1034
>
Count: int64(5),
1035
>
},
1036
>
},
1037
>
resp.Groups,
1038
>
)
1039
>
1040
>
for i := range 2 {
1041
>
s.createClosedWorkflowRecord(
1042
>
startRequests[i],
1043
>
closeTime,
1044
>
enumspb.WORKFLOW_EXECUTION_STATUS_COMPLETED,
1045
>
)
1046
>
}
1047
1049
>
s.ctx,
1050
>
&manager.CountWorkflowExecutionsRequest{
1051
>
NamespaceID: testNamespaceUUID,
1052
>
Query: "GROUP BY ExecutionStatus",
1053
>
},
1054
>
)
1055
>
s.NoError(err)
1056
>
s.Equal(int64(5), resp.Count)
1057
}
1058