NetMesh/components/terminalLayer/TerminalLayerWorkspaceSection.test.ts

import assert from "node:assert/strict";
import { readFileSync } from "node:fs";
import test from "node:test";
import React from "react";
import { renderToStaticMarkup } from "react-dom/server";

import { TerminalLayerWorkspaceSection } from "./TerminalLayerWorkspaceSection.tsx";

test("terminal keyboard focus updates the pane selected for magnification", () => {
  const source = readFileSync(new URL("./TerminalLayerSupport.tsx", import.meta.url), "utf8");

  assert.match(source, /onClick=\{handlePaneClick\}/);
  assert.match(source, /onFocusCapture=\{handlePaneClick\}/);
  assert.match(source, /const isCoveredByMagnification = isVisible/);
  assert.match(source, /inert=\{isVisible && !isCoveredByMagnification \? undefined : true\}/);
});

test("closing a magnified terminal clears the overlay selection", () => {
  const source = readFileSync(new URL("../TerminalLayer.tsx", import.meta.url), "utf8");

  assert.match(source, /isPaneMagnificationSelectionValid\(current, terminalPanes, sidePanelPanes\)/);
  assert.match(source, /current\?\.target\.kind === 'terminal' && current\.target\.sessionId === sessionId/);
});

test("terminal panes expose focus mode and temporary magnification as separate actions", () => {
  const source = readFileSync(new URL("TerminalLayerSupport.tsx", import.meta.url), "utf8");

  assert.match(source, /workspaceFocusHandlersRef,\s+workspaceBroadcastHandlersRef,/);
  assert.match(source, /if \(isMagnified\) \{\s+handleTogglePaneMagnification\(\);\s+\}\s+workspaceFocusHandler\?\.\(\);/);
  assert.match(source, /onExpandToFocus=\{inActiveWorkspace && !isFocusMode \? handleExpandToFocus : undefined\}/);
  assert.match(source, /onTogglePaneMagnification=\{inActiveWorkspace && \(!isFocusMode \|\| isMagnified\) \? handleTogglePaneMagnification : undefined\}/);
});

test("focus mode cannot start temporary magnification but can restore stale magnification", () => {
  const source = readFileSync(new URL("../TerminalLayer.tsx", import.meta.url), "utf8");

  assert.match(source, /const hasCurrentMagnification = magnifiedPaneRef\.current\?\.tabId === tabId;/);
  assert.match(source, /if \(workspace\?\.viewMode === 'focus' && !hasCurrentMagnification\) return null;/);
});

test("closing a magnified terminal restores its pane instead of ending the session", () => {
  const terminalSource = readFileSync(new URL("../Terminal.tsx", import.meta.url), "utf8");
  const viewSource = readFileSync(new URL("../terminal/TerminalView.tsx", import.meta.url), "utf8");
  const toolbarSource = readFileSync(new URL("../terminal/TerminalToolbar.tsx", import.meta.url), "utf8");

  assert.match(viewSource, /renderControls\(\{ showClose: inWorkspace, restorePaneLayout: isPaneMagnified \}\)/);
  assert.match(terminalSource, /opts\?: \{ showClose\?: boolean; restorePaneLayout\?: boolean \}/);
  assert.match(terminalSource, /if \(opts\?\.restorePaneLayout\) \{\s+onTogglePaneMagnification\?\.\(\);\s+return;\s+\}/);
  assert.match(terminalSource, /closeLabel=\{t\(opts\?\.restorePaneLayout\s+\? 'terminal\.paneMagnification\.restore'\s+: 'terminal\.toolbar\.closeSession'\)\}/);
  assert.match(toolbarSource, /aria-label=\{closeLabel \?\? t\('terminal\.toolbar\.closeSession'\)\}/);
});

test("workspace section passes resolved session host ids to terminal panes", () => {
  const resolvedSessionHostIds = new Set(["session-1"]);
  let sawResolvedIds = false;

  const TerminalPanesHost = (props: { resolvedSessionHostIds?: Set<string> }) => {
    sawResolvedIds = true;
    assert.equal(props.resolvedSessionHostIds, resolvedSessionHostIds);
    assert.equal(props.resolvedSessionHostIds?.has("session-1"), true);
    return null;
  };

  const ref = { current: null };
  const noop = () => {};
  const ctx = {
    workspaceInnerRef: ref,
    workspaceOverlayRef: ref,
    draggingSessionId: null,
    isFocusMode: false,
    dropHint: null,
    setDropHint: noop,
    computeSplitHint: () => null,
    handleWorkspaceDrop: noop,
    TerminalPanesHost,
    sessions: [],
    sessionHostsMap: new Map(),
    sessionChainHostsMap: new Map(),
    sessionSudoAutofillPasswordsMap: new Map(),
    sessionSudoAutofillCandidatesMap: new Map(),
    resolvedSessionHostIds,
    workspaceById: new Map(),
    workspaceRectsById: new Map(),
    isTerminalLayerVisible: true,
    workspaceFocusHandlersRef: { current: new Map() },
    workspaceBroadcastHandlersRef: { current: new Map() },
    splitHorizontalHandlersRef: { current: new Map() },
    splitVerticalHandlersRef: { current: new Map() },
    themePreview: { targetSessionId: null, targetHostId: null, globalPreview: false, themeId: null },
    keys: [],
    identities: [],
    snippets: [],
    knownHosts: [],
    terminalFontFamilyId: "default",
    fontSize: 14,
    terminalTheme: {},
    followAppTerminalTheme: false,
    accentMode: "theme",
    customAccent: "",
    terminalSettings: {},
    hotkeyScheme: "mac",
    disableTerminalFontZoom: false,
    restoreTerminalCwd: false,
    keyBindings: [],
    resizing: null,
    isComposeBarOpen: false,
    sessionLogConfig: undefined,
    sshDebugLogsEnabled: false,
    onHotkeyAction: noop,
    handleTerminalFontSizeChange: noop,
    handleOpenSftp: noop,
    handleTerminalCwdChange: noop,
    handleTerminalTitleChange: noop,
    handleTerminalBell: noop,
    handleTerminalOutput: noop,
    handleOpenScripts: noop,
    handleOpenHistory: noop,
    handleOpenSystem: noop,
    handleOpenTheme: noop,
    handleCloseSession: noop,
    handleStatusChange: noop,
    handleSessionExit: noop,
    handleTerminalDataCapture: noop,
    handleOsDetected: noop,
    handleUpdateHost: noop,
    handleAddKnownHost: noop,
    handleCommandExecuted: noop,
    handleCommandSubmitted: noop,
    onSetWorkspaceFocusedSession: noop,
    onSplitSession: noop,
    isBroadcastEnabled: () => false,
    handleBroadcastInput: noop,
    handleBroadcastInterruptPriorityChange: noop,
    handleToggleWorkspaceComposeBar: noop,
    handleSnippetExecutorChange: noop,
    handleProgrammaticCommandLogRewriteChange: noop,
    handleAddSelectionToAI: noop,
    activeResizers: [],
    activeWorkspace: null,
    composeBarThemeColors: null,
    findSplitNode: () => null,
    focusedSessionId: null,
    handleComposeSend: noop,
    handleSnippetFromPanel: noop,
    refocusTerminalSession: noop,
    setIsComposeBarOpen: noop,
    setResizing: noop,
    TerminalComposeBar: () => null,
    Array,
    cn: (...values: unknown[]) => values.filter(Boolean).join(" "),
    onStartSessionRename: noop,
    onRemoveSessionFromWorkspace: noop,
    onReorderTabs: noop,
    onStartSessionDrag: noop,
    onEndSessionDrag: noop,
  };

  renderToStaticMarkup(React.createElement(TerminalLayerWorkspaceSection, { ctx }));

  assert.equal(sawResolvedIds, true);
});