e2e-local.ts 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142
  1. import fs from "node:fs/promises"
  2. import net from "node:net"
  3. import os from "node:os"
  4. import path from "node:path"
  5. async function freePort() {
  6. return await new Promise<number>((resolve, reject) => {
  7. const server = net.createServer()
  8. server.once("error", reject)
  9. server.listen(0, () => {
  10. const address = server.address()
  11. if (!address || typeof address === "string") {
  12. server.close(() => reject(new Error("Failed to acquire a free port")))
  13. return
  14. }
  15. server.close((err) => {
  16. if (err) {
  17. reject(err)
  18. return
  19. }
  20. resolve(address.port)
  21. })
  22. })
  23. })
  24. }
  25. async function waitForHealth(url: string) {
  26. const timeout = Date.now() + 120_000
  27. const errors: string[] = []
  28. while (Date.now() < timeout) {
  29. const result = await fetch(url)
  30. .then((r) => ({ ok: r.ok, error: undefined }))
  31. .catch((error) => ({
  32. ok: false,
  33. error: error instanceof Error ? error.message : String(error),
  34. }))
  35. if (result.ok) return
  36. if (result.error) errors.push(result.error)
  37. await new Promise((r) => setTimeout(r, 250))
  38. }
  39. const last = errors.length ? ` (last error: ${errors[errors.length - 1]})` : ""
  40. throw new Error(`Timed out waiting for server health: ${url}${last}`)
  41. }
  42. const appDir = process.cwd()
  43. const repoDir = path.resolve(appDir, "../..")
  44. const opencodeDir = path.join(repoDir, "packages", "opencode")
  45. const modelsJson = path.join(opencodeDir, "test", "tool", "fixtures", "models-api.json")
  46. const extraArgs = (() => {
  47. const args = process.argv.slice(2)
  48. if (args[0] === "--") return args.slice(1)
  49. return args
  50. })()
  51. const [serverPort, webPort] = await Promise.all([freePort(), freePort()])
  52. const sandbox = await fs.mkdtemp(path.join(os.tmpdir(), "opencode-e2e-"))
  53. const serverEnv = {
  54. ...process.env,
  55. MODELS_DEV_API_JSON: modelsJson,
  56. OPENCODE_DISABLE_SHARE: "true",
  57. OPENCODE_DISABLE_LSP_DOWNLOAD: "true",
  58. OPENCODE_DISABLE_DEFAULT_PLUGINS: "true",
  59. OPENCODE_EXPERIMENTAL_DISABLE_FILEWATCHER: "true",
  60. OPENCODE_TEST_HOME: path.join(sandbox, "home"),
  61. XDG_DATA_HOME: path.join(sandbox, "share"),
  62. XDG_CACHE_HOME: path.join(sandbox, "cache"),
  63. XDG_CONFIG_HOME: path.join(sandbox, "config"),
  64. XDG_STATE_HOME: path.join(sandbox, "state"),
  65. OPENCODE_E2E_PROJECT_DIR: repoDir,
  66. OPENCODE_E2E_SESSION_TITLE: "E2E Session",
  67. OPENCODE_E2E_MESSAGE: "Seeded for UI e2e",
  68. OPENCODE_E2E_MODEL: "opencode/gpt-5-nano",
  69. OPENCODE_CLIENT: "app",
  70. } satisfies Record<string, string>
  71. const runnerEnv = {
  72. ...serverEnv,
  73. PLAYWRIGHT_SERVER_HOST: "127.0.0.1",
  74. PLAYWRIGHT_SERVER_PORT: String(serverPort),
  75. VITE_OPENCODE_SERVER_HOST: "127.0.0.1",
  76. VITE_OPENCODE_SERVER_PORT: String(serverPort),
  77. PLAYWRIGHT_PORT: String(webPort),
  78. } satisfies Record<string, string>
  79. const seed = Bun.spawn(["bun", "script/seed-e2e.ts"], {
  80. cwd: opencodeDir,
  81. env: serverEnv,
  82. stdout: "inherit",
  83. stderr: "inherit",
  84. })
  85. const seedExit = await seed.exited
  86. if (seedExit !== 0) {
  87. process.exit(seedExit)
  88. }
  89. Object.assign(process.env, serverEnv)
  90. process.env.AGENT = "1"
  91. process.env.OPENCODE = "1"
  92. const log = await import("../../opencode/src/util/log")
  93. const install = await import("../../opencode/src/installation")
  94. await log.Log.init({
  95. print: true,
  96. dev: install.Installation.isLocal(),
  97. level: "WARN",
  98. })
  99. const servermod = await import("../../opencode/src/server/server")
  100. const inst = await import("../../opencode/src/project/instance")
  101. const server = servermod.Server.listen({ port: serverPort, hostname: "127.0.0.1" })
  102. console.log(`opencode server listening on http://127.0.0.1:${serverPort}`)
  103. const result = await (async () => {
  104. try {
  105. await waitForHealth(`http://127.0.0.1:${serverPort}/global/health`)
  106. const runner = Bun.spawn(["bun", "test:e2e", ...extraArgs], {
  107. cwd: appDir,
  108. env: runnerEnv,
  109. stdout: "inherit",
  110. stderr: "inherit",
  111. })
  112. return { code: await runner.exited }
  113. } catch (error) {
  114. return { error }
  115. } finally {
  116. await inst.Instance.disposeAll()
  117. await server.stop()
  118. }
  119. })()
  120. if ("error" in result) {
  121. console.error(result.error)
  122. process.exit(1)
  123. }
  124. process.exit(result.code)