elizaos--eliza/packages/scripts/cloud/admin/prune-node-disk.test.ts

// Exercises cloud admin prune node disk.test automation behavior with deterministic script fixtures.
import { describe, expect, it } from "bun:test";
import { parsePruneArgs } from "./prune-node-disk";

const EMPTY: NodeJS.ProcessEnv = {};

describe("parsePruneArgs", () => {
  it("accepts --node-id alone and applies SSH defaults", () => {
    const args = parsePruneArgs(["--node-id", "eliza-core-95ea703e"], EMPTY);
    expect(args).toEqual({
      nodeId: "eliza-core-95ea703e",
      sshPort: 22,
      sshUser: "root",
      dryRun: false,
    });
  });

  it("accepts --host alone with custom ssh user/port", () => {
    const args = parsePruneArgs(
      ["--host", "10.0.0.7", "--ssh-user", "deploy", "--ssh-port", "2222"],
      EMPTY,
    );
    expect(args).toEqual({
      host: "10.0.0.7",
      sshPort: 2222,
      sshUser: "deploy",
      dryRun: false,
    });
  });

  it("requires exactly one of --node-id / --host", () => {
    expect(() => parsePruneArgs([], EMPTY)).toThrow("exactly one of --node-id");
    expect(() =>
      parsePruneArgs(["--node-id", "a", "--host", "b"], EMPTY),
    ).toThrow("only ONE of --node-id or --host");
  });

  it("reads --dry-run", () => {
    const args = parsePruneArgs(["--node-id", "a", "--dry-run"], EMPTY);
    expect(args.dryRun).toBe(true);
  });

  it("falls back to env vars", () => {
    const args = parsePruneArgs([], {
      PRUNE_NODE_HOST: "1.2.3.4",
      PRUNE_NODE_SSH_USER: "ops",
    } as NodeJS.ProcessEnv);
    expect(args.host).toBe("1.2.3.4");
    expect(args.sshUser).toBe("ops");
  });

  it("rejects an invalid ssh-port", () => {
    expect(() =>
      parsePruneArgs(["--host", "h", "--ssh-port", "99999"], EMPTY),
    ).toThrow("Invalid ssh-port");
  });
});