astro/packages/integrations/cloudflare/test/test-utils.js
Nate Moore 74df833708
Update wrangler (#8081)
* chore: update wrangler

* chore(cloudflare): update tests to support wrangler@3

* chore(cloudflare): update debug message

* chore: force ci
2023-08-15 12:10:51 -05:00

83 lines
1.9 KiB
JavaScript

import { spawn } from 'node:child_process';
import { fileURLToPath } from 'node:url';
import kill from 'kill-port';
import { loadFixture as baseLoadFixture } from '../../../astro/test/test-utils.js';
export { fixLineEndings } from '../../../astro/test/test-utils.js';
/**
* @typedef {{ ready: Promise<void>, stop: Promise<void> }} WranglerCLI
* @typedef {import('../../../astro/test/test-utils').Fixture} Fixture
*/
export function loadFixture(config) {
if (config?.root) {
config.root = new URL(config.root, import.meta.url);
}
return baseLoadFixture(config);
}
const wranglerPath = fileURLToPath(
new URL('../node_modules/wrangler/bin/wrangler.js', import.meta.url)
);
/**
* @returns {Promise<WranglerCLI>}
*/
export async function runCLI(basePath, { silent, port }) {
// Hack: force existing process on port to be killed
try {
await kill(port, 'tcp');
} catch {
// Will throw if port is not in use, but that's fine
}
const script = fileURLToPath(new URL(`${basePath}/dist/_worker.js`, import.meta.url));
const p = spawn('node', [wranglerPath, 'dev', script, '--port', port, '--log-level', 'info', '--persist-to', `${basePath}/.wrangler/state`]);
p.stderr.setEncoding('utf-8');
p.stdout.setEncoding('utf-8');
const timeout = 10_000;
const ready = new Promise(async (resolve, reject) => {
const failed = setTimeout(
() => {
p.kill();
reject(new Error(`Timed out starting the wrangler CLI`));
},
timeout
);
(async function () {
for (const msg of p.stderr) {
if (!silent) {
console.error(msg);
}
}
})();
for await (const msg of p.stdout) {
if (!silent) {
console.log(msg);
}
if (msg.includes(`[mf:inf] Ready on`)) {
break;
}
}
clearTimeout(failed);
resolve();
});
return {
ready,
stop() {
return new Promise((resolve, reject) => {
p.on('close', () => resolve());
p.on('error', (err) => reject(err));
p.kill();
});
},
};
}