[Init] Initial commit - NetMesh terminal manager

components/terminal/connectionTimeouts.test.ts

@@ -0,0 +1,99 @@
+import test from "node:test";
+import assert from "node:assert/strict";
+
+import {
+  SSH_AUTH_READY_TIMEOUT_MS,
+  SSH_TCP_CONNECT_TIMEOUT_MS,
+  getConnectionTimeoutMs,
+  hasConnectionPassedTcpDial,
+  resolveActiveConnectionTimeoutHost,
+  shouldRunConnectionTimeout,
+} from "./connectionTimeouts";
+
+test("SSH connection timeout constants separate TCP dial from auth wait", () => {
+  assert.equal(SSH_TCP_CONNECT_TIMEOUT_MS, 20_000);
+  assert.equal(SSH_AUTH_READY_TIMEOUT_MS, 120_000);
+});
+
+const baseTimeoutState = {
+  status: "connecting",
+  needsAuth: false,
+  isLocalConnection: false,
+  isSerialConnection: false,
+  hasSshTcpConnectProgress: true,
+  needsHostKeyVerification: false,
+  isConnectionAwaitingUserInput: false,
+  isConnectionPastTcpDial: false,
+};
+
+test("connection timeout runs for an ordinary remote connect attempt", () => {
+  assert.equal(shouldRunConnectionTimeout(baseTimeoutState), true);
+});
+
+test("connection timeout pauses while SSH waits for user confirmation", () => {
+  assert.equal(shouldRunConnectionTimeout({
+    ...baseTimeoutState,
+    needsHostKeyVerification: true,
+  }), false);
+  assert.equal(shouldRunConnectionTimeout({
+    ...baseTimeoutState,
+    isConnectionAwaitingUserInput: true,
+  }), false);
+});
+
+test("connection timeout switches from TCP dial to auth readiness after TCP connects", () => {
+  assert.equal(getConnectionTimeoutMs(baseTimeoutState), SSH_TCP_CONNECT_TIMEOUT_MS);
+  assert.equal(getConnectionTimeoutMs({
+    ...baseTimeoutState,
+    isConnectionPastTcpDial: true,
+  }), SSH_AUTH_READY_TIMEOUT_MS);
+});
+
+test("connection timeout uses configured timeout values", () => {
+  assert.equal(getConnectionTimeoutMs(baseTimeoutState, {
+    tcpConnectTimeoutMs: 45_000,
+    authReadyTimeoutMs: 300_000,
+  }), 45_000);
+  assert.equal(getConnectionTimeoutMs({
+    ...baseTimeoutState,
+    isConnectionPastTcpDial: true,
+  }, {
+    tcpConnectTimeoutMs: 45_000,
+    authReadyTimeoutMs: 300_000,
+  }), 300_000);
+});
+
+test("connection timeout follows the current jump host before the target", () => {
+  const target = { sshTcpConnectTimeoutSeconds: 20, sshAuthReadyTimeoutSeconds: 120 };
+  const jumps = [
+    { sshTcpConnectTimeoutSeconds: 75, sshAuthReadyTimeoutSeconds: 360 },
+    { sshTcpConnectTimeoutSeconds: 90, sshAuthReadyTimeoutSeconds: 420 },
+  ];
+
+  assert.equal(resolveActiveConnectionTimeoutHost(target, jumps, 1), jumps[0]);
+  assert.equal(resolveActiveConnectionTimeoutHost(target, jumps, 2), jumps[1]);
+  assert.equal(resolveActiveConnectionTimeoutHost(target, jumps, 3), target);
+  assert.equal(resolveActiveConnectionTimeoutHost(target, jumps), target);
+  assert.equal(resolveActiveConnectionTimeoutHost(target, jumps, 1, "forwarding"), jumps[1]);
+  assert.equal(resolveActiveConnectionTimeoutHost(target, jumps, 2, "forwarding"), target);
+});
+
+test("connection timeout keeps the auth-ready window for protocols without SSH TCP progress", () => {
+  assert.equal(getConnectionTimeoutMs({
+    ...baseTimeoutState,
+    hasSshTcpConnectProgress: false,
+  }, {
+    tcpConnectTimeoutMs: 5_000,
+    authReadyTimeoutMs: 5_000,
+  }), SSH_AUTH_READY_TIMEOUT_MS);
+});
+
+test("TCP dial is only considered passed after an actual transport connection", () => {
+  assert.equal(hasConnectionPassedTcpDial("connecting"), false);
+  assert.equal(hasConnectionPassedTcpDial("auth-attempt"), false);
+  assert.equal(hasConnectionPassedTcpDial("error"), false);
+  assert.equal(hasConnectionPassedTcpDial("forwarding"), false);
+  assert.equal(hasConnectionPassedTcpDial("tcp-connected"), true);
+  assert.equal(hasConnectionPassedTcpDial("authenticating"), true);
+  assert.equal(hasConnectionPassedTcpDial("connected"), true);
+});