Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
245 lines
9.3 KiB
JavaScript
245 lines
9.3 KiB
JavaScript
import { test } from 'node:test';
|
|
import assert from 'node:assert/strict';
|
|
import { processBatch } from '../src/lib/process.js';
|
|
|
|
const NOW = new Date('2026-10-25T10:00:00.000Z');
|
|
const NOW_ISO = NOW.toISOString();
|
|
const HOUR = 3600 * 1000;
|
|
const ago = (hours) => new Date(NOW.getTime() - hours * HOUR).toISOString();
|
|
const SETTINGS = { threshold: 5, minViews: 10000 };
|
|
const ID = 'vid00000001';
|
|
|
|
function memoryDb() {
|
|
const stores = { seen: new Map(), channels: new Map(), outliers: new Map() };
|
|
const getter = (name) => async (keys) =>
|
|
new Map(keys.filter((key) => stores[name].has(key)).map((key) => [key, stores[name].get(key)]));
|
|
const putter = (name, keyField) => async (records) => {
|
|
for (const record of records) stores[name].set(record[keyField], record);
|
|
};
|
|
return {
|
|
stores,
|
|
getSeen: getter('seen'),
|
|
getChannels: getter('channels'),
|
|
getOutliers: getter('outliers'),
|
|
putSeen: putter('seen', 'videoId'),
|
|
putChannels: putter('channels', 'channelId'),
|
|
putOutliers: putter('outliers', 'videoId'),
|
|
};
|
|
}
|
|
|
|
function fakeApi({ videos = [], channels = [] } = {}) {
|
|
const calls = { videos: [], channels: [] };
|
|
return {
|
|
calls,
|
|
fetchVideos: async (ids) => {
|
|
calls.videos.push(ids);
|
|
const found = videos.filter((v) => ids.includes(v.videoId));
|
|
return { videos: found, missingIds: ids.filter((id) => !found.some((v) => v.videoId === id)) };
|
|
},
|
|
fetchChannels: async (ids) => {
|
|
calls.channels.push(ids);
|
|
return channels.filter((c) => ids.includes(c.channelId));
|
|
},
|
|
};
|
|
}
|
|
|
|
const item = (videoId = ID, over = {}) => ({ videoId, vidiqScore: null, viaShortsUrl: false, page: 'home', ...over });
|
|
const video = (videoId = ID, over = {}) => ({
|
|
videoId,
|
|
title: 'Title',
|
|
thumbnailUrl: 'https://i.ytimg.com/t.jpg',
|
|
channelId: 'UC1',
|
|
channelTitle: 'Chan',
|
|
views: 60000,
|
|
publishedAt: '2026-10-01T00:00:00Z',
|
|
durationSec: 600,
|
|
...over,
|
|
});
|
|
const channel = (channelId, subscribers) => ({
|
|
channelId,
|
|
title: 'Chan',
|
|
url: 'https://www.youtube.com/@chan',
|
|
subscribers,
|
|
});
|
|
const run = (items, db, api) => processBatch(items, { settings: SETTINGS, db, api, now: NOW });
|
|
|
|
test('saves a video whose score reaches the threshold', async () => {
|
|
const db = memoryDb();
|
|
const api = fakeApi({ videos: [video()], channels: [channel('UC1', 10000)] });
|
|
|
|
const result = await run([item(ID, { page: 'search' })], db, api);
|
|
|
|
assert.deepEqual(result, { checked: 1, saved: 1 });
|
|
assert.deepEqual(db.stores.outliers.get(ID), {
|
|
videoId: ID,
|
|
title: 'Title',
|
|
url: `https://www.youtube.com/watch?v=${ID}`,
|
|
thumbnailUrl: 'https://i.ytimg.com/t.jpg',
|
|
channelId: 'UC1',
|
|
channelTitle: 'Chan',
|
|
channelUrl: 'https://www.youtube.com/@chan',
|
|
subscribers: 10000,
|
|
views: 60000,
|
|
publishedAt: '2026-10-01T00:00:00Z',
|
|
durationSec: 600,
|
|
isShort: false,
|
|
score: 6,
|
|
vidiqScore: null,
|
|
foundOn: 'search',
|
|
firstSeenAt: NOW_ISO,
|
|
updatedAt: NOW_ISO,
|
|
});
|
|
assert.deepEqual(db.stores.seen.get(ID), { videoId: ID, checkedAt: NOW_ISO, status: 'ok' });
|
|
assert.deepEqual(db.stores.channels.get('UC1'), { ...channel('UC1', 10000), fetchedAt: NOW_ISO });
|
|
});
|
|
|
|
test('marks a below-threshold video as seen without saving it', async () => {
|
|
const db = memoryDb();
|
|
const api = fakeApi({ videos: [video()], channels: [channel('UC1', 100000)] });
|
|
const result = await run([item()], db, api);
|
|
assert.deepEqual(result, { checked: 1, saved: 0 });
|
|
assert.equal(db.stores.outliers.size, 0);
|
|
assert.equal(db.stores.seen.get(ID).status, 'ok');
|
|
});
|
|
|
|
test('does not save a video below minViews even with a huge score', async () => {
|
|
const db = memoryDb();
|
|
const api = fakeApi({ videos: [video(ID, { views: 9000 })], channels: [channel('UC1', 100)] });
|
|
await run([item()], db, api);
|
|
assert.equal(db.stores.outliers.size, 0);
|
|
});
|
|
|
|
test('saves a VidIQ-flagged video even when subscribers are hidden', async () => {
|
|
const db = memoryDb();
|
|
const api = fakeApi({ videos: [video()], channels: [channel('UC1', null)] });
|
|
await run([item(ID, { vidiqScore: 7.5 })], db, api);
|
|
const saved = db.stores.outliers.get(ID);
|
|
assert.equal(saved.score, null);
|
|
assert.equal(saved.subscribers, null);
|
|
assert.equal(saved.vidiqScore, 7.5);
|
|
});
|
|
|
|
test('does not save a video with hidden subscribers and no VidIQ score', async () => {
|
|
const db = memoryDb();
|
|
const api = fakeApi({ videos: [video()], channels: [channel('UC1', null)] });
|
|
await run([item()], db, api);
|
|
assert.equal(db.stores.outliers.size, 0);
|
|
});
|
|
|
|
test('treats a channel missing from the API response as unknown subscribers', async () => {
|
|
const db = memoryDb();
|
|
const api = fakeApi({ videos: [video()], channels: [] });
|
|
await run([item(ID, { vidiqScore: 4 })], db, api);
|
|
assert.equal(db.stores.outliers.get(ID).subscribers, null);
|
|
assert.equal(db.stores.outliers.get(ID).channelUrl, 'https://www.youtube.com/channel/UC1');
|
|
});
|
|
|
|
test('skips ids checked less than 24 hours ago', async () => {
|
|
const db = memoryDb();
|
|
db.stores.seen.set(ID, { videoId: ID, checkedAt: ago(1), status: 'ok' });
|
|
const api = fakeApi({ videos: [video()], channels: [channel('UC1', 10000)] });
|
|
const result = await run([item()], db, api);
|
|
assert.deepEqual(result, { checked: 0, saved: 0 });
|
|
assert.equal(api.calls.videos.length, 0);
|
|
});
|
|
|
|
test('rechecks ids older than 24 hours', async () => {
|
|
const db = memoryDb();
|
|
db.stores.seen.set(ID, { videoId: ID, checkedAt: ago(25), status: 'ok' });
|
|
const api = fakeApi({ videos: [video()], channels: [channel('UC1', 10000)] });
|
|
const result = await run([item()], db, api);
|
|
assert.deepEqual(result, { checked: 1, saved: 1 });
|
|
});
|
|
|
|
test('an item with a VidIQ score bypasses a fresh seen record', async () => {
|
|
const db = memoryDb();
|
|
db.stores.seen.set(ID, { videoId: ID, checkedAt: ago(1), status: 'ok' });
|
|
const api = fakeApi({ videos: [video()], channels: [channel('UC1', 1000000)] });
|
|
const result = await run([item(ID, { vidiqScore: 4 })], db, api);
|
|
assert.deepEqual(result, { checked: 1, saved: 1 });
|
|
assert.equal(db.stores.outliers.get(ID).vidiqScore, 4);
|
|
});
|
|
|
|
test('a repeat sighting refreshes stats and keeps the original metadata', async () => {
|
|
const db = memoryDb();
|
|
db.stores.outliers.set(ID, {
|
|
videoId: ID,
|
|
firstSeenAt: '2026-10-20T00:00:00.000Z',
|
|
foundOn: 'home',
|
|
isShort: true,
|
|
vidiqScore: 9,
|
|
});
|
|
db.stores.seen.set(ID, { videoId: ID, checkedAt: ago(30), status: 'ok' });
|
|
const api = fakeApi({ videos: [video(ID, { views: 70000 })], channels: [channel('UC1', 10000)] });
|
|
|
|
await run([item(ID, { page: 'search', viaShortsUrl: false })], db, api);
|
|
|
|
const saved = db.stores.outliers.get(ID);
|
|
assert.equal(saved.firstSeenAt, '2026-10-20T00:00:00.000Z');
|
|
assert.equal(saved.foundOn, 'home');
|
|
assert.equal(saved.isShort, true);
|
|
assert.equal(saved.vidiqScore, 9);
|
|
assert.equal(saved.views, 70000);
|
|
assert.equal(saved.score, 7);
|
|
assert.equal(saved.updatedAt, NOW_ISO);
|
|
});
|
|
|
|
test('an already saved video is updated even when its score fell below the threshold', async () => {
|
|
const db = memoryDb();
|
|
db.stores.outliers.set(ID, { videoId: ID, firstSeenAt: ago(100), foundOn: 'home', isShort: false, vidiqScore: null });
|
|
db.stores.seen.set(ID, { videoId: ID, checkedAt: ago(30), status: 'ok' });
|
|
const api = fakeApi({ videos: [video()], channels: [channel('UC1', 1000000)] });
|
|
const result = await run([item()], db, api);
|
|
assert.deepEqual(result, { checked: 1, saved: 1 });
|
|
assert.equal(db.stores.outliers.get(ID).score, 0.06);
|
|
});
|
|
|
|
test('a short found via a /shorts/ link is flagged', async () => {
|
|
const db = memoryDb();
|
|
const api = fakeApi({ videos: [video()], channels: [channel('UC1', 10000)] });
|
|
await run([item(ID, { viaShortsUrl: true })], db, api);
|
|
assert.equal(db.stores.outliers.get(ID).isShort, true);
|
|
});
|
|
|
|
test('records unavailable videos in seen and does not save them', async () => {
|
|
const db = memoryDb();
|
|
const api = fakeApi({ videos: [], channels: [] });
|
|
const result = await run([item()], db, api);
|
|
assert.deepEqual(result, { checked: 1, saved: 0 });
|
|
assert.deepEqual(db.stores.seen.get(ID), { videoId: ID, checkedAt: NOW_ISO, status: 'unavailable' });
|
|
assert.equal(db.stores.outliers.size, 0);
|
|
});
|
|
|
|
test('reuses a cached channel younger than 24 hours', async () => {
|
|
const db = memoryDb();
|
|
db.stores.channels.set('UC1', { ...channel('UC1', 10000), fetchedAt: ago(1) });
|
|
const api = fakeApi({ videos: [video()], channels: [] });
|
|
const result = await run([item()], db, api);
|
|
assert.equal(api.calls.channels.length, 0);
|
|
assert.deepEqual(result, { checked: 1, saved: 1 });
|
|
});
|
|
|
|
test('refetches a cached channel older than 24 hours', async () => {
|
|
const db = memoryDb();
|
|
db.stores.channels.set('UC1', { ...channel('UC1', 10000), fetchedAt: ago(25) });
|
|
const api = fakeApi({ videos: [video()], channels: [channel('UC1', 20000)] });
|
|
const result = await run([item()], db, api);
|
|
assert.deepEqual(api.calls.channels, [['UC1']]);
|
|
assert.equal(db.stores.channels.get('UC1').subscribers, 20000);
|
|
assert.deepEqual(result, { checked: 1, saved: 0 }); // score 3 < 5
|
|
});
|
|
|
|
test('an API failure writes nothing and does not mark ids as seen', async () => {
|
|
const db = memoryDb();
|
|
const api = {
|
|
...fakeApi(),
|
|
fetchVideos: async () => {
|
|
throw new Error('boom');
|
|
},
|
|
};
|
|
await assert.rejects(run([item()], db, api), /boom/);
|
|
assert.equal(db.stores.seen.size, 0);
|
|
assert.equal(db.stores.outliers.size, 0);
|
|
assert.equal(db.stores.channels.size, 0);
|
|
});
|