47 lines
1.8 KiB
TypeScript
47 lines
1.8 KiB
TypeScript
import { test } from '@japa/runner'
|
|
import db from '@adonisjs/lucid/services/db'
|
|
import testUtils from '@adonisjs/core/services/test_utils'
|
|
import FetchRangeService from '#services/FetchRangeService'
|
|
|
|
async function groupWithRanges(ranges: [number, number, string][]) {
|
|
const now = new Date()
|
|
const [group] = await db
|
|
.table('groups')
|
|
.insert({ name: `test.ranges.${Date.now()}`, active: false, last_indexed_id: 100, last_queued_id: 700, created_at: now, updated_at: now })
|
|
.returning(['id', 'name'])
|
|
for (const [start, end, status] of ranges) {
|
|
await db.table('fetch_ranges').insert({ group_id: group.id, start_id: start, end_id: end, status, created_at: now, updated_at: now })
|
|
}
|
|
return { id: Number(group.id), name: group.name }
|
|
}
|
|
|
|
test.group('FetchRangeService.advanceWatermark', (group) => {
|
|
group.each.setup(() => testUtils.db().withGlobalTransaction())
|
|
|
|
test('stops just below the first range that is not done', async ({ assert }) => {
|
|
const g = await groupWithRanges([
|
|
[101, 200, 'done'],
|
|
[201, 300, 'failed'],
|
|
[301, 400, 'done'],
|
|
[401, 700, 'queued'],
|
|
])
|
|
assert.equal(await FetchRangeService.advanceWatermark(g), 200n)
|
|
// done ranges are forgotten, the gap and the queued range stay
|
|
const left = await db.from('fetch_ranges').where('group_id', g.id).orderBy('start_id').select('status')
|
|
assert.deepEqual(left.map((r) => r.status), ['failed', 'queued'])
|
|
})
|
|
|
|
test('reaches the queue cursor when everything is done', async ({ assert }) => {
|
|
const g = await groupWithRanges([
|
|
[101, 400, 'done'],
|
|
[401, 700, 'done'],
|
|
])
|
|
assert.equal(await FetchRangeService.advanceWatermark(g), 700n)
|
|
})
|
|
|
|
test('never moves backwards', async ({ assert }) => {
|
|
const g = await groupWithRanges([[50, 60, 'failed']])
|
|
assert.equal(await FetchRangeService.advanceWatermark(g), 100n)
|
|
})
|
|
})
|