@test-lab-ai/cli 0.2.16 → 0.2.18

package/bin/testlab.mjs

@@ -40,6 +40,9 @@ Usage:
                                             (credentials, labels, fixtures, plans)
   testlab plans list                       List your test plans
   testlab plans create -f plan.json        Create one plan from JSON
+  testlab plans update <id> [-f patch.json] [--prompt P] [--name N]
+                                           Update an existing plan; only the
+                                           fields you pass change
   testlab projects list                    List your projects
   testlab credentials set <key> --value <value>     Set a credential ({{credentials.<key>}})
   testlab credentials list                 List credential keys (values never shown)
@@ -216,6 +219,34 @@ async function cmdPlansCreate(flags) {
   log(`✓ Created plan #${r.json.testPlan.id}: ${r.json.testPlan.name}`)
 }
 
+async function cmdPlansUpdate(flags, args) {
+  const { apiKey, apiUrl } = requireAuth(flags)
+  const planId = parsePlanIdArg(args[2] ?? flags.plan)
+  if (!Number.isInteger(planId)) {
+    errExit("usage: testlab plans update <id> [-f patch.json] [--prompt P] [--name N]  (numeric plan id; see `testlab plans list`)")
+  }
+  // Build a partial patch. A JSON file is the full surface (name, prompt,
+  // testType, agentType, devices, paused, projectId); --prompt / --name are
+  // quick overrides for the two common edits. Only the fields present are sent,
+  // and the server changes only those.
+  let patch = {}
+  if (flags.file) {
+    try {
+      patch = JSON.parse(fs.readFileSync(flags.file, "utf8"))
+    } catch (e) {
+      errExit(`could not read ${flags.file}: ${e.message}`)
+    }
+  }
+  if (flags.prompt !== undefined) patch.prompt = flags.prompt
+  if (flags.name !== undefined) patch.name = flags.name
+  if (Object.keys(patch).length === 0) {
+    errExit("nothing to update: pass -f <patch.json> and/or --prompt / --name. Editable fields: name, prompt, testType, agentType, devices, paused, projectId")
+  }
+  const r = await apiFetch(apiUrl, apiKey, "PUT", `/api/v1/test-plans/${planId}`, patch)
+  if (!r.ok) errExit(`${r.status}: ${r.json?.error || ""}`)
+  log(`✓ Updated plan #${r.json.testPlan.id}: ${r.json.testPlan.name}`)
+}
+
 async function cmdCredentialsSet(flags, args) {
   const { apiKey, apiUrl } = requireAuth(flags)
   const key = args[2]
@@ -639,7 +670,8 @@ async function main() {
     case "plans":
       if (args[1] === "list") return cmdPlansList(flags)
       if (args[1] === "create") return cmdPlansCreate(flags)
-      return errExit("usage: testlab plans <list|create>")
+      if (args[1] === "update") return cmdPlansUpdate(flags, args)
+      return errExit("usage: testlab plans <list|create|update>")
     case "projects":
       if (args[1] === "list") return cmdProjectsList(flags)
       return errExit("usage: testlab projects list")

package/lib/api.mjs

@@ -1,3 +1,7 @@
+import { readFileSync } from "node:fs"
+import { fileURLToPath } from "node:url"
+import { dirname, join } from "node:path"
+
 /**
  * Tiny fetch wrapper for the test-lab public API. Uses global fetch (Node 18+).
  * Returns { ok, status, json } - json is the parsed body (or { raw } on non-JSON).
@@ -10,11 +14,34 @@
  * short-circuits before the app's JSON error handler) from surfacing as a hard
  * "500: upload failed" when a one-second retry would have gone through.
  */
+
+// Client User-Agent: defaults to the CLI's own name+version so the server's
+// audit log can attribute an action to its source surface. The MCP server reuses
+// this same apiFetch, so it overrides the tag at boot via setClientUserAgent
+// ("testlab-mcp/<ver>"); the browser extension hits the API directly and carries
+// its own UA. Kept module-level (not per-call) so every wrapped call is tagged
+// without threading the value through each higher-level lib function.
+function cliVersion() {
+  try {
+    const pkgPath = join(dirname(fileURLToPath(import.meta.url)), "..", "package.json")
+    return JSON.parse(readFileSync(pkgPath, "utf8")).version || "0.0.0"
+  } catch {
+    return "0.0.0"
+  }
+}
+
+let CLIENT_USER_AGENT = `testlab-cli/${cliVersion()}`
+
+/** Override the User-Agent sent on every apiFetch call (MCP calls this at boot). */
+export function setClientUserAgent(ua) {
+  if (ua && typeof ua === "string") CLIENT_USER_AGENT = ua
+}
+
 export async function apiFetch(apiUrl, apiKey, method, pathname, body, opts = {}) {
   const retries = Number.isInteger(opts.retries) ? opts.retries : method === "GET" ? 2 : 0
   const backoffMs = Number.isInteger(opts.backoffMs) ? opts.backoffMs : 800
 
-  const headers = {}
+  const headers = { "User-Agent": CLIENT_USER_AGENT }
   if (apiKey) headers["Authorization"] = `Bearer ${apiKey}`
   if (body !== undefined) headers["Content-Type"] = "application/json"
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@test-lab-ai/cli",
-  "version": "0.2.16",
+  "version": "0.2.18",
   "description": "Import existing test plans into test-lab.ai from the command line (or an AI agent).",
   "type": "module",
   "bin": {
@@ -18,7 +18,7 @@
     "AGENTS.md"
   ],
   "scripts": {
-    "test": "node test/toposort.test.mjs",
+    "test": "node test/toposort.test.mjs && node test/api-useragent.test.mjs",
     "prepublishOnly": "node scripts/bundle-skill.mjs && npm test"
   },
   "keywords": [