1 files changed, 175 insertions, 3 deletions
diff --git a/packages/integrations/prefetch/test/custom-selectors.test.js b/packages/integrations/prefetch/test/custom-selectors.test.js
index 803e1dc3b..ac15d7d5f 100644
--- a/packages/integrations/prefetch/test/custom-selectors.test.js
+++ b/packages/integrations/prefetch/test/custom-selectors.test.js
@@ -2,11 +2,19 @@ import { expect } from '@playwright/test';
 import { testFactory } from './test-utils.js';
 import prefetch from '../dist/index.js';
 
+const customSelector = 'a[href="/contact"]';
+const customIntentSelector = [
+	'a[href][rel~="custom-intent"]',
+	'a[href][rel~="customer-intent"]',
+	'a[href][rel~="customest-intent"]',
+];
+
 const test = testFactory({
 	root: './fixtures/basic-prefetch/',
 	integrations: [
 		prefetch({
-			selector: 'a[href="/contact"]',
+			selector: customSelector,
+			intentSelector: customIntentSelector,
 		}),
 	],
 });
@@ -50,13 +58,13 @@ test.describe('Custom prefetch selectors', () => {
 	test.describe('build', () => {
 		let previewServer;
 
-		test.beforeAll(async ({ astro }) => {
+		test.beforeEach(async ({ astro }) => {
 			await astro.build();
 			previewServer = await astro.preview();
 		});
 
 		// important: close preview server (free up port and connection)
-		test.afterAll(async () => {
+		test.afterEach(async () => {
 			await previewServer.stop();
 		});
 
@@ -84,3 +92,167 @@ test.describe('Custom prefetch selectors', () => {
 		});
 	});
 });
+
+test.describe('Custom prefetch intent selectors', () => {
+	test.describe('dev', () => {
+		let devServer;
+
+		test.beforeEach(async ({ astro }) => {
+			devServer = await astro.startDevServer();
+		});
+
+		test.afterEach(async () => {
+			await devServer.stop();
+		});
+
+		test('prefetches custom intent links only on hover if provided an array', async ({
+			page,
+			astro,
+		}) => {
+			const requests = [];
+
+			page.on('request', (request) => requests.push(request.url()));
+
+			await page.goto(astro.resolveUrl('/'));
+
+			await page.waitForLoadState('networkidle');
+
+			expect(
+				requests.includes(astro.resolveUrl('/terms')),
+				'/terms was not prefetched initially'
+			).toBeFalsy();
+
+			expect(
+				requests.includes(astro.resolveUrl('/conditions')),
+				'/conditions was not prefetched initially'
+			).toBeFalsy();
+
+			const combinedIntentSelectors = customIntentSelector.join(',');
+			const intentElements = await page.$$(combinedIntentSelectors);
+
+			for (const element of intentElements) {
+				await element.hover();
+			}
+
+			await page.waitForLoadState('networkidle');
+
+			expect(
+				requests.includes(astro.resolveUrl('/terms')),
+				'/terms was prefetched on hover'
+			).toBeTruthy();
+			expect(
+				requests.includes(astro.resolveUrl('/conditions')),
+				'/conditions was prefetched on hover'
+			).toBeTruthy();
+		});
+
+		test('prefetches custom intent links only on hover if provided a string', async ({
+			page,
+			astro,
+		}) => {
+			const requests = [];
+
+			page.on('request', (request) => requests.push(request.url()));
+
+			await page.goto(astro.resolveUrl('/'));
+
+			await page.waitForLoadState('networkidle');
+
+			expect(
+				requests.includes(astro.resolveUrl('/terms')),
+				'/terms was not prefetched initially'
+			).toBeFalsy();
+
+			await page.hover(customIntentSelector[0]);
+
+			await page.waitForLoadState('networkidle');
+
+			expect(
+				requests.includes(astro.resolveUrl('/terms')),
+				'/terms was prefetched on hover'
+			).toBeTruthy();
+		});
+	});
+
+	test.describe('build', () => {
+		let previewServer;
+
+		test.beforeEach(async ({ astro }) => {
+			await astro.build();
+			previewServer = await astro.preview();
+		});
+
+		// important: close preview server (free up port and connection)
+		test.afterEach(async () => {
+			await previewServer.stop();
+		});
+
+		test('prefetches custom intent links only on hover if provided an array', async ({
+			page,
+			astro,
+		}) => {
+			const requests = [];
+
+			page.on('request', (request) => requests.push(request.url()));
+
+			await page.goto(astro.resolveUrl('/'));
+
+			await page.waitForLoadState('networkidle');
+
+			expect(
+				requests.includes(astro.resolveUrl('/terms')),
+				'/terms was not prefetched initially'
+			).toBeFalsy();
+
+			expect(
+				requests.includes(astro.resolveUrl('/conditions')),
+				'/conditions was not prefetched initially'
+			).toBeFalsy();
+
+			const combinedIntentSelectors = customIntentSelector.join(',');
+			const intentElements = await page.$$(combinedIntentSelectors);
+
+			for (const element of intentElements) {
+				await element.hover();
+			}
+
+			await page.waitForLoadState('networkidle');
+
+			expect(
+				requests.includes(astro.resolveUrl('/terms')),
+				'/terms was prefetched on hover'
+			).toBeTruthy();
+			expect(
+				requests.includes(astro.resolveUrl('/conditions')),
+				'/conditions was prefetched on hover'
+			).toBeTruthy();
+		});
+
+		test('prefetches custom intent links only on hover if provided a string', async ({
+			page,
+			astro,
+		}) => {
+			const requests = [];
+
+			page.on('request', (request) => requests.push(request.url()));
+
+			await page.goto(astro.resolveUrl('/'));
+
+			await page.waitForLoadState('networkidle');
+
+			expect(
+				requests.includes(astro.resolveUrl('/terms')),
+				'/terms was not prefetched initially'
+			).toBeFalsy();
+
+			await page.hover(customIntentSelector[0]);
+
+			await page.waitForLoadState('networkidle');
+
+			expect(
+				requests.includes(astro.resolveUrl('/terms')),
+				'/terms was prefetched on hover'
+			).toBeTruthy();
+		});
+	});
+});