71 lines
2.0 KiB
TypeScript
71 lines
2.0 KiB
TypeScript
import { describe, expect, it } from 'vitest'
|
|
import { LocalStorageBackend } from './storage/local'
|
|
import type { ImageIndexRow } from './image-index'
|
|
import { collectMediaAssetKeys } from './media-assets'
|
|
|
|
const row: ImageIndexRow = {
|
|
key: 'videos/2026/10/example.mp4',
|
|
backend_id: 'local',
|
|
user_id: 1,
|
|
folder: 'videos',
|
|
media_type: 'video',
|
|
original_name: 'example.mp4',
|
|
content_type: 'video/mp4',
|
|
size: 100,
|
|
uploaded_at: '2026-10-17T00:00:00.000Z',
|
|
hls_key: 'videos/2026/10/example/index.m3u8',
|
|
transcode_status: 'done',
|
|
transcode_progress: 100,
|
|
thumb_key: 'videos/2026/10/example.webp',
|
|
source_deleted: 0,
|
|
transcode_preset: 'veryfast',
|
|
transcode_error: null,
|
|
transcode_started_at: null,
|
|
transcode_duration_ms: null
|
|
}
|
|
|
|
async function createBackend() {
|
|
const backend = new LocalStorageBackend()
|
|
const meta = {
|
|
originalName: 'example',
|
|
uploadedAt: row.uploaded_at,
|
|
contentType: 'application/octet-stream',
|
|
size: 0,
|
|
userId: row.user_id,
|
|
mediaType: 'video' as const
|
|
}
|
|
await backend.put(row.key, new Uint8Array([1]), meta)
|
|
await backend.put(row.thumb_key!, new Uint8Array([2]), meta)
|
|
await backend.put(
|
|
row.hls_key!,
|
|
new TextEncoder().encode('#EXTM3U\n#EXTINF:6,\nseg000.ts\n'),
|
|
meta
|
|
)
|
|
await backend.put('videos/2026/10/example/seg000.ts', new Uint8Array([3]), meta)
|
|
return backend
|
|
}
|
|
|
|
describe('collectMediaAssetKeys', () => {
|
|
it('includes source, thumbnail, playlist and HLS segments', async () => {
|
|
const backend = await createBackend()
|
|
expect(new Set(await collectMediaAssetKeys(row, backend))).toEqual(new Set([
|
|
row.key,
|
|
row.thumb_key,
|
|
row.hls_key,
|
|
'videos/2026/10/example/seg000.ts'
|
|
]))
|
|
})
|
|
|
|
it('omits the source when transcoding already removed it', async () => {
|
|
const backend = await createBackend()
|
|
expect(new Set(await collectMediaAssetKeys({
|
|
...row,
|
|
source_deleted: 1
|
|
}, backend))).toEqual(new Set([
|
|
row.thumb_key,
|
|
row.hls_key,
|
|
'videos/2026/10/example/seg000.ts'
|
|
]))
|
|
})
|
|
})
|