postgresai 0.14.0-dev.36 → 0.14.0-dev.38

package/test/init.integration.test.cjs

@@ -157,7 +157,7 @@ async function withTempPostgres(t) {
 async function runCliInit(args, env = {}) {
   const node = process.execPath;
   const cliPath = path.resolve(__dirname, "..", "dist", "bin", "postgres-ai.js");
-  const res = spawnSync(node, [cliPath, "init", ...args], {
+  const res = spawnSync(node, [cliPath, "prepare-db", ...args], {
     encoding: "utf8",
     env: { ...process.env, ...env },
   });
@@ -165,7 +165,7 @@ async function runCliInit(args, env = {}) {
 }
 
 test(
-  "integration: init supports URI / conninfo / psql-like connection styles",
+  "integration: prepare-db supports URI / conninfo / psql-like connection styles",
   { skip: !havePostgresBinaries() },
   async (t) => {
     const pg = await withTempPostgres(t);
@@ -207,7 +207,7 @@ test(
 );
 
 test(
-  "integration: init requires explicit monitoring password in non-interactive mode (unless --print-password)",
+  "integration: prepare-db requires explicit monitoring password in non-interactive mode (unless --print-password)",
   { skip: !havePostgresBinaries() },
   async (t) => {
     const pg = await withTempPostgres(t);
@@ -231,7 +231,7 @@ test(
 );
 
 test(
-  "integration: init fixes slightly-off permissions idempotently",
+  "integration: prepare-db fixes slightly-off permissions idempotently",
   { skip: !havePostgresBinaries() },
   async (t) => {
     const pg = await withTempPostgres(t);
@@ -285,7 +285,7 @@ test(
   }
 );
 
-test("integration: init reports nicely when lacking permissions", { skip: !havePostgresBinaries() }, async (t) => {
+test("integration: prepare-db reports nicely when lacking permissions", { skip: !havePostgresBinaries() }, async (t) => {
   const pg = await withTempPostgres(t);
   const { Client } = require("pg");
 
@@ -310,13 +310,13 @@ test("integration: init reports nicely when lacking permissions", { skip: !haveP
   const limitedUri = `postgresql://limited:${limitedPw}@127.0.0.1:${pg.port}/testdb`;
   const r = await runCliInit([limitedUri, "--password", "monpw", "--skip-optional-permissions"]);
   assert.notEqual(r.status, 0);
-  assert.match(r.stderr, /Error: init:/);
+  assert.match(r.stderr, /Error: prepare-db:/);
   // Should include step context and hint.
   assert.match(r.stderr, /Failed at step "/);
   assert.match(r.stderr, /Fix: connect as a superuser/i);
 });
 
-test("integration: init --verify returns 0 when ok and non-zero when missing", { skip: !havePostgresBinaries() }, async (t) => {
+test("integration: prepare-db --verify returns 0 when ok and non-zero when missing", { skip: !havePostgresBinaries() }, async (t) => {
   const pg = await withTempPostgres(t);
   const { Client } = require("pg");
 
@@ -330,7 +330,7 @@ test("integration: init --verify returns 0 when ok and non-zero when missing", {
   {
     const r = await runCliInit([pg.adminUri, "--verify", "--skip-optional-permissions"]);
     assert.equal(r.status, 0, r.stderr || r.stdout);
-    assert.match(r.stdout, /init verify: OK/i);
+    assert.match(r.stdout, /prepare-db verify: OK/i);
   }
 
   // Break a required privilege and ensure verify fails
@@ -345,12 +345,12 @@ test("integration: init --verify returns 0 when ok and non-zero when missing", {
   {
     const r = await runCliInit([pg.adminUri, "--verify", "--skip-optional-permissions"]);
     assert.notEqual(r.status, 0);
-    assert.match(r.stderr, /init verify failed/i);
+    assert.match(r.stderr, /prepare-db verify failed/i);
     assert.match(r.stderr, /pg_catalog\.pg_index/i);
   }
 });
 
-test("integration: init --reset-password updates the monitoring role login password", { skip: !havePostgresBinaries() }, async (t) => {
+test("integration: prepare-db --reset-password updates the monitoring role login password", { skip: !havePostgresBinaries() }, async (t) => {
   const pg = await withTempPostgres(t);
   const { Client } = require("pg");
 

package/test/init.test.cjs

@@ -177,8 +177,8 @@ test("resolveAdminConnection rejects when connection is missing", () => {
   assert.throws(() => init.resolveAdminConnection({}), /Connection is required/);
 });
 
-test("cli: init with missing connection prints init help/options", () => {
-  const r = runCli(["init"]);
+test("cli: prepare-db with missing connection prints help/options", () => {
+  const r = runCli(["prepare-db"]);
   assert.notEqual(r.status, 0);
   // We should show options, not just the error message.
   assert.match(r.stderr, /--print-sql/);
@@ -307,8 +307,8 @@ test("redactPasswordsInSql redacts password literals with embedded quotes", asyn
   assert.match(redacted, /password '<redacted>'/i);
 });
 
-test("cli: init --print-sql works without connection (offline mode)", () => {
-  const r = runCli(["init", "--print-sql", "-d", "mydb", "--password", "monpw"]);
+test("cli: prepare-db --print-sql works without connection (offline mode)", () => {
+  const r = runCli(["prepare-db", "--print-sql", "-d", "mydb", "--password", "monpw"]);
   assert.equal(r.status, 0, r.stderr || r.stdout);
   assert.match(r.stdout, /SQL plan \(offline; not connected\)/);
   assert.match(r.stdout, new RegExp(`grant connect on database "mydb" to "${DEFAULT_MONITORING_USER}"`, "i"));
@@ -320,4 +320,73 @@ test("pgai wrapper forwards to postgresai CLI", () => {
   assert.match(r.stdout, /postgresai|PostgresAI/i);
 });
 
+test("cli: prepare-db command exists and shows help", () => {
+  const r = runCli(["prepare-db", "--help"]);
+  assert.equal(r.status, 0, r.stderr || r.stdout);
+  assert.match(r.stdout, /monitoring user/i);
+  assert.match(r.stdout, /--print-sql/);
+});
+
+test("cli: prepare-db with missing connection prints help/options", () => {
+  const r = runCli(["prepare-db"]);
+  assert.notEqual(r.status, 0);
+  assert.match(r.stderr, /--print-sql/);
+  assert.match(r.stderr, /--monitoring-user/);
+});
+
+test("cli: prepare-db --print-sql works without connection (offline mode)", () => {
+  const r = runCli(["prepare-db", "--print-sql", "-d", "mydb", "--password", "monpw"]);
+  assert.equal(r.status, 0, r.stderr || r.stdout);
+  assert.match(r.stdout, /SQL plan \(offline; not connected\)/);
+  assert.match(r.stdout, new RegExp(`grant connect on database "mydb" to "${DEFAULT_MONITORING_USER}"`, "i"));
+});
+
+test("cli: mon local-install command exists and shows help", () => {
+  const r = runCli(["mon", "local-install", "--help"]);
+  assert.equal(r.status, 0, r.stderr || r.stdout);
+  assert.match(r.stdout, /--demo/);
+  assert.match(r.stdout, /--api-key/);
+});
+
+// Auth --set-key tests
+test("cli: auth --set-key stores key without OAuth", () => {
+  const fs = require("node:fs");
+  const path = require("node:path");
+  const os = require("node:os");
+  
+  // Use a temp directory for config to avoid modifying user's actual config
+  const tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "pgai-auth-test-"));
+  
+  try {
+    // Create the postgresai subdirectory so we know exactly where config goes
+    const postgresaiDir = path.join(tmpDir, "postgresai");
+    fs.mkdirSync(postgresaiDir, { recursive: true });
+    
+    // Set XDG_CONFIG_HOME to redirect config to temp dir
+    const r = runCli(["auth", "--set-key", "test-api-key-12345"], {
+      XDG_CONFIG_HOME: tmpDir,
+      // Also clear HOME to prevent fallbacks
+      HOME: tmpDir,
+    });
+    
+    assert.equal(r.status, 0, r.stderr || r.stdout);
+    assert.match(r.stdout, /API key saved/i);
+    
+    // Verify the config file was created with the API key
+    const actualConfigPath = path.join(postgresaiDir, "config.json");
+    assert.ok(fs.existsSync(actualConfigPath), "Config file should exist at " + actualConfigPath);
+    
+    const config = JSON.parse(fs.readFileSync(actualConfigPath, "utf8"));
+    assert.equal(config.apiKey, "test-api-key-12345");
+  } finally {
+    // Cleanup
+    fs.rmSync(tmpDir, { recursive: true, force: true });
+  }
+});
+
+test("cli: auth --help shows --set-key option", () => {
+  const r = runCli(["auth", "--help"]);
+  assert.equal(r.status, 0, r.stderr || r.stdout);
+  assert.match(r.stdout, /--set-key/);
+});