116
}
117
119
>
shardID := rand.Int31()
120
>
taskID := int64(1)
121
>
122
>
task := s.newRandomTransferTaskRow(shardID, taskID)
123
>
result, err := s.store.InsertIntoTransferTasks(newExecutionContext(), []sqlplugin.TransferTasksRow{task})
124
>
s.NoError(err)
125
>
rowsAffected, err := result.RowsAffected()
126
>
s.NoError(err)
127
>
s.Equal(1, int(rowsAffected))
128
>
129
>
rangeFilter := sqlplugin.TransferTasksRangeFilter{
130
>
ShardID: shardID,
131
>
InclusiveMinTaskID: taskID,
132
>
ExclusiveMaxTaskID: taskID + 1,
133
>
PageSize: 1,
134
>
}
135
>
rows, err := s.store.RangeSelectFromTransferTasks(newExecutionContext(), rangeFilter)
136
>
s.NoError(err)
137
>
for index := range rows {
138
>
rows[index].ShardID = shardID
139
>
}
140
>
s.Equal([]sqlplugin.TransferTasksRow{task}, rows)
141
}
142