33 *
44 * @vitest -environment node
55 */
6- import { loggerMock } from '@sim/testing'
6+ import { auditMock , loggerMock } from '@sim/testing'
77import { NextRequest } from 'next/server'
88import { afterEach , beforeEach , describe , expect , it , vi } from 'vitest'
99
10+ vi . mock ( '@/lib/audit/log' , ( ) => auditMock )
11+
1012vi . mock ( '@/lib/core/config/feature-flags' , ( ) => ( {
1113 isDev : true ,
1214 isHosted : false ,
@@ -48,7 +50,14 @@ describe('Chat Edit API Route', () => {
4850 } ) )
4951
5052 vi . doMock ( '@sim/db/schema' , ( ) => ( {
51- chat : { id : 'id' , identifier : 'identifier' , userId : 'userId' } ,
53+ chat : {
54+ id : 'id' ,
55+ identifier : 'identifier' ,
56+ userId : 'userId' ,
57+ workflowId : 'workflowId' ,
58+ title : 'title' ,
59+ } ,
60+ workflow : { id : 'id' , workspaceId : 'workspaceId' } ,
5261 } ) )
5362
5463 // Mock logger - use loggerMock from @sim/testing
@@ -217,7 +226,7 @@ describe('Chat Edit API Route', () => {
217226 }
218227
219228 mockCheckChatAccess . mockResolvedValue ( { hasAccess : true , chat : mockChat } )
220- mockLimit . mockResolvedValueOnce ( [ ] ) // No identifier conflict
229+ mockLimit . mockResolvedValueOnce ( [ { workspaceId : 'workspace-123' } ] )
221230
222231 const req = new NextRequest ( 'http://localhost:3000/api/chat/manage/chat-123' , {
223232 method : 'PATCH' ,
@@ -312,7 +321,7 @@ describe('Chat Edit API Route', () => {
312321 }
313322
314323 mockCheckChatAccess . mockResolvedValue ( { hasAccess : true , chat : mockChat } )
315- mockLimit . mockResolvedValueOnce ( [ ] )
324+ mockLimit . mockResolvedValueOnce ( [ { workspaceId : 'workspace-123' } ] )
316325
317326 const req = new NextRequest ( 'http://localhost:3000/api/chat/manage/chat-123' , {
318327 method : 'PATCH' ,
@@ -372,7 +381,8 @@ describe('Chat Edit API Route', () => {
372381 } ) )
373382
374383 mockCheckChatAccess . mockResolvedValue ( { hasAccess : true } )
375- mockWhere . mockResolvedValue ( undefined )
384+ mockLimit . mockResolvedValueOnce ( [ { workflowId : 'workflow-123' , title : 'Test Chat' } ] )
385+ mockLimit . mockResolvedValueOnce ( [ { workspaceId : 'workspace-123' } ] )
376386
377387 const req = new NextRequest ( 'http://localhost:3000/api/chat/manage/chat-123' , {
378388 method : 'DELETE' ,
@@ -394,7 +404,8 @@ describe('Chat Edit API Route', () => {
394404 } ) )
395405
396406 mockCheckChatAccess . mockResolvedValue ( { hasAccess : true } )
397- mockWhere . mockResolvedValue ( undefined )
407+ mockLimit . mockResolvedValueOnce ( [ { workflowId : 'workflow-123' , title : 'Test Chat' } ] )
408+ mockLimit . mockResolvedValueOnce ( [ { workspaceId : 'workspace-123' } ] )
398409
399410 const req = new NextRequest ( 'http://localhost:3000/api/chat/manage/chat-123' , {
400411 method : 'DELETE' ,
0 commit comments