42 lines
1.6 KiB
TypeScript
42 lines
1.6 KiB
TypeScript
import { strict as assert } from "node:assert";
|
|
import test from "node:test";
|
|
import {
|
|
CONNECTION_ATTEMPT_TIMEOUT_BUFFER_MS,
|
|
MONGO_LEGACY_FALLBACK_TIMEOUT_BUFFER_MS,
|
|
connectionAttemptTimeoutMessage,
|
|
connectionAttemptTimeoutMs,
|
|
} from "../../apps/desktop/src/lib/connectionAttemptTimeout.ts";
|
|
|
|
test("uses connection timeout with a small UI buffer", () => {
|
|
assert.equal(connectionAttemptTimeoutMs({ connect_timeout_secs: 8 }), 8_000 + CONNECTION_ATTEMPT_TIMEOUT_BUFFER_MS);
|
|
});
|
|
|
|
test("falls back to the default connection timeout", () => {
|
|
assert.equal(connectionAttemptTimeoutMs({}), 5_000 + CONNECTION_ATTEMPT_TIMEOUT_BUFFER_MS);
|
|
assert.equal(connectionAttemptTimeoutMs({ connect_timeout_secs: 0 }), 5_000 + CONNECTION_ATTEMPT_TIMEOUT_BUFFER_MS);
|
|
});
|
|
|
|
test("honors slower SSH tunnel connection timeouts", () => {
|
|
assert.equal(
|
|
connectionAttemptTimeoutMs({
|
|
connect_timeout_secs: 5,
|
|
ssh_enabled: true,
|
|
ssh_connect_timeout_secs: 12,
|
|
ssh_tunnels: [{ id: "hop-1", host: "bastion", port: 22, user: "dbx", connect_timeout_secs: 20 }],
|
|
}),
|
|
20_000 + CONNECTION_ATTEMPT_TIMEOUT_BUFFER_MS,
|
|
);
|
|
});
|
|
|
|
test("allows MongoDB legacy agent fallback after native driver timeout", () => {
|
|
assert.equal(
|
|
connectionAttemptTimeoutMs({ db_type: "mongodb", connect_timeout_secs: 5 }),
|
|
5_000 + CONNECTION_ATTEMPT_TIMEOUT_BUFFER_MS + MONGO_LEGACY_FALLBACK_TIMEOUT_BUFFER_MS,
|
|
);
|
|
});
|
|
|
|
test("formats connection attempt timeout messages", () => {
|
|
assert.match(connectionAttemptTimeoutMessage(7_001), /timed out after 8s/);
|
|
assert.match(connectionAttemptTimeoutMessage(7_001), /VPN/);
|
|
});
|