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 => {
    const doc = new PDFDocument(options);
    const buffers = [];

    fn(doc);

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

export { runDocTest };