1 files changed, 119 insertions, 0 deletions
diff --git a/packages/astro/test/astro-mode.test.js b/packages/astro/test/astro-mode.test.js
new file mode 100644
index 000000000..09585831a
--- /dev/null
+++ b/packages/astro/test/astro-mode.test.js
@@ -0,0 +1,119 @@
+import assert from 'node:assert/strict';
+import { after, afterEach, before, describe, it } from 'node:test';
+import * as cheerio from 'cheerio';
+import { loadFixture } from './test-utils.js';
+
+describe('--mode', () => {
+	/** @type {import('./test-utils.js').Fixture} */
+	let fixture;
+
+	before(async () => {
+		fixture = await loadFixture({
+			root: './fixtures/astro-mode/',
+		});
+	});
+
+	afterEach(() => {
+		// Reset so it doesn't interfere with other tests as builds below
+		// will interact with env variables loaded from .env files
+		delete process.env.NODE_ENV;
+		// `astro:env` writes to `process.env` currently which should be avoided,
+		// otherwise consecutive builds can't get the latest `.env` values. Workaround
+		// here for now be resetting it.
+		delete process.env.TITLE;
+	});
+
+	describe('build', () => {
+		before(async () => {
+			await fixture.build();
+		});
+
+		it('works', async () => {
+			const html = await fixture.readFile('/index.html');
+			const $ = cheerio.load(html);
+			assert.equal($('#env-mode').text(), 'production');
+			assert.equal($('#env-dev').text(), 'false');
+			assert.equal($('#env-prod').text(), 'true');
+			assert.equal($('#env-title').text(), 'production');
+			assert.equal($('#env-astro-title').text(), 'production');
+		});
+	});
+
+	describe('build --mode testing --devOutput', () => {
+		before(async () => {
+			await fixture.build({ mode: 'testing' }, { devOutput: true });
+		});
+
+		it('works', async () => {
+			const html = await fixture.readFile('/index.html');
+			const $ = cheerio.load(html);
+			assert.equal($('#env-mode').text(), 'testing');
+			assert.equal($('#env-dev').text(), 'true');
+			assert.equal($('#env-prod').text(), 'false');
+			assert.equal($('#env-title').text(), '');
+			assert.equal($('#env-astro-title').text(), 'unset');
+		});
+	});
+
+	describe('build --mode staging', () => {
+		before(async () => {
+			await fixture.build({ mode: 'staging' });
+		});
+
+		it('works', async () => {
+			const html = await fixture.readFile('/index.html');
+			const $ = cheerio.load(html);
+			assert.equal($('#env-mode').text(), 'staging');
+			assert.equal($('#env-dev').text(), 'false');
+			assert.equal($('#env-prod').text(), 'true');
+			assert.equal($('#env-title').text(), 'staging');
+			assert.equal($('#env-astro-title').text(), 'staging');
+		});
+	});
+
+	describe('dev', () => {
+		/** @type {import('./test-utils.js').DevServer} */
+		let devServer;
+		before(async () => {
+			devServer = await fixture.startDevServer();
+		});
+		after(async () => {
+			await devServer.stop();
+		});
+
+		it('works', async () => {
+			const res = await fixture.fetch('/');
+			assert.equal(res.status, 200);
+			const html = await res.text();
+			const $ = cheerio.load(html);
+			assert.equal($('#env-mode').text(), 'development');
+			assert.equal($('#env-dev').text(), 'true');
+			assert.equal($('#env-prod').text(), 'false');
+			assert.equal($('#env-title').text(), 'development');
+			assert.equal($('#env-astro-title').text(), 'development');
+		});
+	});
+
+	describe('dev --mode develop', () => {
+		/** @type {import('./test-utils.js').DevServer} */
+		let devServer;
+		before(async () => {
+			devServer = await fixture.startDevServer({ mode: 'develop' });
+		});
+		after(async () => {
+			await devServer.stop();
+		});
+
+		it('works', async () => {
+			const res = await fixture.fetch('/');
+			assert.equal(res.status, 200);
+			const html = await res.text();
+			const $ = cheerio.load(html);
+			assert.equal($('#env-mode').text(), 'develop');
+			assert.equal($('#env-dev').text(), 'true');
+			assert.equal($('#env-prod').text(), 'false');
+			assert.equal($('#env-title').text(), '');
+			assert.equal($('#env-astro-title').text(), 'unset');
+		});
+	});
+});