Something went wrong. Try again.
[READ-ONLY] Mirror of https://github.com/just-cameron/tangled-cli. Automation-first CLI for Tangled.org
Something went wrong. Try again.
14 kB · 439 lines
TypeScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440import { beforeEach, describe, expect, it, vi } from 'vitest';import type { TangledApiClient } from '../../src/lib/api-client.js';import { getBacklinks } from '../../src/lib/constellation.js';import { createPull, deletePull, getPull, getPullState, listPulls, resolveSequentialPullNumber,} from '../../src/lib/pulls-api.js';
vi.mock('../../src/lib/constellation.js');
// Mock API client factoryconst createMockClient = (authenticated = true): TangledApiClient => { const mockAgent = { com: { atproto: { repo: { createRecord: vi.fn(), getRecord: vi.fn(), uploadBlob: vi.fn(), }, }, }, };
return { isAuthenticated: vi.fn(() => authenticated), getSession: vi.fn(() => authenticated ? { did: 'did:plc:test123', handle: 'test.bsky.social' } : null ), getAgent: vi.fn(() => mockAgent), } as unknown as TangledApiClient;};
const REPO_DID = 'did:plc:owner';const PULL_AT_URI = 'at://did:plc:test123/sh.tangled.repo.pull/abc123';
describe('createPull', () => { let mockClient: TangledApiClient;
beforeEach(() => { mockClient = createMockClient(true); });
it('should upload blob and create pull record', async () => { const mockBlob = { $type: 'blob', ref: { $link: 'bafyreiabc123' }, mimeType: 'application/gzip', size: 42, }; const mockUploadBlob = vi.fn().mockResolvedValue({ data: { blob: mockBlob } }); const mockCreateRecord = vi.fn().mockResolvedValue({ data: { uri: PULL_AT_URI, cid: 'cid123', }, });
vi.mocked(mockClient.getAgent).mockReturnValue({ com: { atproto: { repo: { uploadBlob: mockUploadBlob, createRecord: mockCreateRecord, }, }, }, } as never);
const patchBuffer = Buffer.from('fake gzip content'); const result = await createPull({ client: mockClient, repoDid: REPO_DID, title: 'Add new feature', body: 'Description', targetBranch: 'main', sourceBranch: 'feature/new-thing', patchBuffer, });
expect(mockUploadBlob).toHaveBeenCalledWith(patchBuffer, { encoding: 'application/gzip' }); expect(mockCreateRecord).toHaveBeenCalledWith({ repo: 'did:plc:test123', collection: 'sh.tangled.repo.pull', record: expect.objectContaining({ $type: 'sh.tangled.repo.pull', target: { repo: REPO_DID, branch: 'main', repoDid: REPO_DID }, title: 'Add new feature', body: 'Description', rounds: [{ createdAt: expect.any(String), patchBlob: mockBlob }], source: { branch: 'feature/new-thing' }, createdAt: expect.any(String), }), });
expect(result).toMatchObject({ uri: PULL_AT_URI, cid: 'cid123', author: 'did:plc:test123', title: 'Add new feature', }); });
it('should create pull without body', async () => { const mockBlob = { $type: 'blob', ref: { $link: 'bafyreiabc123' }, mimeType: 'application/gzip', size: 10, }; const mockUploadBlob = vi.fn().mockResolvedValue({ data: { blob: mockBlob } }); const mockCreateRecord = vi.fn().mockResolvedValue({ data: { uri: PULL_AT_URI, cid: 'cid123' }, });
vi.mocked(mockClient.getAgent).mockReturnValue({ com: { atproto: { repo: { uploadBlob: mockUploadBlob, createRecord: mockCreateRecord } } }, } as never);
const result = await createPull({ client: mockClient, repoDid: REPO_DID, title: 'Fix bug', targetBranch: 'main', sourceBranch: 'fix/bug', patchBuffer: Buffer.from('patch'), });
expect(result.body).toBeUndefined(); expect(result.title).toBe('Fix bug'); });
it('should throw when not authenticated', async () => { const unauthClient = createMockClient(false); await expect( createPull({ client: unauthClient, repoDid: REPO_DID, title: 'Test', targetBranch: 'main', sourceBranch: 'feature', patchBuffer: Buffer.from('patch'), }) ).rejects.toThrow(); });});
describe('listPulls', () => { let mockClient: TangledApiClient;
beforeEach(() => { mockClient = createMockClient(true); vi.mocked(getBacklinks).mockResolvedValue({ total: 0, records: [], cursor: null, }); });
it('should return empty list when no pulls exist', async () => { const result = await listPulls({ client: mockClient, repoDid: REPO_DID }); expect(result.pulls).toHaveLength(0); expect(result.cursor).toBeUndefined(); });
it('should query constellation with correct parameters', async () => { await listPulls({ client: mockClient, repoDid: REPO_DID, limit: 25 }); expect(getBacklinks).toHaveBeenCalledWith( REPO_DID, 'sh.tangled.repo.pull', '.target.repo', 25, undefined ); });
it('should fetch records from backlinks and return pulls', async () => { vi.mocked(getBacklinks).mockResolvedValue({ total: 1, records: [{ did: 'did:plc:test123', collection: 'sh.tangled.repo.pull', rkey: 'abc123' }], cursor: null, });
const mockRecord = { $type: 'sh.tangled.repo.pull', target: { repo: REPO_DID, branch: 'main' }, title: 'Test PR', rounds: [{ createdAt: '2024-01-01T00:00:00.000Z', patchBlob: {} }], source: { branch: 'feature', sha: 'abc', repo: REPO_DID }, createdAt: '2024-01-01T00:00:00.000Z', };
const mockGetRecord = vi.fn().mockResolvedValue({ data: { value: mockRecord, uri: PULL_AT_URI, cid: 'cid123' }, });
vi.mocked(mockClient.getAgent).mockReturnValue({ com: { atproto: { repo: { getRecord: mockGetRecord } } }, } as never);
const result = await listPulls({ client: mockClient, repoDid: REPO_DID }); expect(result.pulls).toHaveLength(1); expect(result.pulls[0].title).toBe('Test PR'); expect(result.pulls[0].uri).toBe(PULL_AT_URI); expect(result.pulls[0].author).toBe('did:plc:test123'); });});
describe('getPull', () => { let mockClient: TangledApiClient;
beforeEach(() => { mockClient = createMockClient(true); });
it('should fetch pull record by AT-URI', async () => { const mockRecord = { $type: 'sh.tangled.repo.pull', target: { repo: REPO_DID, branch: 'main' }, title: 'Test PR', rounds: [{ createdAt: '2024-01-01T00:00:00.000Z', patchBlob: {} }], createdAt: '2024-01-01T00:00:00.000Z', };
const mockGetRecord = vi.fn().mockResolvedValue({ data: { value: mockRecord, uri: PULL_AT_URI, cid: 'cid123' }, });
vi.mocked(mockClient.getAgent).mockReturnValue({ com: { atproto: { repo: { getRecord: mockGetRecord } } }, } as never);
const result = await getPull({ client: mockClient, pullUri: PULL_AT_URI }); expect(result.title).toBe('Test PR'); expect(result.uri).toBe(PULL_AT_URI); expect(result.author).toBe('did:plc:test123'); expect(mockGetRecord).toHaveBeenCalledWith({ repo: 'did:plc:test123', collection: 'sh.tangled.repo.pull', rkey: 'abc123', }); });
it('should throw for invalid AT-URI', async () => { await expect(getPull({ client: mockClient, pullUri: 'not-a-uri' })).rejects.toThrow( 'Invalid pull request AT-URI' ); });});
describe('deletePull', () => { let mockClient: TangledApiClient;
beforeEach(() => { mockClient = createMockClient(true); });
it('should delete the record when the caller is the author', async () => { const mockDeleteRecord = vi.fn().mockResolvedValue({}); vi.mocked(mockClient.getAgent).mockReturnValue({ com: { atproto: { repo: { deleteRecord: mockDeleteRecord } } }, } as never);
await deletePull({ client: mockClient, pullUri: PULL_AT_URI });
expect(mockDeleteRecord).toHaveBeenCalledWith({ repo: 'did:plc:test123', collection: 'sh.tangled.repo.pull', rkey: 'abc123', }); });
it('should throw when the caller is not the author', async () => { const otherAuthorUri = 'at://did:plc:someoneelse/sh.tangled.repo.pull/abc123'; await expect(deletePull({ client: mockClient, pullUri: otherAuthorUri })).rejects.toThrow( 'Cannot delete pull request: you are not the author' ); });
it('should throw for invalid AT-URI', async () => { await expect(deletePull({ client: mockClient, pullUri: 'not-a-uri' })).rejects.toThrow( 'Invalid pull request AT-URI' ); });});
describe('getPullState', () => { let mockClient: TangledApiClient;
beforeEach(() => { mockClient = createMockClient(true); vi.mocked(getBacklinks).mockResolvedValue({ total: 0, records: [], cursor: null }); });
it('should return open when no state records exist', async () => { const state = await getPullState({ client: mockClient, pullUri: PULL_AT_URI }); expect(state).toBe('open'); expect(getBacklinks).toHaveBeenCalledWith( PULL_AT_URI, 'sh.tangled.repo.pull.status', '.pull', 100 ); });
it('should return closed for closed status', async () => { vi.mocked(getBacklinks).mockResolvedValue({ total: 1, records: [ { did: 'did:plc:test123', collection: 'sh.tangled.repo.pull.status', rkey: 'rkey1' }, ], cursor: null, }); const mockGetRecord = vi.fn().mockResolvedValue({ data: { value: { status: 'sh.tangled.repo.pull.status.closed' }, uri: '', cid: '' }, }); vi.mocked(mockClient.getAgent).mockReturnValue({ com: { atproto: { repo: { getRecord: mockGetRecord } } }, } as never);
const state = await getPullState({ client: mockClient, pullUri: PULL_AT_URI }); expect(state).toBe('closed'); });
it('should return merged for merged status', async () => { vi.mocked(getBacklinks).mockResolvedValue({ total: 1, records: [ { did: 'did:plc:test123', collection: 'sh.tangled.repo.pull.status', rkey: 'rkey1' }, ], cursor: null, }); const mockGetRecord = vi.fn().mockResolvedValue({ data: { value: { status: 'sh.tangled.repo.pull.status.merged' }, uri: '', cid: '' }, }); vi.mocked(mockClient.getAgent).mockReturnValue({ com: { atproto: { repo: { getRecord: mockGetRecord } } }, } as never);
const state = await getPullState({ client: mockClient, pullUri: PULL_AT_URI }); expect(state).toBe('merged'); });
it('should use latest rkey when multiple state records exist', async () => { vi.mocked(getBacklinks).mockResolvedValue({ total: 2, records: [ { did: 'did:plc:test123', collection: 'sh.tangled.repo.pull.status', rkey: 'rkey2' }, { did: 'did:plc:test123', collection: 'sh.tangled.repo.pull.status', rkey: 'rkey1' }, ], cursor: null, }); const mockGetRecord = vi .fn() .mockResolvedValueOnce({ data: { value: { status: 'sh.tangled.repo.pull.status.closed' }, uri: '', cid: '' }, }) .mockResolvedValueOnce({ data: { value: { status: 'sh.tangled.repo.pull.status.open' }, uri: '', cid: '' }, }); vi.mocked(mockClient.getAgent).mockReturnValue({ com: { atproto: { repo: { getRecord: mockGetRecord } } }, } as never);
// rkey2 > rkey1 alphabetically, so rkey2 (closed) should win const state = await getPullState({ client: mockClient, pullUri: PULL_AT_URI }); expect(state).toBe('closed'); });});
describe('resolveSequentialPullNumber', () => { let mockClient: TangledApiClient;
beforeEach(() => { mockClient = createMockClient(true); vi.mocked(getBacklinks).mockClear(); vi.mocked(getBacklinks).mockResolvedValue({ total: 0, records: [], cursor: null }); });
it('should use fast path for #N displayId', async () => { const num = await resolveSequentialPullNumber('#3', PULL_AT_URI, mockClient, REPO_DID); expect(num).toBe(3); expect(getBacklinks).not.toHaveBeenCalled(); });
it('should scan pulls when displayId is not #N', async () => { const pullUri1 = 'at://did:plc:test123/sh.tangled.repo.pull/rkey1'; const pullUri2 = 'at://did:plc:test123/sh.tangled.repo.pull/rkey2';
vi.mocked(getBacklinks).mockResolvedValue({ total: 2, records: [ { did: 'did:plc:test123', collection: 'sh.tangled.repo.pull', rkey: 'rkey1' }, { did: 'did:plc:test123', collection: 'sh.tangled.repo.pull', rkey: 'rkey2' }, ], cursor: null, });
const mockGetRecord = vi .fn() .mockResolvedValueOnce({ data: { value: { $type: 'sh.tangled.repo.pull', target: { repo: REPO_DID, branch: 'main' }, title: 'First', rounds: [{ createdAt: '2024-01-01T00:00:00.000Z', patchBlob: {} }], createdAt: '2024-01-01T00:00:00.000Z', }, uri: pullUri1, cid: 'cid1', }, }) .mockResolvedValueOnce({ data: { value: { $type: 'sh.tangled.repo.pull', target: { repo: REPO_DID, branch: 'main' }, title: 'Second', rounds: [{ createdAt: '2024-01-01T00:00:00.000Z', patchBlob: {} }], createdAt: '2024-01-02T00:00:00.000Z', }, uri: pullUri2, cid: 'cid2', }, });
vi.mocked(mockClient.getAgent).mockReturnValue({ com: { atproto: { repo: { getRecord: mockGetRecord } } }, } as never);
const num = await resolveSequentialPullNumber('rkey2', pullUri2, mockClient, REPO_DID); expect(num).toBe(2); });});