mirror of
https://github.com/deepseek-ai/deepseek-harness.git
synced 2026-08-29 04:26:38 +00:00
106 lines
3.9 KiB
TypeScript
106 lines
3.9 KiB
TypeScript
import { Context } from '@deepseek-ai/cordis'
|
|
import { HostConnectionService } from '@deepseek-ai/dsh-client-connection'
|
|
import type { BrowserAuth } from '@deepseek-ai/dsh-client-connection/src/browser-auth.ts'
|
|
import type { SessionHeader, SessionId } from '@deepseek-ai/dsh-session'
|
|
import type { SessionRawArtifact } from '@deepseek-ai/dsh-session-persistence'
|
|
import { strFromU8, unzipSync } from 'fflate'
|
|
import { describe, expect, it } from 'vitest'
|
|
import {
|
|
Config,
|
|
SESSION_LOG_EXPORT_PATH,
|
|
apply,
|
|
inject,
|
|
} from '../src/index.ts'
|
|
|
|
const sid = (value: string): SessionId => value as SessionId
|
|
|
|
function artifact(id: string): SessionRawArtifact {
|
|
const header: SessionHeader = {
|
|
version: 0,
|
|
id: sid(id),
|
|
createdAt: 1,
|
|
cwd: '/workspace',
|
|
delegationDepth: 0,
|
|
}
|
|
return {
|
|
meta: header,
|
|
filename: 'session.jsonl',
|
|
content: `${JSON.stringify({ type: 'session', ...header })}\n`,
|
|
}
|
|
}
|
|
|
|
async function mounted(withServices: boolean): Promise<{
|
|
readonly connection: HostConnectionService
|
|
readonly dispose: () => Promise<void>
|
|
}> {
|
|
const ctx = new Context()
|
|
ctx.provide('commands', { register: () => () => {} } as never)
|
|
if (withServices) {
|
|
ctx.provide('sessionQuery', {
|
|
traceSession: async () => ({ descendants: [] }),
|
|
} as never)
|
|
ctx.provide('sessionPersistence', {
|
|
supportsRawArtifacts: true,
|
|
readRaw: async (id: SessionId) => artifact(String(id)),
|
|
} as never)
|
|
ctx.provide('attachments', {
|
|
readImage: async () => { throw new Error('fixture has no images') },
|
|
} as never)
|
|
}
|
|
const connection = new HostConnectionService(ctx, [], {} as BrowserAuth)
|
|
const fiber = ctx.plugin({ inject: [...inject], apply })
|
|
await fiber
|
|
return { connection, dispose: () => fiber.dispose() }
|
|
}
|
|
|
|
describe('Session log export Fetch route', () => {
|
|
it('registers one GET/HEAD route and removes it with the plugin fiber', async () => {
|
|
const { connection, dispose } = await mounted(true)
|
|
const fallback = { fetch: async () => new Response('fallback', { status: 418 }) }
|
|
const shared = connection.createSharedFetchHandler('/api', fallback)
|
|
|
|
const response = await shared.fetch(new Request(
|
|
`http://host${SESSION_LOG_EXPORT_PATH}?sessionId=session-1`,
|
|
))
|
|
expect(response.status).toBe(200)
|
|
expect(response.headers.get('content-type')).toBe('application/zip')
|
|
const files = unzipSync(new Uint8Array(await response.arrayBuffer()))
|
|
expect(strFromU8(files['session.jsonl'] as Uint8Array)).toContain('"id":"session-1"')
|
|
|
|
const head = await shared.fetch(new Request(
|
|
`http://host${SESSION_LOG_EXPORT_PATH}?sessionId=session-1`, { method: 'HEAD' },
|
|
))
|
|
expect(head.status).toBe(200)
|
|
expect(head.body).toBeNull()
|
|
|
|
await dispose()
|
|
expect((await shared.fetch(new Request(
|
|
`http://host${SESSION_LOG_EXPORT_PATH}?sessionId=session-1`,
|
|
))).status).toBe(418)
|
|
})
|
|
|
|
it('validates the query before reporting missing export services', async () => {
|
|
const { connection, dispose } = await mounted(false)
|
|
const shared = connection.createSharedFetchHandler('/api', {
|
|
fetch: async () => new Response('fallback', { status: 418 }),
|
|
})
|
|
expect((await shared.fetch(new Request(`http://host${SESSION_LOG_EXPORT_PATH}`))).status).toBe(400)
|
|
expect((await shared.fetch(new Request(
|
|
`http://host${SESSION_LOG_EXPORT_PATH}?sessionId=session-1&includeDescendants=1`,
|
|
))).status).toBe(400)
|
|
expect((await shared.fetch(new Request(
|
|
`http://host${SESSION_LOG_EXPORT_PATH}?sessionId=session-1`,
|
|
))).status).toBe(500)
|
|
await dispose()
|
|
})
|
|
|
|
it('validates the compression level', () => {
|
|
expect(Config({})).toEqual({ compressionLevel: 6 })
|
|
expect(Config({ compressionLevel: 0 })).toEqual({ compressionLevel: 0 })
|
|
expect(Config({ compressionLevel: 9 })).toEqual({ compressionLevel: 9 })
|
|
for (const compressionLevel of [-1, 10, 1.5]) {
|
|
expect(() => Config({ compressionLevel } as never)).toThrow()
|
|
}
|
|
})
|
|
})
|