fix tests

Author: icecrasher321

apps/sim/app/api/knowledge/[id]/documents/route.test.ts

@@ -43,6 +43,10 @@ vi.mock('@/lib/knowledge/documents/service', () => ({
   KnowledgeBaseFileOwnershipError: class KnowledgeBaseFileOwnershipError extends Error {},
 }))
 
+vi.mock('@/lib/billing/calculations/usage-monitor', () => ({
+  checkActorUsageLimits: vi.fn().mockResolvedValue({ isExceeded: false }),
+}))
+
 vi.mock('@sim/audit', () => auditMock)
 
 import {

apps/sim/app/api/knowledge/[id]/documents/upsert/route.test.ts

@@ -27,6 +27,10 @@ vi.mock('@/lib/auth/hybrid', () => hybridAuthMock)
 vi.mock('@/app/api/knowledge/utils', () => knowledgeApiUtilsMock)
 vi.mock('@sim/audit', () => auditMock)
 
+vi.mock('@/lib/billing/calculations/usage-monitor', () => ({
+  checkActorUsageLimits: vi.fn().mockResolvedValue({ isExceeded: false }),
+}))
+
 vi.mock('@/lib/knowledge/documents/service', () => ({
   createDocumentRecords: vi.fn(),
   deleteDocument: vi.fn(),

apps/sim/app/api/v1/knowledge/search/route.test.ts

@@ -42,6 +42,10 @@ vi.mock('@/app/api/knowledge/search/utils', () => ({
 
 vi.mock('@/app/api/knowledge/utils', () => knowledgeApiUtilsMock)
 
+vi.mock('@/lib/billing/calculations/usage-monitor', () => ({
+  checkActorUsageLimits: vi.fn().mockResolvedValue({ isExceeded: false }),
+}))
+
 vi.mock('@/app/api/v1/middleware', () => ({
   authenticateRequest: mockAuthenticateRequest,
   validateWorkspaceAccess: mockValidateWorkspaceAccess,

apps/sim/lib/copilot/request/lifecycle/headless.test.ts

@@ -6,7 +6,6 @@ import { propagation, trace } from '@opentelemetry/api'
 import { W3CTraceContextPropagator } from '@opentelemetry/core'
 import { BasicTracerProvider } from '@opentelemetry/sdk-trace-base'
 import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
-import { RequestTraceV1Outcome } from '@/lib/copilot/generated/request-trace-v1'
 import type { OrchestratorResult } from '@/lib/copilot/request/types'
 
 const { runCopilotLifecycle } = vi.hoisted(() => ({
@@ -34,22 +33,13 @@ describe('runHeadlessCopilotLifecycle', () => {
   beforeEach(() => {
     trace.setGlobalTracerProvider(new BasicTracerProvider())
     propagation.setGlobalPropagator(new W3CTraceContextPropagator())
-    vi.stubGlobal(
-      'fetch',
-      vi.fn().mockResolvedValue(
-        new Response(null, {
-          status: 200,
-        })
-      )
-    )
   })
 
   afterEach(() => {
     vi.clearAllMocks()
-    vi.unstubAllGlobals()
   })
 
-  it('reports a successful headless trace', async () => {
+  it('runs the lifecycle and returns its result', async () => {
     runCopilotLifecycle.mockResolvedValueOnce(
       createLifecycleResult({
         usage: { prompt: 10, completion: 5 },
@@ -77,32 +67,13 @@ describe('runHeadlessCopilotLifecycle', () => {
       expect.objectContaining({
         simRequestId: 'req-1',
         trace: expect.any(Object),
+        otelContext: expect.any(Object),
         chatId: 'chat-1',
       })
     )
-
-    expect(fetch).toHaveBeenCalledTimes(1)
-    const [url, init] = vi.mocked(fetch).mock.calls[0] as [string, RequestInit]
-    expect(url).toContain('/api/traces')
-    const body = JSON.parse(String(init.body))
-    expect(body).toEqual(
-      expect.objectContaining({
-        simRequestId: 'req-1',
-        outcome: RequestTraceV1Outcome.success,
-        chatId: 'chat-1',
-        usage: expect.objectContaining({
-          inputTokens: 10,
-          outputTokens: 5,
-        }),
-        cost: {
-          rawTotalCost: 3,
-          billedTotalCost: 3,
-        },
-      })
-    )
   })
 
-  it('reports an error trace when the lifecycle result is unsuccessful', async () => {
+  it('returns an unsuccessful result from the lifecycle', async () => {
     runCopilotLifecycle.mockResolvedValueOnce(
       createLifecycleResult({
         success: false,
@@ -125,9 +96,6 @@ describe('runHeadlessCopilotLifecycle', () => {
     )
 
     expect(result.success).toBe(false)
-    const [, init] = vi.mocked(fetch).mock.calls[0] as [string, RequestInit]
-    const body = JSON.parse(String(init.body))
-    expect(body.outcome).toBe(RequestTraceV1Outcome.error)
   })
 
   it('prefers an explicit simRequestId over the payload messageId', async () => {
@@ -154,13 +122,9 @@ describe('runHeadlessCopilotLifecycle', () => {
         simRequestId: 'workflow-request-id',
       })
     )
-
-    const [, init] = vi.mocked(fetch).mock.calls[0] as [string, RequestInit]
-    const body = JSON.parse(String(init.body))
-    expect(body.simRequestId).toBe('workflow-request-id')
   })
 
-  it('passes an OTel context to the lifecycle and trace report', async () => {
+  it('threads a valid OTel context into the lifecycle', async () => {
     let lifecycleTraceparent = ''
     runCopilotLifecycle.mockImplementationOnce(async (_payload, options) => {
       const { traceHeaders } = await import('@/lib/copilot/request/go/propagation')
@@ -183,18 +147,9 @@ describe('runHeadlessCopilotLifecycle', () => {
     )
 
     expect(lifecycleTraceparent).toMatch(/^00-[0-9a-f]{32}-[0-9a-f]{16}-0[0-9a-f]$/)
-    const [, init] = vi.mocked(fetch).mock.calls[0] as [string, RequestInit]
-    const headers = init.headers as Record<string, string>
-    // The outbound trace report now runs inside its own OTel child span, so
-    // traceparent has the same trace-id as the lifecycle but a different
-    // span-id. Both must stay on the same trace.
-    const lifecycleTraceId = lifecycleTraceparent.split('-')[1]
-    expect(headers.traceparent).toMatch(/^00-[0-9a-f]{32}-[0-9a-f]{16}-0[0-9a-f]$/)
-    expect(headers.traceparent.split('-')[1]).toBe(lifecycleTraceId)
-    expect(headers.traceparent.split('-')[2]).not.toBe(lifecycleTraceparent.split('-')[2])
   })
 
-  it('reports an error trace when the lifecycle throws', async () => {
+  it('rethrows when the lifecycle throws', async () => {
     runCopilotLifecycle.mockRejectedValueOnce(new Error('kaboom'))
 
     await expect(
@@ -212,9 +167,5 @@ describe('runHeadlessCopilotLifecycle', () => {
         }
       )
     ).rejects.toThrow('kaboom')
-
-    const [, init] = vi.mocked(fetch).mock.calls[0] as [string, RequestInit]
-    const body = JSON.parse(String(init.body))
-    expect(body.outcome).toBe(RequestTraceV1Outcome.error)
   })
 })

apps/sim/lib/copilot/tools/handlers/workflow/mutations.test.ts

@@ -1,6 +1,7 @@
 /**
  * @vitest-environment node
  */
+import { createEnvMock } from '@sim/testing'
 import { beforeEach, describe, expect, it, vi } from 'vitest'
 
 const {
@@ -34,9 +35,7 @@ vi.mock('@/lib/api-key/orchestration', () => ({
   performCreateWorkspaceApiKey: vi.fn(),
 }))
 
-vi.mock('@/lib/core/config/env', () => ({
-  env: { INTERNAL_API_SECRET: 'secret' },
-}))
+vi.mock('@/lib/core/config/env', () => createEnvMock({ INTERNAL_API_SECRET: 'secret' }))
 
 vi.mock('@/lib/core/utils/request', () => ({
   generateRequestId: () => 'request-1',

apps/sim/lib/mcp/service.test.ts

@@ -13,11 +13,38 @@ const {
   mockValidateDomain,
   mockValidateSsrf,
   mockIsDomainAllowed,
+  mockCacheAdapter,
 } = vi.hoisted(() => {
   const mockListTools = vi.fn()
   const mockConnect = vi.fn()
   const mockDisconnect = vi.fn()
+  // In-memory cache adapter so the service never touches the real Redis the
+  // local .env points at (unreachable in CI/sandbox → hangs). Honors TTL via
+  // an expiry timestamp so negative-cache assertions behave like production.
+  const cacheStore = new Map<string, { tools: unknown[]; expiry: number }>()
+  const mockCacheAdapter = {
+    get: async (key: string) => {
+      const entry = cacheStore.get(key)
+      if (!entry) return null
+      if (entry.expiry <= Date.now()) {
+        cacheStore.delete(key)
+        return null
+      }
+      return entry
+    },
+    set: async (key: string, tools: unknown[], ttlMs: number) => {
+      cacheStore.set(key, { tools, expiry: Date.now() + ttlMs })
+    },
+    delete: async (key: string) => {
+      cacheStore.delete(key)
+    },
+    clear: async () => {
+      cacheStore.clear()
+    },
+    dispose: () => {},
+  }
   return {
+    mockCacheAdapter,
     MockMcpClient: vi.fn().mockImplementation(
       class {
         constructor() {
@@ -91,6 +118,11 @@ vi.mock('@/lib/mcp/resolve-config', () => ({
   resolveMcpConfigEnvVars: (...args: unknown[]) => mockResolveEnvVars(...args),
 }))
 
+vi.mock('@/lib/mcp/storage', () => ({
+  createMcpCacheAdapter: () => mockCacheAdapter,
+  getMcpCacheType: () => 'memory',
+}))
+
 import { mcpService } from '@/lib/mcp/service'
 import { McpOauthAuthorizationRequiredError } from '@/lib/mcp/types'
 

packages/testing/src/mocks/audit.mock.ts

@@ -112,6 +112,7 @@ export const auditMock = {
     ORG_MEMBER_ADDED: 'org_member.added',
     ORG_MEMBER_REMOVED: 'org_member.removed',
     ORG_MEMBER_ROLE_CHANGED: 'org_member.role_changed',
+    ORG_MEMBER_USAGE_LIMIT_CHANGED: 'org_member.usage_limit_changed',
     ORG_INVITATION_CREATED: 'org_invitation.created',
     ORG_INVITATION_UPDATED: 'org_invitation.updated',
     ORG_INVITATION_ACCEPTED: 'org_invitation.accepted',