loader-composition.spec.ts 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  1. import { createHmac } from 'node:crypto'
  2. import { mkdtemp, rm, writeFile } from 'node:fs/promises'
  3. import { tmpdir } from 'node:os'
  4. import { join } from 'node:path'
  5. import { pathToFileURL } from 'node:url'
  6. import { Context } from '@deepseek-ai/cordis'
  7. import Include from '@deepseek-ai/cordis-plugin-include'
  8. import Loader from '@deepseek-ai/cordis-plugin-loader'
  9. import WebServer from '@deepseek-ai/dsh-host-webserver'
  10. import { afterEach, describe, expect, it, vi } from 'vitest'
  11. import * as GitHubAdapter from '../src/index.ts'
  12. let root: string | undefined
  13. let context: Context | undefined
  14. afterEach(async () => {
  15. await context?.fiber.dispose()
  16. context = undefined
  17. if (root !== undefined) await rm(root, { recursive: true, force: true })
  18. root = undefined
  19. })
  20. describe('real Loader composition', () => {
  21. it('registers on a real WebServer and dispatches a signed request', { timeout: 60_000 }, async () => {
  22. root = await mkdtemp(join(tmpdir(), 'dsh-webhook-github-loader-'))
  23. const configPath = join(root, 'cordis.yml')
  24. await writeFile(configPath, [
  25. '- name: fixture-dependencies',
  26. "- name: '@deepseek-ai/dsh-host-webserver'",
  27. ' config:',
  28. " host: '127.0.0.1'",
  29. ' port: 0',
  30. "- name: '@deepseek-ai/dsh-webhook-github'",
  31. ' config:',
  32. ' source: loader',
  33. ' path: /github',
  34. ' secretEnv: DSH_GITHUB_WEBHOOK_SECRET',
  35. ' maxBodyBytes: 1024',
  36. '',
  37. ].join('\n'))
  38. const dispatch = vi.fn()
  39. const dependencies = {
  40. name: 'fixture-dependencies',
  41. apply(ctx: Context) {
  42. ctx.provide('webhookRuntime', { dispatch } as never)
  43. ctx.provide('credentials', {
  44. resolve: async () => ({ value: 'loader-secret', source: 'environment' }),
  45. } as never)
  46. },
  47. }
  48. context = new Context()
  49. context.baseUrl = pathToFileURL(root).href + '/'
  50. await context.plugin(Loader)
  51. context.loader.builtins.include = Include
  52. const modules = new Map<string, unknown>([
  53. ['fixture-dependencies', dependencies],
  54. ['@deepseek-ai/dsh-host-webserver', WebServer],
  55. ['@deepseek-ai/dsh-webhook-github', GitHubAdapter],
  56. ])
  57. context.loader.internal = {
  58. version: 'v2',
  59. async import(specifier: string) {
  60. if (!modules.has(specifier)) throw new Error(`unexpected Loader import: ${specifier}`)
  61. return modules.get(specifier)
  62. },
  63. } as unknown as NonNullable<typeof context.loader.internal>
  64. await context.loader.create({
  65. name: 'cordis:include',
  66. config: { path: pathToFileURL(configPath).href },
  67. })
  68. await context.loader.await()
  69. expect([...context.loader.entries()].filter(entry => entry.fiber === undefined && !entry.disabled)).toEqual([])
  70. const body = JSON.stringify({ action: 'ready_for_review' })
  71. const signature = `sha256=${createHmac('sha256', 'loader-secret').update(body).digest('hex')}`
  72. const response = await fetch(`http://127.0.0.1:${String(context.webServer.port)}/github`, {
  73. method: 'POST',
  74. headers: {
  75. 'content-type': 'application/json',
  76. 'x-hub-signature-256': signature,
  77. 'x-github-event': 'pull_request',
  78. 'x-github-delivery': 'loader-delivery',
  79. },
  80. body,
  81. })
  82. expect(response.status).toBe(202)
  83. expect(dispatch).toHaveBeenCalledOnce()
  84. })
  85. })