151
}
152
154
>
numTasks := 20
155
>
immediateTasks := s.AddRandomTasks(
156
>
fakeImmediateTaskCategory,
157
>
numTasks,
158
>
func(workflowKey definition.WorkflowKey, taskID int64, visibilityTimestamp time.Time) tasks.Task {
159
>
fakeTask := tasks.NewFakeTask(
160
>
workflowKey,
161
>
fakeImmediateTaskCategory,
162
>
visibilityTimestamp,
163
>
)
164
>
fakeTask.SetTaskID(taskID)
165
>
return fakeTask
166
>
},
167
)
168
170
>
loadedTasks := s.PaginateTasks(
171
>
fakeImmediateTaskCategory,
172
>
inclusiveMinTaskKey,
173
>
exclusiveMaxTaskKey,
174
>
rand.Intn(len(immediateTasks)*2)+1,
175
>
)
176
>
s.Equal(immediateTasks, loadedTasks)
177
>
178
>
err := s.ExecutionManager.RangeCompleteHistoryTasks(s.Ctx, &p.RangeCompleteHistoryTasksRequest{
179
>
ShardID: s.ShardID,
180
>
TaskCategory: fakeImmediateTaskCategory,
181
>
InclusiveMinTaskKey: tasks.NewImmediateKey(0),
182
>
ExclusiveMaxTaskKey: tasks.NewImmediateKey(math.MaxInt64),
183
>
})
184
>
s.NoError(err)
185
>
186
>
loadedTasks = s.PaginateTasks(
187
>
fakeImmediateTaskCategory,
188
>
inclusiveMinTaskKey,
189
>
exclusiveMaxTaskKey,
190
>
1,
191
>
)
192
>
s.Empty(loadedTasks)
193
}
194