1 files changed, 59 insertions, 0 deletions
diff --git a/packages/astro/test/custom-404-locals.test.js b/packages/astro/test/custom-404-locals.test.js
new file mode 100644
index 000000000..246e9e31e
--- /dev/null
+++ b/packages/astro/test/custom-404-locals.test.js
@@ -0,0 +1,59 @@
+import assert from 'node:assert/strict';
+import { after, before, describe, it } from 'node:test';
+import * as cheerio from 'cheerio';
+import { loadFixture } from './test-utils.js';
+
+describe('Custom 404 locals', () => {
+	let fixture;
+
+	before(async () => {
+		fixture = await loadFixture({
+			root: './fixtures/custom-404-locals/',
+			site: 'http://example.com',
+		});
+	});
+ 
+	describe('dev', () => {
+		let devServer;
+		let $;
+
+		before(async () => {
+			devServer = await fixture.startDevServer();
+		});
+
+		after(async () => {
+			await devServer.stop();
+		});
+
+		it('renders /', async () => {
+			const html = await fixture.fetch('/').then((res) => res.text());
+			$ = cheerio.load(html);
+
+			assert.equal($('h1').text(), 'Home');
+		});
+
+		it('renders 404 for /a', async () => {
+			const res = await fixture.fetch('/a');
+			assert.equal(res.status, 404);
+
+			const html = await res.text();
+			$ = cheerio.load(html);
+
+			assert.equal($('h1').text(), 'Page not found');
+			assert.equal($('p.message').text(), 'This 404 is a dynamic HTML file.');
+			assert.equal($('p.runtime').text(), 'locals');
+		});
+
+		it('renders 404 for /404-return', async () => {
+			const res = await fixture.fetch('/404-return');
+			assert.equal(res.status, 404);
+
+			const html = await res.text();
+			$ = cheerio.load(html);
+
+			assert.equal($('h1').text(), 'Page not found');
+			assert.equal($('p.message').text(), 'This 404 is a dynamic HTML file.');
+			assert.equal($('p.runtime').text(), 'locals');
+		});
+	});
+});