preload.ts 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. // IMPORTANT: Set env vars BEFORE any imports from src/ directory
  2. // xdg-basedir reads env vars at import time, so we must set these first
  3. import os from "os"
  4. import path from "path"
  5. import fs from "fs/promises"
  6. import fsSync from "fs"
  7. import { afterAll } from "bun:test"
  8. const dir = path.join(os.tmpdir(), "opencode-test-data-" + process.pid)
  9. await fs.mkdir(dir, { recursive: true })
  10. afterAll(() => {
  11. fsSync.rmSync(dir, { recursive: true, force: true })
  12. })
  13. process.env["XDG_DATA_HOME"] = path.join(dir, "share")
  14. process.env["XDG_CACHE_HOME"] = path.join(dir, "cache")
  15. process.env["XDG_CONFIG_HOME"] = path.join(dir, "config")
  16. process.env["XDG_STATE_HOME"] = path.join(dir, "state")
  17. // Pre-fetch models.json so tests don't need the macro fallback
  18. // Also write the cache version file to prevent global/index.ts from clearing the cache
  19. const cacheDir = path.join(dir, "cache", "opencode")
  20. await fs.mkdir(cacheDir, { recursive: true })
  21. await fs.writeFile(path.join(cacheDir, "version"), "14")
  22. const response = await fetch("https://models.dev/api.json")
  23. if (response.ok) {
  24. await fs.writeFile(path.join(cacheDir, "models.json"), await response.text())
  25. }
  26. // Disable models.dev refresh to avoid race conditions during tests
  27. process.env["OPENCODE_DISABLE_MODELS_FETCH"] = "true"
  28. // Clear provider env vars to ensure clean test state
  29. delete process.env["ANTHROPIC_API_KEY"]
  30. delete process.env["OPENAI_API_KEY"]
  31. delete process.env["GOOGLE_API_KEY"]
  32. delete process.env["GOOGLE_GENERATIVE_AI_API_KEY"]
  33. delete process.env["AZURE_OPENAI_API_KEY"]
  34. delete process.env["AWS_ACCESS_KEY_ID"]
  35. delete process.env["AWS_PROFILE"]
  36. delete process.env["OPENROUTER_API_KEY"]
  37. delete process.env["GROQ_API_KEY"]
  38. delete process.env["MISTRAL_API_KEY"]
  39. delete process.env["PERPLEXITY_API_KEY"]
  40. delete process.env["TOGETHER_API_KEY"]
  41. delete process.env["XAI_API_KEY"]
  42. delete process.env["DEEPSEEK_API_KEY"]
  43. delete process.env["FIREWORKS_API_KEY"]
  44. delete process.env["CEREBRAS_API_KEY"]
  45. delete process.env["SAMBANOVA_API_KEY"]
  46. // Now safe to import from src/
  47. const { Log } = await import("../src/util/log")
  48. Log.init({
  49. print: false,
  50. dev: true,
  51. level: "DEBUG",
  52. })