57 lines
2.6 KiB
TypeScript
57 lines
2.6 KiB
TypeScript
import { test } from '@japa/runner'
|
|
import db from '@adonisjs/lucid/services/db'
|
|
import testUtils from '@adonisjs/core/services/test_utils'
|
|
import Par2Service from '#services/Par2Service'
|
|
import type { Par2Set } from '#lib/par2/index'
|
|
|
|
const VOLUME = 10_000_952 // a standard split-volume size that recurs across posts
|
|
|
|
async function release(key: string, name: string) {
|
|
const now = new Date()
|
|
const [row] = await db
|
|
.table('releases')
|
|
.insert({ release_key: key.padEnd(40, '0'), name, source: 'weak-stem', groups: '[]', posted_at: now, first_seen_at: now, last_seen_at: now })
|
|
.returning('id')
|
|
return Number(row.id)
|
|
}
|
|
|
|
async function binary(key: string, fields: Record<string, unknown>) {
|
|
const now = new Date()
|
|
const [row] = await db
|
|
.table('binaries')
|
|
.insert({ file_key: key.padEnd(40, '0'), poster: 'x', subject: '"x" yEnc', name: 'Xq8r2KdLm3Zw91PqA7', total_parts: 14, groups: '[]', posted_at: now, first_seen_at: now, last_seen_at: now, ...fields })
|
|
.returning(['id', 'poster', 'groups', 'posted_at'])
|
|
return row
|
|
}
|
|
|
|
function set(id: string, files: { name: string; md5_16k: string }[]): Par2Set {
|
|
return {
|
|
recoverySetId: id.padEnd(32, '0'),
|
|
files: files.map((f, i) => ({ fileId: `${id}${i}`.padEnd(32, '0'), md5: '0'.repeat(32), md5_16k: f.md5_16k, length: VOLUME, name: f.name })),
|
|
}
|
|
}
|
|
|
|
test.group('Par2Service.attach', (group) => {
|
|
group.each.setup(() => testUtils.db().withGlobalTransaction())
|
|
|
|
test('a set does not join another post just because a size matches', async ({ assert }) => {
|
|
const other = await release('other', 'OtherPost')
|
|
await binary('other-vol', { size: VOLUME, md5_16k: 'a'.repeat(32), release_id: other })
|
|
const par2 = await binary('par2-b', { size: 50_000, total_parts: 1 })
|
|
|
|
const result = await Par2Service.attach(par2, set('b', [{ name: 'PostB.part1.rar', md5_16k: 'b'.repeat(32) }]))
|
|
assert.notEqual(result.releaseId, other)
|
|
assert.equal(await db.from('release_files').where('release_id', other).count('* as n').first().then((r) => Number(r.n)), 0)
|
|
})
|
|
|
|
test('an exactly matching file pulls the set into its release and gets its real name', async ({ assert }) => {
|
|
const stemRelease = await release('stem', 'Xq8r2KdLm3Zw91PqA7')
|
|
const vol = await binary('vol', { size: VOLUME, md5_16k: 'c'.repeat(32), release_id: stemRelease })
|
|
const par2 = await binary('par2-c', { size: 50_000, total_parts: 1 })
|
|
|
|
const result = await Par2Service.attach(par2, set('c', [{ name: 'Real.Name.part1.rar', md5_16k: 'c'.repeat(32) }]))
|
|
assert.equal(result.releaseId, stemRelease)
|
|
assert.equal((await db.from('binaries').where('id', vol.id).first()).real_name, 'Real.Name.part1.rar')
|
|
})
|
|
})
|