paperclip/.github/scripts/tests/fetch-pr-files.test.mjs

38 lines
1.3 KiB
JavaScript

import { test } from 'node:test';
import assert from 'node:assert/strict';
import { fetchAllPullRequestFiles } from '../fetch-pr-files.mjs';
test('fetchAllPullRequestFiles: returns a single short page', async () => {
const seenPaths = [];
const files = await fetchAllPullRequestFiles(async (path) => {
seenPaths.push(path);
return [{ filename: 'src/only.ts' }];
}, 'paperclipai/paperclip', 6469, 'token');
assert.deepEqual(seenPaths, [
'/repos/paperclipai/paperclip/pulls/6469/files?per_page=100&page=1',
]);
assert.equal(files.length, 1);
});
test('fetchAllPullRequestFiles: keeps fetching when a page is full', async () => {
const seenPaths = [];
const files = await fetchAllPullRequestFiles(async (path) => {
seenPaths.push(path);
const page = Number(new URL(`https://github.com${path}`).searchParams.get('page'));
if (page === 1) {
return Array.from({ length: 100 }, (_, index) => ({
filename: `src/file-${index + 1}.ts`,
}));
}
return [{ filename: 'src/file-101.ts' }];
}, 'paperclipai/paperclip', 6469, 'token');
assert.deepEqual(seenPaths, [
'/repos/paperclipai/paperclip/pulls/6469/files?per_page=100&page=1',
'/repos/paperclipai/paperclip/pulls/6469/files?per_page=100&page=2',
]);
assert.equal(files.length, 101);
assert.equal(files.at(-1)?.filename, 'src/file-101.ts');
});