deepseek-harness/packages/client/connection/tests/node-half.spec.ts

/** Node half: registers the /api prefix route bridging to the api gateway. */
import { Context } from 'cordis'
import { describe, expect, it } from 'vitest'
import type { IncomingMessage, ServerResponse } from 'node:http'
import type { ApiProxy } from '@deepseek-ai/dsh-host-apiproxy/api'
import type { HttpServerService, WebRoute } from '@deepseek-ai/dsh-host-webserver'
import type { AttachmentStore } from '@deepseek-ai/dsh-attachment'
import { API_PATH, apply, inject } from '../src/index.ts'

describe('connection node half', () => {
  it('registers the /api prefix route and removes it with the fiber', async () => {
    const ctx = new Context()
    const routes: WebRoute[] = []
    // Structural fake: the plugin only touches register(); the service class
    // carries private state a literal cannot (and need not) reproduce.
    const httpServer: Pick<HttpServerService, 'register' | 'tapIndex' | 'port'> = {
      register(route) {
        routes.push(route)
        return () => { routes.splice(routes.indexOf(route), 1) }
      },
      tapIndex: () => () => {},
      port: 0,
    }
    ctx.provide('httpServer', httpServer as HttpServerService)
    ctx.provide('apiProxy', {} as unknown as ApiProxy)
    ctx.provide('attachments', {
      imageLimits: { maxMessageImageBytes: 20 * 1024 * 1024 },
    } as AttachmentStore)

    const fiber = ctx.plugin({ inject: [...inject], apply })
    await fiber.await()
    expect(routes).toHaveLength(1)
    expect(routes[0]).toMatchObject({ kind: 'prefix', path: API_PATH })

    for (const url of ['/api/host.pickDirectory', '/api/host.openPath']) {
      let status: number | undefined
      let body: unknown
      const deniedRequest = {
        url,
        headers: {
          host: 'harness.example', origin: 'http://harness.example', 'sec-fetch-site': 'same-origin',
        },
        socket: { remoteAddress: '192.168.1.8' },
      } as unknown as IncomingMessage
      const deniedResponse = {
        writeHead(value: number) { status = value; return this },
        end(value?: unknown) { body = value; return this },
      } as unknown as ServerResponse
      await routes[0]!.handler(deniedRequest, deniedResponse)
      expect(status).toBe(403)
      expect(body).toBe('forbidden')
    }

    await fiber.dispose()
    expect(routes).toHaveLength(0)
  })
})