wrangler.test.js — astro Source File

Source Code

import * as assert from 'node:assert/strict';
import { describe, it } from 'node:test';
import {
	cloudflareConfigCustomizer,
	DEFAULT_ASSETS_BINDING_NAME,
	DEFAULT_IMAGES_BINDING_NAME,
	DEFAULT_SESSION_KV_BINDING_NAME,
} from '../dist/wrangler.js';

describe('cloudflareConfigCustomizer', () => {
	describe('main entrypoint', () => {
		it('sets main to the server entrypoint when none exists', () => {
			const customizer = cloudflareConfigCustomizer();
			const result = customizer({});

			assert.equal(result.main, '@astrojs/cloudflare/entrypoints/server');
		});

		it('preserves user main entrypoint', () => {
			const customizer = cloudflareConfigCustomizer();
			const result = customizer({ main: './src/worker.ts' });

			assert.equal(result.main, './src/worker.ts');
		});
	});

	describe('session KV binding', () => {
		it('adds default SESSION KV binding when none exists', () => {
			const customizer = cloudflareConfigCustomizer();
			const result = customizer({});

			assert.deepEqual(result.kv_namespaces, [{ binding: DEFAULT_SESSION_KV_BINDING_NAME }]);
		});

		it('adds custom SESSION KV binding name when specified', () => {
			const customizer = cloudflareConfigCustomizer({ sessionKVBindingName: 'MY_SESSION' });
			const result = customizer({});

			assert.deepEqual(result.kv_namespaces, [{ binding: 'MY_SESSION' }]);
		});

		it('does not add SESSION binding when one already exists with default name', () => {
			const customizer = cloudflareConfigCustomizer();
			const result = customizer({
				kv_namespaces: [{ binding: DEFAULT_SESSION_KV_BINDING_NAME, id: 'existing-id' }],
			});

			assert.equal(result.kv_namespaces, undefined);
		});

		it('does not add SESSION binding when one already exists with custom name', () => {
			const customizer = cloudflareConfigCustomizer({ sessionKVBindingName: 'MY_SESSION' });
			const result = customizer({
				kv_namespaces: [{ binding: 'MY_SESSION', id: 'existing-id' }],
			});

			assert.equal(result.kv_namespaces, undefined);
		});

		it('adds SESSION binding when other KV bindings exist but not the session one', () => {
// ... (74 more lines)