prefetch.test.js — astro Source File

Source Code

import { expect } from '@playwright/test';
import { testFactory } from './test-utils.js';

const test = testFactory(import.meta.url, {
	root: './fixtures/prefetch/',
});

// Used to track fetch request urls
/** @type {string[]} */
const reqUrls = [];
test.beforeEach(async ({ page }) => {
	page.on('request', (req) => {
		const urlObj = new URL(req.url());
		reqUrls.push(urlObj.pathname + urlObj.search);
	});
});
test.afterEach(() => {
	reqUrls.length = 0;
});

/**
 * Check if url is prefetched via `link[rel="prefetch"]` or `fetch()` (from `reqUrls`)
 * @param {string} url
 * @param {import('@playwright/test').Page} page
 * @param {number} [count] Also expect that it's prefetched this amount of times
 */
async function expectUrlPrefetched(url, page, count) {
	try {
		await expect(page.locator(`link[rel="prefetch"][href$="${url}"]`)).toBeAttached();
	} catch {
		// If link is not found, check if it was fetched via `fetch()`
		expect(reqUrls, `${url} is not prefetched via link or fetch`).toContainEqual(url);
	}

	if (count != null) {
		const linkCount = await page.locator(`link[rel="prefetch"][href$="${url}"]`).count();
		try {
			expect(linkCount).toBe(count);
		} catch {
			const fetchCount = reqUrls.filter((u) => u.includes(url)).length;
			expect(
				fetchCount,
				`${url} should be prefetched ${count} time(s), but is prefetch with link ${linkCount} time(s) and with fetch ${fetchCount} time(s)`,
			).toEqual(count);
		}
	}
}

/**
 * Check if url is not prefetched via `link[rel="prefetch"]` and `fetch()` (from `reqUrls`)
 * @param {string} url
 * @param {import('@playwright/test').Page} page
 */
async function expectUrlNotPrefetched(url, page) {
	await expect(page.locator(`link[rel="prefetch"][href$="${url}"]`)).not.toBeAttached();
	expect(reqUrls).not.toContainEqual(url);
}

/**
 * @param {import('@playwright/test').Page} page
// ... (426 more lines)