Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
JavaScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345import assert from "node:assert/strict";import { createHash } from "node:crypto";import { readFile } from "node:fs/promises";import { test } from "node:test";import { unstable_dev } from "wrangler";import { parse } from "jsonc-parser";import { customerBindings } from "./fixtures/config.mjs";import { loadArtifact, verifyUpgradeIdentity,} from "../control-plane/artifact.ts";
const readJson = async (path) => JSON.parse(await readFile(path, "utf8"));const digest = (bytes) => createHash("sha256").update(bytes).digest("hex");
test("release contains intact Worker, assets and a consistent SQLite lifecycle", async () => { const manifestBytes = await readFile("dist/release/manifest.json"); assert.equal( digest(manifestBytes), (await readFile("dist/release/manifest.sha256", "utf8")).trim(), ); const manifest = JSON.parse(manifestBytes); const config = await readJson("dist/release/deployment.json"); const source = parse(await readFile("wrangler.jsonc", "utf8"), [], { allowTrailingComma: true, }); assert.equal(manifest.schemaVersion, 1); assert.equal(manifest.release, (await readJson("package.json")).version); assert.equal( manifest.lockfileSha256, digest(await readFile("pnpm-lock.yaml")), ); assert.equal(config.no_bundle, true); assert.deepEqual(config.compatibility_flags, [ "nodejs_compat", "global_fetch_strictly_public", ]); assert.equal(config.name, undefined); assert.equal(config.account_id, undefined); assert.equal(config.vars, undefined); assert.equal(config.env, undefined); assert.equal(config.migrations, undefined); assert.deepEqual(config.exports, { PersonalAgent: { type: "durable-object", storage: "sqlite" }, Sandbox: { type: "durable-object", storage: "sqlite" }, }); assert.deepEqual(config.durable_objects, source.durable_objects); assert.deepEqual(config.containers, source.containers); assert.equal(config.containers[0].image, manifest.shell.image); assert.match(manifest.shell.image, /0\.12\.9@sha256:[0-9a-f]{64}$/); assert.equal(config.containers[0].class_name, manifest.shell.className); assert.equal(config.containers[0].max_instances, manifest.shell.maxInstances); assert.equal(config.containers[0].instance_type, manifest.shell.instanceType); assert.equal( config.durable_objects.bindings[0].class_name, manifest.identity.durableObjectClass, ); assert.equal( config.durable_objects.bindings[0].name, manifest.identity.durableObjectBinding, ); assert.deepEqual(config.ai, { binding: "AI" }); assert.equal(config.browser.binding, "BROWSER"); assert.deepEqual(config.worker_loaders, [{ binding: "LOADER" }]); assert.ok( manifest.files.some( ({ path }) => path === config.main.replace(/^\.\//, ""), ), ); assert.ok( manifest.files.some( ({ path }) => path.startsWith("assets/") && path.endsWith(".js"), ), ); for (const file of manifest.files) { assert.match(file.path, /^(worker\/|assets\/|deployment\.json$)/); assert.ok(!file.path.split("/").includes("..")); const bytes = await readFile(`dist/release/${file.path}`); assert.equal(bytes.length, file.size, file.path); assert.equal(digest(bytes), file.sha256, file.path); assert.ok( !bytes.includes(customerBindings.FLAREBOT_SESSION_SECRET), file.path, ); if (file.path.startsWith("assets/")) { assert.ok( !bytes.includes("FLAREBOT_SESSION_SECRET"), "server configuration must not enter browser assets", ); } if (file.path.startsWith("worker/")) { assert.ok( !bytes.includes("__golden__") && !bytes.includes("TEST_RUNNER_PORT"), "golden-path fixture routes and boundaries must not enter the release", ); assert.ok( !bytes.includes("fixtureEcho"), "test tools must not enter the release", ); assert.ok( !bytes.includes("MockLanguageModel"), "test model must not enter the release", ); assert.ok( !bytes.includes("react.production"), "React runtime must not enter the Worker", ); assert.ok( !bytes.includes("FLAREBOT_OAUTH_CLIENT_SECRET"), "control-plane secret loader must not enter customer Worker", ); } }});
test("public fetch release retains legacy artifacts and requires an explicit forward edge", async () => { const source = parse(await readFile("wrangler.jsonc", "utf8")); const publisher = parse( await readFile("wrangler.control-plane.jsonc", "utf8"), ); const contract = await readJson("deployment/manifest.json"); assert.deepEqual(source.compatibility_flags, [ "nodejs_compat", "global_fetch_strictly_public", ]); assert.deepEqual(publisher.compatibility_flags, source.compatibility_flags); assert.equal((await readJson("package.json")).version, "0.1.0-dev.16"); assert.ok( contract.compatibility.fromArtifacts.includes( "cd703c946a3865211efe98e6e68e7d4b875ac6f825db8ad03711e5adf7c52843", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "fe4c23a4371d331dd7fe2c9c15a6f236bef06324cf999a998b27e832f12b1a31", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "ac05648bb6b8fe93d46018d1aa2b4fb08b0034005f9f10ed92314cadfe6d022d", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "a4e69efa2b85b88d02f51ebbaa47a0ea4c211afcbffada4b49354896f5a91a9d", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "2306652b8f72ad772243e3a24ba1164a59e966539082c9eccdac4346cd74ec5d", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "cb19b72d84999e8e0a71f165c94715bb2a62a9254ce064ee0b291d123253e020", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "f978451c3c8e3ef1a62d0c558059426f78a09e4b608b5e1df25d9c6c7b8b2a4f", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "7b52b11b9b2ddb492309607664ca229433dfb28968391634b32cdced1f87b65c", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "d4e44b5430b871ced18c624901e09ea19d0ee67215aeed9f8c1beed25d860bc2", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "ecf047e0c3cec1fc92429c7589801c7a7d5660d9ff65e9884667c29fda2e0281", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "4c0e2ed811f0f254890af50c92c73bd6ea9e224ae77cbca066e285f29ddf92ca", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "0f54bfd186c7577b8cfb69f88ff9f2e8540abd344a09d198c97a7619c76da208", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "ca98843d197f6a4fbaf8e1039beede3db706336ca4610bd20e1f8a8c6feb6c73", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "b48bc5126a2b2a7ba34f94c73ed0b524d52fb1d11ee62bf5a995424028ac3366", ), ); assert.ok( contract.compatibility.fromArtifacts.includes( "362dcc03f0a80013b96a4972f9a7192469738716d92ee90f12041865b1520068", ), ); async function entry(flags, version, fromArtifacts = []) { const deployment = await readJson("dist/release/deployment.json"); deployment.compatibility_flags = flags; const bytes = Object.fromEntries( Object.entries({ "deployment.json": JSON.stringify(deployment), "worker/index.js": "export default {};", "assets/index.html": "<!doctype html><title>Fixture</title>", }).map(([path, text]) => [path, new TextEncoder().encode(text).buffer]), ); const manifest = { schemaVersion: 1, kind: "flarebot-customer-runtime", configurationVersion: 1, release: version, sourceRevision: "a".repeat(40), sourceDirty: false, deployment: "deployment.json", shell: { image: contract.shell.image }, compatibility: { ...contract.compatibility, fromArtifacts }, files: Object.entries(bytes).map(([path, data]) => ({ path, size: data.byteLength, sha256: digest(Buffer.from(data)), mime: path.endsWith(".js") ? "application/javascript" : path.endsWith(".html") ? "text/html" : "application/json", ...(path.startsWith("assets/") ? { assetHash: "b".repeat(32) } : {}), })), }; const manifestBytes = new TextEncoder().encode( JSON.stringify(manifest), ).buffer; return { identity: { version, sourceRevision: manifest.sourceRevision, artifactDigest: digest(Buffer.from(manifestBytes)), }, development: false, files: { ...bytes, "manifest.json": manifestBytes }, }; } const legacyEntry = await entry(["nodejs_compat"], "0.1.0-fixture.legacy"); const legacy = await loadArtifact(legacyEntry, legacyEntry.identity); assert.deepEqual(legacy.deployment.compatibility_flags, ["nodejs_compat"]); const currentEntry = await entry( source.compatibility_flags, "0.1.0-fixture.public", [legacy.identity.artifactDigest], ); const current = await loadArtifact(currentEntry, currentEntry.identity); assert.doesNotThrow(() => verifyUpgradeIdentity(legacy.identity, current)); assert.throws( () => verifyUpgradeIdentity(current.identity, legacy), /artifact_unavailable/, ); assert.throws( () => verifyUpgradeIdentity(legacy.identity, { ...current, compatibility: { ...current.compatibility, fromArtifacts: [] }, }), /artifact_unavailable/, ); for (const flags of [ [], ["global_fetch_strictly_public"], ["nodejs_compat", "global_fetch_private_origin"], ["nodejs_compat", "nodejs_compat"], [...source.compatibility_flags, "unknown_flag"], ]) { await assert.rejects( loadArtifact(await entry(flags, "0.1.0-fixture.invalid")), /artifact_unavailable/, ); }});
test("packaged release is accepted by the production artifact validator", async () => { const manifestBytes = await readFile("dist/release/manifest.json"); const manifest = JSON.parse(manifestBytes); const identity = { version: manifest.release, sourceRevision: manifest.sourceRevision, artifactDigest: digest(manifestBytes), }; const files = Object.fromEntries( await Promise.all( ["manifest.json", ...manifest.files.map((file) => file.path)].map( async (path) => [ path, new Uint8Array(await readFile(`dist/release/${path}`)).buffer, ], ), ), ); const artifact = await loadArtifact( { identity, files, development: manifest.sourceDirty }, identity, manifest.sourceDirty, ); assert.deepEqual(artifact.deployment.ai, { binding: "AI" });});
test( "packaged PersonalAgent instantiates and persists native SDK state", { timeout: 60_000 }, async () => { const worker = await unstable_dev("tests/fixtures/deployment-worker.js", { config: "dist/release/deployment.json", vars: customerBindings, bundle: true, local: true, ip: "127.0.0.1", port: 0, inspectorPort: 0, persist: false, logLevel: "error", experimental: { disableExperimentalWarning: true, watch: false }, }); try { const written = await worker.fetch("/", { method: "POST" }); assert.equal(written.status, 200); const state = await written.json(); assert.equal(state.schemaVersion, 1); assert.ok(Number.isFinite(Date.parse(state.createdAt))); const read = await worker.fetch("/"); assert.equal(read.status, 200); assert.deepEqual(await read.json(), state); const facet = await worker.fetch("/conversation"); assert.equal(facet.status, 200); const created = await facet.json(); assert.match(created.id, /^[0-9a-f-]{36}$/); assert.equal(created.ready, true); } finally { await worker.stop(); } },);