import { toMatchImageSnapshot } from 'jest-image-snapshot';
import { defaultImageSnapshotConfig } from './config';
import { puppeteerTest } from './puppeteerTest';
expect.extend({ toMatchImageSnapshot });
export const imageSnapshot = (customConfig = {}) => {
    const config = { ...defaultImageSnapshotConfig, ...customConfig };
    const { getMatchOptions, getScreenshotOptions, beforeScreenshot, afterScreenshot } = config;
    return puppeteerTest({
        ...config,
        async testBody(page, options) {
            expect.hasAssertions();
            const element = await beforeScreenshot(page, options);
            const image = await (element || page).screenshot(getScreenshotOptions(options));
            await afterScreenshot({ image, context: options.context });
            expect(image).toMatchImageSnapshot(getMatchOptions(options));
        },
    });
};