113
}
114
116
>
rangeID := rand.Int63()
117
>
taskQueue := s.createTaskQueue(rangeID)
118
>
119
>
var tasks []*persistencespb.AllocatedTaskInfo
120
>
for pass := int64(1); pass <= 5; pass++ {
121
>
tasks = append(tasks, s.randomTask(pass, pass))
122
>
}
123
>
_, err := s.taskManager.CreateTasks(s.ctx, &p.CreateTasksRequest{
124
>
TaskQueueInfo: &p.PersistedTaskQueueInfo{RangeID: rangeID, Data: taskQueue},
125
>
Tasks: tasks,
126
>
})
127
>
s.NoError(err)
128
>
129
>
_, err = s.taskManager.CompleteTasksLessThan(s.ctx, &p.CompleteTasksLessThanRequest{
130
>
NamespaceID: s.namespaceID,
131
>
TaskQueueName: s.taskQueueName,
132
>
TaskType: s.taskQueueType,
133
>
ExclusiveMaxPass: 3,
134
>
ExclusiveMaxTaskID: 0,
135
>
Limit: 10,
136
>
})
137
>
s.NoError(err)
138
>
139
>
resp, err := s.taskManager.GetTasks(s.ctx, &p.GetTasksRequest{
140
>
NamespaceID: s.namespaceID,
141
>
TaskQueue: s.taskQueueName,
142
>
TaskType: s.taskQueueType,
143
>
InclusiveMinPass: 1,
144
>
InclusiveMinTaskID: 0,
145
>
ExclusiveMaxTaskID: math.MaxInt64,
146
>
PageSize: 10,
147
>
})
148
>
s.NoError(err)
149
>
150
>
expected := []*persistencespb.AllocatedTaskInfo{tasks[2], tasks[3], tasks[4]}
151
>
protorequire.ProtoSliceEqual(s.T(), expected, resp.Tasks)
152
>
s.Nil(resp.NextPageToken)
153
}
154