NetMesh/application/state/vaultImportProgress.test.ts

import assert from "node:assert/strict";
import test from "node:test";

import {
  countVaultImportDuplicates,
  ensureVaultImportPersisted,
  mergeVaultImportedGroups,
  rebaseVaultImportedHosts,
  resolveUniqueManagedImportGroupName,
  rollbackVaultImportedHosts,
  waitForVaultImportProgressPaint,
} from "./vaultImportProgress.ts";

test("vault import rollback preserves unrelated concurrent host changes", () => {
  const baseline = [{
    id: "existing",
    label: "Existing",
    hostname: "existing.test",
    username: "root",
    port: 22,
    group: "Original",
    tags: [],
    os: "linux" as const,
  }];
  const applied = [
    { ...baseline[0], group: "Managed", managedSourceId: "source-1" },
    {
      ...baseline[0],
      id: "imported",
      label: "Imported",
      hostname: "imported.test",
    },
    {
      ...baseline[0],
      id: "edited-import",
      label: "Edited import",
      hostname: "edited-import.test",
    },
  ];
  const concurrent = {
    ...baseline[0],
    id: "concurrent",
    label: "Concurrent",
    hostname: "concurrent.test",
  };

  const rolledBack = rollbackVaultImportedHosts({
    baselineHosts: baseline,
    appliedHosts: applied,
    currentHosts: [
      { ...applied[0], notes: "changed while importing" },
      applied[1],
      { ...applied[2], notes: "edited after import started" },
      concurrent,
    ],
  });

  assert.deepEqual(rolledBack, [
    { ...baseline[0], notes: "changed while importing" },
    { ...applied[2], notes: "edited after import started" },
    concurrent,
  ]);

  const rolledBackPerField = rollbackVaultImportedHosts({
    baselineHosts: baseline,
    appliedHosts: [{
      ...baseline[0],
      label: "ExistingWithoutSpaces",
      group: "Managed",
      managedSourceId: "source-1",
    }],
    currentHosts: [{
      ...baseline[0],
      label: "Renamed concurrently",
      group: "Managed",
      managedSourceId: "source-1",
    }],
  });
  assert.deepEqual(rolledBackPerField, [{
    ...baseline[0],
    label: "Renamed concurrently",
  }]);
});

test("vault import rebase preserves concurrent edits and restores missing imported data", () => {
  const baseline = [{
    id: "existing",
    label: "Existing",
    hostname: "existing.test",
    username: "root",
    port: 22,
    group: "Original",
    tags: [],
    os: "linux" as const,
  }];
  const applied = [
    { ...baseline[0], group: "Imported", managedSourceId: "source-1" },
    { ...baseline[0], id: "imported", label: "Imported", hostname: "imported.test" },
  ];
  const concurrent = [
    { ...baseline[0], label: "Renamed concurrently" },
    { ...baseline[0], id: "concurrent", hostname: "concurrent.test" },
  ];

  assert.deepEqual(rebaseVaultImportedHosts({
    currentHosts: concurrent,
    baselineHosts: baseline,
    appliedHosts: applied,
  }), [
    {
      ...baseline[0],
      label: "Renamed concurrently",
      group: "Imported",
      managedSourceId: "source-1",
    },
    { ...baseline[0], id: "concurrent", hostname: "concurrent.test" },
    applied[1],
  ]);
});

test("vault import group merge keeps concurrent changes without restoring deleted groups", () => {
  assert.deepEqual(mergeVaultImportedGroups({
    currentGroups: ["Renamed", "Concurrent"],
    baselineGroups: ["Original", "Kept"],
    appliedGroups: ["Original", "Kept", "Imported"],
  }), ["Renamed", "Concurrent", "Imported"]);
});

test("managed import group naming ignores its own retry and avoids concurrent groups", () => {
  const host = {
    id: "imported",
    label: "Imported",
    hostname: "imported.test",
    username: "root",
    port: 22,
    group: "sessions - Managed",
    managedSourceId: "source-1",
    tags: [],
    os: "linux" as const,
  };
  assert.equal(resolveUniqueManagedImportGroupName({
    baseName: "sessions",
    customGroups: ["sessions - Managed"],
    hosts: [host],
    managedSources: [],
    ownerSourceId: "source-1",
  }), "sessions - Managed (1)");
  assert.equal(resolveUniqueManagedImportGroupName({
    baseName: "sessions",
    customGroups: [],
    hosts: [host],
    managedSources: [],
    ownerSourceId: "source-1",
  }), "sessions - Managed");
});

test("vault import duplicate count includes hosts that already exist", () => {
  assert.equal(countVaultImportDuplicates({
    importedHostCount: 8000,
    newHostCount: 0,
    fileDuplicateCount: 3,
    managed: false,
  }), 8003);

  assert.equal(countVaultImportDuplicates({
    importedHostCount: 8000,
    newHostCount: 0,
    fileDuplicateCount: 3,
    managed: true,
  }), 3);
});

test("vault import treats an explicit persistence failure as an import failure", async () => {
  let committed = 0;
  let rolledBack = 0;
  await assert.doesNotReject(() => ensureVaultImportPersisted(
    undefined,
    "not saved",
    () => { committed++; },
  ));
  await assert.rejects(
    ensureVaultImportPersisted(
      false,
      "not saved",
      () => { committed++; },
      () => { rolledBack++; },
    ),
    /not saved/,
  );
  assert.equal(committed, 1);
  assert.equal(rolledBack, 1);

  await assert.rejects(
    ensureVaultImportPersisted(
      false,
      "not saved",
      undefined,
      async () => { throw new Error("rollback also failed"); },
    ),
    /not saved/,
  );
});

test("vault import keeps moving when animation frames are paused in a background window", async () => {
  let timeoutCallback: (() => void) | undefined;
  let resolved = false;

  const promise = waitForVaultImportProgressPaint({
    requestFrame: () => 1,
    setTimer: (callback) => {
      timeoutCallback = callback;
      return 1;
    },
    clearTimer: () => {},
  }).then(() => {
    resolved = true;
  });

  await Promise.resolve();
  assert.equal(resolved, false);

  timeoutCallback?.();
  await promise;
  assert.equal(resolved, true);
});