| 1 | import { afterEach, describe, expect, it, vi } from 'vitest'; |
| 2 | |
| 3 | import { |
| 4 | assertGitHubRepositoryReadable, |
| 5 | listGitHubRepositories, |
| 6 | } from './github.js'; |
| 7 | import { |
| 8 | GitHubNotConnectedError, |
| 9 | GitHubRepositoryNotFoundError, |
| 10 | GitHubUnavailableError, |
| 11 | } from './imports.errors.js'; |
| 12 | |
| 13 | const raw = { full_name: 'octo/repo', private: true, description: null }; |
| 14 | |
| 15 | function answer(response: Response | Error) { |
| 16 | const fetch = |
| 17 | response instanceof Error |
| 18 | ? vi.fn().mockRejectedValue(response) |
| 19 | : vi.fn().mockResolvedValue(response); |
| 20 | vi.stubGlobal('fetch', fetch); |
| 21 | return fetch; |
| 22 | } |
| 23 | |
| 24 | afterEach(() => vi.unstubAllGlobals()); |
| 25 | |
| 26 | describe('assertGitHubRepositoryReadable', () => { |
| 27 | it('passes a readable repository', async () => { |
| 28 | const fetch = answer(Response.json({})); |
| 29 | await assertGitHubRepositoryReadable('tok', 'octo/repo'); |
| 30 | expect(fetch.mock.calls[0][0]).toBe( |
| 31 | 'https://api.github.com/repos/octo/repo', |
| 32 | ); |
| 33 | expect(fetch.mock.calls[0][1].headers.Authorization).toBe('Bearer tok'); |
| 34 | }); |
| 35 | |
| 36 | it('turns a 404 into not found', async () => { |
| 37 | answer(new Response('', { status: 404 })); |
| 38 | await expect( |
| 39 | assertGitHubRepositoryReadable('tok', 'octo/repo'), |
| 40 | ).rejects.toBeInstanceOf(GitHubRepositoryNotFoundError); |
| 41 | }); |
| 42 | |
| 43 | it('asks for GitHub to be linked again when the token is revoked', async () => { |
| 44 | answer(new Response('', { status: 401 })); |
| 45 | await expect( |
| 46 | assertGitHubRepositoryReadable('tok', 'octo/repo'), |
| 47 | ).rejects.toBeInstanceOf(GitHubNotConnectedError); |
| 48 | }); |
| 49 | |
| 50 | it('turns other failures into GitHub being unavailable', async () => { |
| 51 | answer(new Response('', { status: 500 })); |
| 52 | await expect( |
| 53 | assertGitHubRepositoryReadable('tok', 'octo/repo'), |
| 54 | ).rejects.toBeInstanceOf(GitHubUnavailableError); |
| 55 | answer(new Error('ENOTFOUND')); |
| 56 | await expect( |
| 57 | assertGitHubRepositoryReadable('tok', 'octo/repo'), |
| 58 | ).rejects.toBeInstanceOf(GitHubUnavailableError); |
| 59 | }); |
| 60 | }); |
| 61 | |
| 62 | describe('listGitHubRepositories', () => { |
| 63 | const page = (size: number) => |
| 64 | Array.from({ length: size }, (_, index) => ({ |
| 65 | ...raw, |
| 66 | full_name: `octo/repo-${index}`, |
| 67 | })); |
| 68 | |
| 69 | it('lists repositories the user has direct access to, page by page', async () => { |
| 70 | const fetch = vi |
| 71 | .fn() |
| 72 | .mockResolvedValueOnce(Response.json(page(100))) |
| 73 | .mockResolvedValueOnce(Response.json(page(1))); |
| 74 | vi.stubGlobal('fetch', fetch); |
| 75 | |
| 76 | const repositories = await listGitHubRepositories('tok'); |
| 77 | expect(repositories).toHaveLength(101); |
| 78 | expect(repositories[0]).toEqual({ |
| 79 | fullName: 'octo/repo-0', |
| 80 | private: true, |
| 81 | description: null, |
| 82 | }); |
| 83 | expect(fetch.mock.calls[0][0]).toContain( |
| 84 | '/user/repos?sort=pushed&per_page=100&page=1&affiliation=owner,collaborator,organization_member', |
| 85 | ); |
| 86 | expect(fetch.mock.calls[1][0]).toContain('page=2'); |
| 87 | }); |
| 88 | |
| 89 | it('stops after ten pages', async () => { |
| 90 | const fetch = vi |
| 91 | .fn() |
| 92 | .mockImplementation(async () => Response.json(page(100))); |
| 93 | vi.stubGlobal('fetch', fetch); |
| 94 | expect(await listGitHubRepositories('tok')).toHaveLength(1000); |
| 95 | expect(fetch).toHaveBeenCalledTimes(10); |
| 96 | }); |
| 97 | |
| 98 | it('lists nothing when GitHub has nothing', async () => { |
| 99 | answer(new Response('', { status: 404 })); |
| 100 | expect(await listGitHubRepositories('tok')).toEqual([]); |
| 101 | }); |
| 102 | }); |