232
}
233
235
>
shardID := rand.Int31()
236
>
taskID := int64(1)
237
>
238
>
task := s.newRandomVisibilityTaskRow(shardID, taskID)
239
>
result, err := s.store.InsertIntoVisibilityTasks(newExecutionContext(), []sqlplugin.VisibilityTasksRow{task})
240
>
s.NoError(err)
241
>
rowsAffected, err := result.RowsAffected()
242
>
s.NoError(err)
243
>
s.Equal(1, int(rowsAffected))
244
>
245
>
filter := sqlplugin.VisibilityTasksFilter{
246
>
ShardID: shardID,
247
>
TaskID: taskID,
248
>
}
249
>
result, err = s.store.DeleteFromVisibilityTasks(newExecutionContext(), filter)
250
>
s.NoError(err)
251
>
rowsAffected, err = result.RowsAffected()
252
>
s.NoError(err)
253
>
s.Equal(1, int(rowsAffected))
254
>
255
>
rangeFilter := sqlplugin.VisibilityTasksRangeFilter{
256
>
ShardID: shardID,
257
>
InclusiveMinTaskID: taskID,
258
>
ExclusiveMaxTaskID: taskID + 1,
259
>
PageSize: 1,
260
>
}
261
>
rows, err := s.store.RangeSelectFromVisibilityTasks(newExecutionContext(), rangeFilter)
262
>
s.NoError(err)
263
>
for index := range rows {
264
rows[index].ShardID = shardID
265
}
267
}
268