import { expect, test, Snapshots } from "vitest"; 

declare module 'vitest' {
		interface Assertion<T = any> {
				toMatchCustomSnapshot(snapshot?: string): T;
				toMatchAsyncCustomSnapshot(snapshot?: string): Promise<T>;
		}
}

function syncResult(value: string): string {
		return value;
}

function asyncResult(value: string): Promise<string> {
		return Promise.resolve(value); 
}

expect.extend({
		toMatchCustomSnapshot(received: string, inlineSnapshot?: string) {
				const snapshot = syncResult(received);
				return Snapshots.toMatchInlineSnapshot.call(this, snapshot, inlineSnapshot);
		},
		async toMatchAsyncCustomSnapshot(received: string, inlineSnapshot?: string) {
				const snapshot = await asyncResult(received);
				return Snapshots.toMatchInlineSnapshot.call(this, snapshot, inlineSnapshot);
		},
});

test('sync snapshot', () => {
		expect('foo').toMatchCustomSnapshot(`"foo"`);
});

test('async snapshot', async () => {
		await expect('foo').toMatchAsyncCustomSnapshot(`"foo"`);
});