@accounter/server
Version:
Accounter GraphQL server
188 lines • 8.35 kB
JavaScript
import { beforeEach, describe, expect, it, vi } from 'vitest';
const pgTypedRuntimeMock = vi.hoisted(() => {
const runMocks = {
getStepStatusesRun: vi.fn(),
getStepStatusRun: vi.fn(),
upsertStepStatusRun: vi.fn(),
upsertStep09StatusRun: vi.fn(),
resetStep09ForTemplateRun: vi.fn(),
getBalanceChargeRun: vi.fn(),
};
const sql = vi.fn((strings) => {
const query = strings.join(' ');
if (query.includes('WHERE owner_id = $ownerId AND year = $year AND step_id = $stepId;')) {
return { run: runMocks.getStepStatusRun };
}
if (query.includes('WHERE owner_id = $ownerId AND year = $year;')) {
return { run: runMocks.getStepStatusesRun };
}
if (query.includes("VALUES ($ownerId, $year, '9', 'COMPLETED'")) {
return { run: runMocks.upsertStep09StatusRun };
}
if (query.includes('VALUES ($ownerId, $year, $stepId, $status')) {
return { run: runMocks.upsertStepStatusRun };
}
if (query.includes("evidence_json->>'lockedTemplateName' = $templateName")) {
return { run: runMocks.resetStep09ForTemplateRun };
}
if (query.includes('user_description ILIKE')) {
return { run: runMocks.getBalanceChargeRun };
}
// Catch-all: SQL from transitively-imported modules
return { run: vi.fn().mockResolvedValue([]) };
});
return {
runMocks,
sql,
reset() {
for (const mock of Object.values(runMocks)) {
mock.mockReset();
}
},
};
});
vi.mock('@pgtyped/runtime', () => ({
sql: pgTypedRuntimeMock.sql,
}));
const dynamicReportProviderMock = {
lockTemplate: vi.fn(),
unlockTemplate: vi.fn(),
};
vi.mock('../../reports/providers/dynamic-report.provider.js', () => ({
DynamicReportProvider: class {
lockTemplate = dynamicReportProviderMock.lockTemplate;
unlockTemplate = dynamicReportProviderMock.unlockTemplate;
},
}));
const adminContextProviderMock = {
getVerifiedAdminContext: vi.fn(),
};
vi.mock('../../admin-context/providers/admin-context.provider.js', () => ({
AdminContextProvider: class {
getVerifiedAdminContext = adminContextProviderMock.getVerifiedAdminContext;
},
}));
import { AnnualAuditProvider } from '../annual-audit.provider.js';
function makeStepRow(overrides = {}) {
return {
owner_id: 'owner-1',
year: 2024,
step_id: '9',
status: 'COMPLETED',
notes: null,
evidence_json: null,
updated_at: new Date(),
completed_at: new Date(),
...overrides,
};
}
describe('AnnualAuditProvider — setStep09Status', () => {
let db;
let provider;
beforeEach(() => {
vi.clearAllMocks();
pgTypedRuntimeMock.reset();
db = { query: vi.fn() };
adminContextProviderMock.getVerifiedAdminContext.mockResolvedValue({ ownerId: 'owner-1' });
provider = new AnnualAuditProvider(adminContextProviderMock, dynamicReportProviderMock, db);
});
it('locks the new template and sets step 09 to COMPLETED on first call (no existing)', async () => {
pgTypedRuntimeMock.runMocks.getStepStatusRun.mockResolvedValue([]);
dynamicReportProviderMock.lockTemplate.mockResolvedValue({ name: 'template-A', is_locked: true });
pgTypedRuntimeMock.runMocks.upsertStep09StatusRun.mockResolvedValue([
makeStepRow({ evidence_json: { lockedTemplateName: 'template-A' } }),
]);
const result = await provider.setStep09Status({
ownerId: 'owner-1',
year: 2024,
templateName: 'template-A',
});
expect(dynamicReportProviderMock.unlockTemplate).not.toHaveBeenCalled();
expect(dynamicReportProviderMock.lockTemplate).toHaveBeenCalledWith({
name: 'template-A',
ownerId: 'owner-1',
});
expect(result.status).toBe('COMPLETED');
expect(result.stepId).toBe('9');
expect(result.evidence).toBe(JSON.stringify({ lockedTemplateName: 'template-A' }));
});
it('unlocks the old template and locks the new one on re-selection', async () => {
pgTypedRuntimeMock.runMocks.getStepStatusRun.mockResolvedValue([
makeStepRow({ evidence_json: { lockedTemplateName: 'template-A' } }),
]);
dynamicReportProviderMock.unlockTemplate.mockResolvedValue({ name: 'template-A', is_locked: false });
dynamicReportProviderMock.lockTemplate.mockResolvedValue({ name: 'template-B', is_locked: true });
pgTypedRuntimeMock.runMocks.upsertStep09StatusRun.mockResolvedValue([
makeStepRow({ evidence_json: { lockedTemplateName: 'template-B' } }),
]);
await provider.setStep09Status({
ownerId: 'owner-1',
year: 2024,
templateName: 'template-B',
});
expect(dynamicReportProviderMock.unlockTemplate).toHaveBeenCalledWith({
name: 'template-A',
ownerId: 'owner-1',
});
expect(dynamicReportProviderMock.lockTemplate).toHaveBeenCalledWith({
name: 'template-B',
ownerId: 'owner-1',
});
});
it('does not unlock the old template when re-selecting the same template', async () => {
pgTypedRuntimeMock.runMocks.getStepStatusRun.mockResolvedValue([
makeStepRow({ evidence_json: { lockedTemplateName: 'template-A' } }),
]);
dynamicReportProviderMock.lockTemplate.mockResolvedValue({ name: 'template-A', is_locked: true });
pgTypedRuntimeMock.runMocks.upsertStep09StatusRun.mockResolvedValue([
makeStepRow({ evidence_json: { lockedTemplateName: 'template-A' } }),
]);
await provider.setStep09Status({
ownerId: 'owner-1',
year: 2024,
templateName: 'template-A',
});
expect(dynamicReportProviderMock.unlockTemplate).not.toHaveBeenCalled();
expect(dynamicReportProviderMock.lockTemplate).toHaveBeenCalledWith({
name: 'template-A',
ownerId: 'owner-1',
});
});
it('swallows unlock error when previous template has been deleted', async () => {
pgTypedRuntimeMock.runMocks.getStepStatusRun.mockResolvedValue([
makeStepRow({ evidence_json: { lockedTemplateName: 'deleted-template' } }),
]);
dynamicReportProviderMock.unlockTemplate.mockRejectedValue(new Error('not found'));
dynamicReportProviderMock.lockTemplate.mockResolvedValue({ name: 'template-B', is_locked: true });
pgTypedRuntimeMock.runMocks.upsertStep09StatusRun.mockResolvedValue([
makeStepRow({ evidence_json: { lockedTemplateName: 'template-B' } }),
]);
await expect(provider.setStep09Status({
ownerId: 'owner-1',
year: 2024,
templateName: 'template-B',
})).resolves.toBeDefined();
});
it('throws when upsertStep09Status returns no rows', async () => {
pgTypedRuntimeMock.runMocks.getStepStatusRun.mockResolvedValue([]);
dynamicReportProviderMock.lockTemplate.mockResolvedValue({ name: 'template-A', is_locked: true });
pgTypedRuntimeMock.runMocks.upsertStep09StatusRun.mockResolvedValue([]);
await expect(provider.setStep09Status({ ownerId: 'owner-1', year: 2024, templateName: 'template-A' })).rejects.toThrow('Failed to upsert annual audit step 09 status');
});
});
describe('AnnualAuditProvider — resetStep09ForTemplate', () => {
let db;
let provider;
beforeEach(() => {
vi.clearAllMocks();
pgTypedRuntimeMock.reset();
db = { query: vi.fn() };
provider = new AnnualAuditProvider(adminContextProviderMock, dynamicReportProviderMock, db);
});
it('calls the reset SQL and resolves', async () => {
pgTypedRuntimeMock.runMocks.resetStep09ForTemplateRun.mockResolvedValue([]);
await expect(provider.resetStep09ForTemplate('owner-1', 'template-A')).resolves.toBeUndefined();
expect(pgTypedRuntimeMock.runMocks.resetStep09ForTemplateRun).toHaveBeenCalledWith({ ownerId: 'owner-1', templateName: 'template-A' }, db);
});
});
//# sourceMappingURL=annual-audit-step09.provider.test.js.map