vercel-labs--agent-browser/packages/@agent-browser/sandbox/test/shared.test.mjs

import assert from "node:assert/strict";
import { readFileSync } from "node:fs";
import test from "node:test";

import {
  AGENT_BROWSER_SANDBOX_VERSION,
  DEFAULT_AGENT_BROWSER_INSTALL_SPEC,
  buildAgentBrowserArgv,
  buildShellCommand,
  defaultSessionName,
  quoteShellArg,
  resolveAgentBrowserInstallSpec,
} from "../dist/index.js";

test("defaults install spec to the package version", () => {
  const packageJson = JSON.parse(
    readFileSync(new URL("../package.json", import.meta.url), "utf-8"),
  );
  const expectedInstallSpec = `agent-browser@${packageJson.version}`;

  assert.equal(AGENT_BROWSER_SANDBOX_VERSION, packageJson.version);
  assert.equal(DEFAULT_AGENT_BROWSER_INSTALL_SPEC, expectedInstallSpec);
  assert.equal(resolveAgentBrowserInstallSpec(), expectedInstallSpec);
  assert.equal(resolveAgentBrowserInstallSpec({ installSpec: "latest" }), "latest");
});

test("builds argv with session and json by default", () => {
  assert.deepEqual(buildAgentBrowserArgv(["open", "https://example.com"], { session: "s1" }), [
    "--session",
    "s1",
    "open",
    "https://example.com",
    "--json",
  ]);

  assert.deepEqual(buildAgentBrowserArgv(["snapshot", "--json"], { session: "s1" }), [
    "--session",
    "s1",
    "snapshot",
    "--json",
  ]);
});

test("quotes shell args and builds commands", () => {
  assert.equal(quoteShellArg("simple"), "simple");
  assert.equal(quoteShellArg("hello world"), "'hello world'");
  assert.equal(quoteShellArg("can't"), "'can'\\''t'");

  assert.equal(
    buildShellCommand(["open", "https://example.com/a b"], {
      env: { AGENT_BROWSER_HOME: "/tmp/agent browser" },
      session: "s1",
    }),
    "AGENT_BROWSER_HOME='/tmp/agent browser' agent-browser --session s1 open 'https://example.com/a b' --json",
  );
});

test("sanitizes default session names", () => {
  assert.equal(defaultSessionName("eve", "sandbox/id 1"), "eve-sandbox-id-1");
});

test("keeps generated default session names short", () => {
  const session = defaultSessionName(
    "eve",
    "eve-sbx-ses-vercel-1d940340bdba4563-wrun_01KVKDK1Z3GC3XEC86DGWRWRMH-__root__",
  );

  assert.equal(session.length <= 48, true);
  assert.match(session, /^eve-eve-sbx-ses-vercel-.+-[a-f0-9]{8}$/);
});