102 lines
5.3 KiB
TypeScript
102 lines
5.3 KiB
TypeScript
import { test } from '@japa/runner'
|
|
import db from '@adonisjs/lucid/services/db'
|
|
import testUtils from '@adonisjs/core/services/test_utils'
|
|
import { ReleaseJudge } from '#services/release_judge'
|
|
import SearchService from '#services/search_service'
|
|
|
|
test.group('ReleaseJudge', (group) => {
|
|
group.each.setup(() => testUtils.db().withGlobalTransaction())
|
|
|
|
test('judges dead, fake, repeated, locked and unusable releases, and takes a verdict back', async ({ assert }) => {
|
|
let n = 0
|
|
const settled = new Date(Date.now() - 3 * 86400_000)
|
|
const release = async (name: string, extra: Record<string, unknown> = {}) => {
|
|
n++
|
|
const key = `jj${String(n).padStart(6, '0')}`.padEnd(40, '0')
|
|
const [row] = await db
|
|
.table('releases')
|
|
.insert({ release_key: key, identity_key: key, name, source: 'stem', groups: '[]', group_id: 1, files: 3, posted_at: settled, readable: true, settled_at: settled, ...extra })
|
|
.returning('id')
|
|
return Number(row.id)
|
|
}
|
|
let candidate = 9000
|
|
const probe = async (releaseId: number, status: string) => {
|
|
candidate++
|
|
await db.table('probe_candidates').insert({ id: candidate, release_id: releaseId, kind: 'lone', message_id: `<${candidate}@x>`, file: 'x', status })
|
|
return candidate
|
|
}
|
|
const inspect = async (releaseId: number, kind: string, result: Record<string, unknown>) => {
|
|
const id = await probe(releaseId, 'done')
|
|
await db.table('release_inspections').insert({ release_id: releaseId, candidate_id: id, kind, result: JSON.stringify(result) })
|
|
}
|
|
|
|
const dead = await release('Gone.Movie.2020.1080p')
|
|
await probe(dead, 'missing')
|
|
await probe(dead, 'missing')
|
|
const halfDead = await release('Half.Movie.2020.1080p')
|
|
await probe(halfDead, 'missing')
|
|
await probe(halfDead, 'done')
|
|
|
|
const fakeArchive = await release('Das.A-Team.S02E23.Das.Spiel.ist.aus')
|
|
await inspect(fakeArchive, 'archive', { type: 'rar4', names: ['Das.A-Team.S02E23.Das.Spiel.ist.aus.exe'], entries: 1 })
|
|
const fakeFile = await release('Some.Film.2021.1080p.WEB-DL')
|
|
// Only "H.264" and "WEB-DL" say it's media here (a "?" in the pattern never matched).
|
|
const fakeCodec = await release('Some.Show.H.264.WEB-DL')
|
|
await inspect(fakeCodec, 'lone', { type: 'exe', yencName: 'setup.exe' })
|
|
await inspect(fakeFile, 'lone', { type: 'exe', yencName: 'Some.Film.2021.1080p.WEB-DL.mkv' })
|
|
// A program under a program's name is software, not fake.
|
|
const software = await release('LimeWire 4.19.6')
|
|
await inspect(software, 'archive', { type: 'rar4', names: ['LimeWire 4.19.6.exe'], entries: 1 })
|
|
|
|
const hash = 'a'.repeat(32)
|
|
const repeatedIds: number[] = []
|
|
for (let i = 0; i < 20; i++) {
|
|
const id = await release(`kantodo${i}`)
|
|
repeatedIds.push(id)
|
|
await db.table('file_hashes').insert({ candidate_id: 20000 + i, release_id: id, hash16k: hash, size: 1073741824 })
|
|
}
|
|
|
|
const scraped = await release('Jamie Lee Curtis Nude Celebrity Pictures', { poster: '@_CelebMovie_@', source: 'subject' })
|
|
const celeborn = await release('Some.Album.2010', { poster: 'celeborn@not.here (Celeborn)', source: 'subject' })
|
|
const locked = await release('Locked.Thing', { passworded: true })
|
|
|
|
const unlocked = await release('Known.Password', { passworded: true })
|
|
await db.table('release_passwords').insert({ release_id: unlocked, password: 'pw', source: 'spot-text' })
|
|
|
|
const unusable = await release('ab12CD34ef56GH78ij90', { readable: false })
|
|
const fresh = await release('zz12CD34ef56GH78ij90', { readable: false, settled_at: new Date() })
|
|
const spotted = await release('yy12CD34ef56GH78ij90', { readable: false })
|
|
await db.table('release_spots').insert({ release_id: spotted, spot_id: 's@x', files: 1 })
|
|
|
|
const round = await new ReleaseJudge().round()
|
|
const verdicts = new Map((await db.from('release_verdicts').select('release_id', 'rule')).map((v) => [Number(v.release_id), v.rule]))
|
|
assert.equal(verdicts.get(dead), 'dead')
|
|
assert.isUndefined(verdicts.get(halfDead))
|
|
assert.equal(verdicts.get(fakeArchive), 'fake')
|
|
assert.equal(verdicts.get(fakeFile), 'fake')
|
|
assert.equal(verdicts.get(fakeCodec), 'fake')
|
|
assert.isUndefined(verdicts.get(software))
|
|
assert.deepEqual([...new Set(repeatedIds.map((id) => verdicts.get(id)))], ['repeated'])
|
|
assert.equal(verdicts.get(scraped), 'scraped')
|
|
assert.isUndefined(verdicts.get(celeborn))
|
|
assert.equal(verdicts.get(locked), 'locked')
|
|
assert.isUndefined(verdicts.get(unlocked))
|
|
assert.equal(verdicts.get(unusable), 'unusable')
|
|
assert.isUndefined(verdicts.get(fresh))
|
|
assert.isUndefined(verdicts.get(spotted))
|
|
assert.equal(round.rules.repeated.added, 20)
|
|
assert.equal(round.rules.scraped.matched, 1)
|
|
|
|
// Search leaves them out, unless asked.
|
|
const found = await SearchService.search({ q: 'Locked' })
|
|
assert.equal(found.total, 0)
|
|
assert.equal((await SearchService.search({ q: 'Locked', judged: true })).total, 1)
|
|
|
|
// A password found: the next round takes the verdict back.
|
|
await db.table('release_passwords').insert({ release_id: locked, password: 'found', source: 'nzb' })
|
|
const again = await new ReleaseJudge().round()
|
|
assert.equal(again.rules.locked.removed, 1)
|
|
assert.isNull(await db.from('release_verdicts').where('release_id', locked).first())
|
|
})
|
|
})
|