Stabilize tests

Author: brandonkachen

cli/src/__tests__/test-utils.ts

@@ -69,7 +69,9 @@ const TEST_CLIENT_ENV_DEFAULTS: Record<string, string> = {
 const TEST_SERVER_ENV_DEFAULTS: Record<string, string> = {
   OPEN_ROUTER_API_KEY: 'test',
   OPENAI_API_KEY: 'test',
+  ANTHROPIC_API_KEY: 'test',
   LINKUP_API_KEY: 'test',
+  GRAVITY_API_KEY: 'test',
   PORT: '4242',
   DATABASE_URL: 'postgres://user:pass@localhost:5432/db',
   CODEBUFF_GITHUB_ID: 'test-id',

packages/internal/src/env.ts

@@ -13,7 +13,9 @@ if (isCI) {
 
   ensureEnvDefault('OPEN_ROUTER_API_KEY', 'test')
   ensureEnvDefault('OPENAI_API_KEY', 'test')
+  ensureEnvDefault('ANTHROPIC_API_KEY', 'test')
   ensureEnvDefault('LINKUP_API_KEY', 'test')
+  ensureEnvDefault('GRAVITY_API_KEY', 'test')
   ensureEnvDefault('PORT', '4242')
   ensureEnvDefault('DATABASE_URL', 'postgres://user:pass@localhost:5432/db')
   ensureEnvDefault('CODEBUFF_GITHUB_ID', 'test-id')

web/jest.config.cjs

@@ -13,6 +13,7 @@ const config = {
     '^common/(.*)$': '<rootDir>/../common/src/$1',
     '^@codebuff/internal/env$': '<rootDir>/../packages/internal/src/env.ts',
     '^@codebuff/internal/xml-parser$': '<rootDir>/src/test-stubs/xml-parser.ts',
+    '^bun:test$': '<rootDir>/src/test-stubs/bun-test.ts',
     '^react$': '<rootDir>/node_modules/react',
     '^react-dom$': '<rootDir>/node_modules/react-dom',
   },

web/src/test-stubs/bun-test.ts

@@ -0,0 +1,33 @@
+import {
+  afterEach,
+  beforeEach,
+  describe,
+  expect,
+  it,
+  jest,
+  test,
+} from '@jest/globals'
+
+type MockFactory = <T extends (...args: any[]) => any>(
+  impl?: T,
+) => jest.Mock<ReturnType<T>, Parameters<T>>
+
+const mock = ((impl?: (...args: any[]) => any) => jest.fn(impl)) as MockFactory & {
+  restore: () => void
+  clearAllMocks: () => void
+  module: (moduleName: string, factory: () => unknown) => void
+}
+
+mock.restore = () => {
+  jest.restoreAllMocks()
+}
+
+mock.clearAllMocks = () => {
+  jest.clearAllMocks()
+}
+
+mock.module = (moduleName, factory) => {
+  jest.mock(moduleName, factory)
+}
+
+export { afterEach, beforeEach, describe, expect, it, test, mock }