Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

apps/desktop/electron/window-connection-route.test.ts

@@ -0,0 +1,122 @@
+import assert from 'node:assert/strict'
+
+import { test } from 'vitest'
+
+import {
+  normalizeWindowConnectionRoute,
+  registrySshScopeForWindowRoute,
+  WindowConnectionRouteRegistry
+} from './window-connection-route'
+
+test('normalizes an exact registry-scoped connection and profile', () => {
+  assert.deepEqual(
+    normalizeWindowConnectionRoute({
+      connectionId: 'source-b',
+      profile: 'research',
+      registryScoped: true
+    }),
+    {
+      connectionId: 'source-b',
+      profile: 'research',
+      registryScoped: true
+    }
+  )
+})
+
+test('keeps legacy/profile-only routes distinct from registry identities', () => {
+  assert.deepEqual(normalizeWindowConnectionRoute({ profile: 'work' }), {
+    connectionId: null,
+    profile: 'work',
+    registryScoped: false
+  })
+})
+
+test('preserves a registry connection when no profile is selected', () => {
+  assert.deepEqual(
+    normalizeWindowConnectionRoute({
+      connectionId: 'source-b',
+      registryScoped: true
+    }),
+    {
+      connectionId: 'source-b',
+      profile: undefined,
+      registryScoped: true
+    }
+  )
+})
+
+test('isolates active routes by webContents id', () => {
+  const routes = new WindowConnectionRouteRegistry()
+
+  routes.set(11, { connectionId: 'source-a', profile: 'default', registryScoped: true })
+  routes.set(22, { connectionId: 'source-b', profile: 'worker', registryScoped: true })
+
+  assert.equal(routes.get(11)?.connectionId, 'source-a')
+  assert.equal(routes.get(22)?.connectionId, 'source-b')
+
+  routes.delete(11)
+
+  assert.equal(routes.get(11), null)
+  assert.equal(routes.get(22)?.connectionId, 'source-b')
+})
+
+test('invalid publications clear only the sender route', () => {
+  const routes = new WindowConnectionRouteRegistry()
+
+  routes.set(11, { connectionId: 'source-a', profile: 'default', registryScoped: true })
+  routes.set(22, { connectionId: 'source-b', profile: 'worker', registryScoped: true })
+
+  routes.set(11, null)
+
+  assert.equal(routes.get(11), null)
+  assert.equal(routes.get(22)?.connectionId, 'source-b')
+})
+
+test('routes a non-primary SSH connection independently from another window', () => {
+  const registry = {
+    primary: 'source-a',
+    connections: [
+      { id: 'source-a', kind: 'ssh' },
+      { id: 'source-b', kind: 'ssh' },
+      { id: 'source-c', kind: 'remote' }
+    ]
+  } as never
+
+  const routes = new WindowConnectionRouteRegistry()
+
+  routes.set(11, {
+    connectionId: 'source-b',
+    profile: 'worker',
+    registryScoped: true
+  })
+  routes.set(22, {
+    connectionId: 'source-c',
+    profile: 'default',
+    registryScoped: true
+  })
+
+  assert.equal(registrySshScopeForWindowRoute(routes.get(11), registry), 'conn:source-b::worker')
+  assert.equal(registrySshScopeForWindowRoute(routes.get(22), registry), null)
+})
+
+test('uses the canonical default profile scope when a registry SSH route has no profile', () => {
+  const registry = {
+    primary: 'source-a',
+    connections: [
+      { id: 'source-a', kind: 'ssh' },
+      { id: 'source-b', kind: 'ssh' }
+    ]
+  } as never
+
+  assert.equal(
+    registrySshScopeForWindowRoute(
+      {
+        connectionId: 'source-b',
+        profile: undefined,
+        registryScoped: true
+      },
+      registry
+    ),
+    'conn:source-b::default'
+  )
+})