148
}
149
150
>
func (s *matchingTaskSuite) TestInsertSelect_Multiple() {
matching_task.go
151
>
queueID := shuffle.Bytes(testMatchingTaskTaskQueueID)
152
>
taskID := int64(100)
153
>
154
>
task1 := s.newRandomTasksRow(queueID, taskID)
155
>
taskID++
156
>
task2 := s.newRandomTasksRow(queueID, taskID)
157
>
result, err := s.store.InsertIntoTasks(newExecutionContext(), []sqlplugin.TasksRow{task1, task2})
158
>
s.NoError(err)
159
>
rowsAffected, err := result.RowsAffected()
160
>
s.NoError(err)
161
>
s.Equal(2, int(rowsAffected))
162
>
163
>
inclusiveMinTaskID := new(taskID - 1)
164
>
exclusiveMaxTaskID := new(taskID + 1)
165
>
pageSize := new(2)
166
>
filter := sqlplugin.TasksFilter{
167
>
RangeHash: testMatchingTaskRangeHash,
168
>
TaskQueueID: queueID,
169
>
InclusiveMinTaskID: inclusiveMinTaskID,
170
>
ExclusiveMaxTaskID: exclusiveMaxTaskID,
171
>
PageSize: pageSize,
172
>
}
173
>
rows, err := s.store.SelectFromTasks(newExecutionContext(), filter)
174
>
s.NoError(err)
175
>
// fill in some omitted info
176
>
for index := range rows {
177
>
rows[index].RangeHash = testMatchingTaskRangeHash
178
>
rows[index].TaskQueueID = queueID
179
>
}
180
>
s.Equal([]sqlplugin.TasksRow{task1, task2}, rows)
181
}
182