@@ -236,7 +236,22 @@ describe('POST /api/v2/workflows/[id]/execute', () => {
236236 rateLimitSubscription : null ,
237237 keyType : 'workspace' ,
238238 } )
239- dbChainMockFns . limit . mockResolvedValue ( [ applicationContext ] )
239+ dbChainMockFns . limit
240+ . mockResolvedValueOnce ( [
241+ {
242+ workflowId : workflowRecord . id ,
243+ workflow : workflowRecord ,
244+ workspaceId : workflowRecord . workspaceId ,
245+ } ,
246+ ] )
247+ . mockResolvedValueOnce ( [
248+ {
249+ id : applicationContext . workspaceId ,
250+ organizationId : applicationContext . workspaceOrganizationId ,
251+ allowPersonalApiKeys : applicationContext . allowPersonalApiKeys ,
252+ billedAccountUserId : applicationContext . billedAccountUserId ,
253+ } ,
254+ ] )
240255 mockAuthorize . mockResolvedValue ( { allowed : true , workflow : workflowRecord } )
241256 mockClaimExecutionId . mockImplementation ( async ( executionId : string ) => ( {
242257 key : `workflow-execution-id:${ executionId } ` ,
@@ -420,9 +435,23 @@ describe('POST /api/v2/workflows/[id]/execute', () => {
420435 rateLimitSubscription : null ,
421436 keyType : 'personal' ,
422437 } )
423- dbChainMockFns . limit . mockResolvedValueOnce ( [
424- { ...applicationContext , allowPersonalApiKeys : false } ,
425- ] )
438+ dbChainMockFns . limit . mockReset ( )
439+ dbChainMockFns . limit
440+ . mockResolvedValueOnce ( [
441+ {
442+ workflowId : workflowRecord . id ,
443+ workflow : workflowRecord ,
444+ workspaceId : workflowRecord . workspaceId ,
445+ } ,
446+ ] )
447+ . mockResolvedValueOnce ( [
448+ {
449+ id : applicationContext . workspaceId ,
450+ organizationId : applicationContext . workspaceOrganizationId ,
451+ allowPersonalApiKeys : false ,
452+ billedAccountUserId : applicationContext . billedAccountUserId ,
453+ } ,
454+ ] )
426455
427456 const res = await callExecute ( { input : { } } )
428457
@@ -487,6 +516,7 @@ describe('POST /api/v2/workflows/[id]/execute', () => {
487516 } )
488517
489518 it ( 'runs the anonymous public path sync but refuses async' , async ( ) => {
519+ dbChainMockFns . limit . mockReset ( )
490520 dbChainMockFns . limit . mockResolvedValueOnce ( [
491521 { isPublicApi : true , isDeployed : true , userId : 'owner-1' , workspaceId : 'workspace-1' } ,
492522 ] )
@@ -526,6 +556,7 @@ describe('POST /api/v2/workflows/[id]/execute', () => {
526556 } )
527557
528558 it ( '401s non-public workflows without a key' , async ( ) => {
559+ dbChainMockFns . limit . mockReset ( )
529560 dbChainMockFns . limit . mockResolvedValueOnce ( [
530561 { isPublicApi : false , isDeployed : true , userId : 'owner-1' , workspaceId : 'workspace-1' } ,
531562 ] )
@@ -551,6 +582,7 @@ describe('POST /api/v2/workflows/[id]/execute', () => {
551582 } )
552583
553584 it ( 'returns a safe error when canonical workflow lookup fails' , async ( ) => {
585+ dbChainMockFns . limit . mockReset ( )
554586 dbChainMockFns . limit . mockRejectedValueOnce ( new Error ( 'database connection details' ) )
555587
556588 const response = await callExecute ( { input : { } } )
0 commit comments