Fix some border cases

packages/safe-chain/src/packagemanager/pip/utils/pipCommands.spec.js

@@ -0,0 +1,88 @@
+import { test } from "node:test";
+import assert from "node:assert";
+import {
+  getPipCommandForArgs,
+  hasDryRunArg,
+  pipInstallCommand,
+  pipDownloadCommand,
+  pipWheelCommand,
+  pipUninstallCommand,
+} from "./pipCommands.js";
+
+test("getPipCommandForArgs", async (t) => {
+  await t.test("should return null for empty args", () => {
+    assert.strictEqual(getPipCommandForArgs([]), null);
+  });
+
+  await t.test("should return null for null args", () => {
+    assert.strictEqual(getPipCommandForArgs(null), null);
+  });
+
+  await t.test("should return the first non-flag argument", () => {
+    assert.strictEqual(getPipCommandForArgs(["install"]), "install");
+  });
+
+  await t.test("should skip flags and return command", () => {
+    assert.strictEqual(
+      getPipCommandForArgs(["-v", "--verbose", "install"]),
+      "install"
+    );
+  });
+
+  await t.test("should return install command", () => {
+    assert.strictEqual(
+      getPipCommandForArgs(["install", "requests"]),
+      "install"
+    );
+  });
+
+  await t.test("should return uninstall command", () => {
+    assert.strictEqual(
+      getPipCommandForArgs(["uninstall", "requests"]),
+      "uninstall"
+    );
+  });
+
+  await t.test("should return null if only flags", () => {
+    assert.strictEqual(getPipCommandForArgs(["--version", "-v"]), null);
+  });
+});
+
+test("hasDryRunArg", async (t) => {
+  await t.test("should return false for empty args", () => {
+    assert.strictEqual(hasDryRunArg([]), false);
+  });
+
+  await t.test("should return true if --dry-run is present", () => {
+    assert.strictEqual(hasDryRunArg(["install", "--dry-run", "requests"]), true);
+  });
+
+  await t.test("should return false if --dry-run is not present", () => {
+    assert.strictEqual(hasDryRunArg(["install", "requests"]), false);
+  });
+
+  await t.test("should return true for --dry-run with other flags", () => {
+    assert.strictEqual(
+      hasDryRunArg(["install", "-v", "--dry-run", "--upgrade", "requests"]),
+      true
+    );
+  });
+});
+
+test("command constants", async (t) => {
+  await t.test("should have correct install command", () => {
+    assert.strictEqual(pipInstallCommand, "install");
+  });
+
+  await t.test("should have correct download command", () => {
+    assert.strictEqual(pipDownloadCommand, "download");
+  });
+
+  await t.test("should have correct wheel command", () => {
+    assert.strictEqual(pipWheelCommand, "wheel");
+  });
+
+  await t.test("should have correct uninstall command", () => {
+    assert.strictEqual(pipUninstallCommand, "uninstall");
+  });
+});