107
}
108
110
>
rangeID := rand.Int63()
111
>
taskQueue := s.createTaskQueue(rangeID)
112
>
113
>
taskID := rand.Int63()
114
>
task := s.randomTask(taskID)
115
>
_, err := s.taskManager.CreateTasks(s.ctx, &p.CreateTasksRequest{
116
>
TaskQueueInfo: &p.PersistedTaskQueueInfo{
117
>
RangeID: rangeID,
118
>
Data: taskQueue,
119
>
},
120
>
Tasks: []*persistencespb.AllocatedTaskInfo{task},
121
>
})
122
>
s.NoError(err)
123
>
124
>
resp, err := s.taskManager.GetTasks(s.ctx, &p.GetTasksRequest{
125
>
NamespaceID: s.namespaceID,
126
>
TaskQueue: s.taskQueueName,
127
>
TaskType: s.taskQueueType,
128
>
InclusiveMinTaskID: taskID,
129
>
ExclusiveMaxTaskID: taskID + 1,
130
>
PageSize: 100,
131
>
NextPageToken: nil,
132
>
})
133
>
s.NoError(err)
134
>
protorequire.ProtoSliceEqual(s.T(), []*persistencespb.AllocatedTaskInfo{task}, resp.Tasks)
135
>
s.Nil(resp.NextPageToken)
136
>
}
137
138
func (s *TaskQueueTaskSuite) TestCreateGet_Multiple() {