pdfkit/tests/visual/helpers.js

import PDFDocument from '../../lib/document';
import { pdf2png } from './pdf2png.js';

function runDocTest(options, fn) {
  if (typeof options === 'function') {
    fn = options;
    options = {};
  }
  if (!options.info) {
    options.info = {};
  }

  return new Promise((resolve, reject) => {
    const doc = new PDFDocument(options);
    const buffers = [];

    (async () => {
      await fn(doc);
    })()
      .then(() => {
        doc.on('error', (err) => reject(err));
        doc.on('data', buffers.push.bind(buffers));
        doc.on('end', async () => {
          try {
            const pdfData = Buffer.concat(buffers);
            const { systemFonts = false } = options;
            const images = await pdf2png(pdfData, { systemFonts });
            for (let image of images) {
              expect(image).toMatchImageSnapshot();
            }
            resolve();
          } catch (err) {
            reject(err);
          }
        });
        doc.end();
      })
      .catch((err) => reject(err));
  });
}

export { runDocTest };