db.test.ts

import fs from 'fs';
import path from 'path';
import SQL from 'sql-template-strings';
import { Pool } from 'pg';
import { v4 as uuid } from 'uuid';

import {
  persistAudit,
  retrieveAuditById,
  deleteAuditById,
  retrieveAuditList,
  retrieveAuditCount,
} from './db';
import { Audit } from './models';
import { NotFoundError } from '../../errors';

const LIGHTHOUSE_REPORT_FIXTURE = fs
  .readFileSync(path.join(__dirname, '__fixtures__', 'lighthouse-report.json'))
  .toString();

async function wait(ms: number = 0): Promise<void> {
  await new Promise(r => setTimeout(r, ms));
}

describe('audit db methods', () => {
  let conn: Pool;

  beforeAll(() => {
    conn = new Pool();
  });

  beforeEach(async () => {
    await conn.query(SQL`BEGIN;`);
  });

  afterEach(async () => {
    await conn.query(SQL`ROLLBACK;`);
  });

  afterAll(() => {
    conn.end();
  });

  describe('#persistAudit', () => {
    afterEach(async () => {
      await conn.query(SQL`DELETE FROM lighthouse_audits; COMMIT;`);
    });

    it('creates a new audit', async () => {
      const audit = Audit.buildForUrl('https://spotify.com');
      await persistAudit(conn, audit);
      const resp = await conn.query(
        SQL`SELECT url FROM lighthouse_audits WHERE id = ${audit.id}`,
      );
      expect(resp.rows[0].url).toBe('https://spotify.com');
    });

    it('updates an existing audit', async () => {
      const audit = Audit.buildForUrl('https://spotify.com');
      await persistAudit(conn, audit);
      const resp = await conn.query(
        SQL`SELECT url, report_json FROM lighthouse_audits WHERE id = ${audit.id}`,
      );
      expect(resp.rows[0].url).toBe('https://spotify.com');
      expect(resp.rows[0].report_json).toBeFalsy();

      audit.updateWithReport(JSON.parse(LIGHTHOUSE_REPORT_FIXTURE));
      await persistAudit(conn, audit);
      const resp2 = await conn.query(
        SQL`SELECT url, report_json FROM lighthouse_audits WHERE id = ${audit.id}`,