Skip to content

Commit 653228c

Browse files
charlesliencharleslien
committed
fix typecheck for unit tests
1 parent 29d0165 commit 653228c

File tree

3 files changed

+62
-4
lines changed

3 files changed

+62
-4
lines changed

‎web/src/app/api/v1/agent-runs/[runId]/steps/__tests__/steps.test.ts‎

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,20 @@
11
import { TEST_USER_ID } from '@codebuff/common/old-constants'
2-
import { beforeEach, describe, expect, test } from 'bun:test'
2+
import { beforeEach, describe, expect, mock, test } from 'bun:test'
33
import { NextRequest } from 'next/server'
44

55
import { postAgentRunsSteps } from '../_post'
66

77
import type { TrackEventFn } from '@codebuff/common/types/contracts/analytics'
88
import type { GetUserInfoFromApiKeyFn } from '@codebuff/common/types/contracts/database'
9-
import type { Logger } from '@codebuff/common/types/contracts/logger'
9+
import type {
10+
Logger,
11+
LoggerWithContextFn,
12+
} from '@codebuff/common/types/contracts/logger'
1013

1114
describe('agentRunsStepsPost', () => {
1215
let mockGetUserInfoFromApiKey: GetUserInfoFromApiKeyFn
1316
let mockLogger: Logger
17+
let mockLoggerWithContext: LoggerWithContextFn
1418
let mockTrackEvent: TrackEventFn
1519
let mockDb: any
1620

@@ -42,6 +46,8 @@ describe('agentRunsStepsPost', () => {
4246
debug: () => {},
4347
}
4448

49+
mockLoggerWithContext = mock(() => mockLogger)
50+
4551
mockTrackEvent = () => {}
4652

4753
// Default mock DB with successful operations
@@ -73,6 +79,7 @@ describe('agentRunsStepsPost', () => {
7379
runId: 'run-123',
7480
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
7581
logger: mockLogger,
82+
loggerWithContext: mockLoggerWithContext,
7683
trackEvent: mockTrackEvent,
7784
db: mockDb,
7885
})
@@ -97,6 +104,7 @@ describe('agentRunsStepsPost', () => {
97104
runId: 'run-123',
98105
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
99106
logger: mockLogger,
107+
loggerWithContext: mockLoggerWithContext,
100108
trackEvent: mockTrackEvent,
101109
db: mockDb,
102110
})
@@ -121,6 +129,7 @@ describe('agentRunsStepsPost', () => {
121129
runId: 'run-123',
122130
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
123131
logger: mockLogger,
132+
loggerWithContext: mockLoggerWithContext,
124133
trackEvent: mockTrackEvent,
125134
db: mockDb,
126135
})
@@ -145,6 +154,7 @@ describe('agentRunsStepsPost', () => {
145154
runId: 'run-123',
146155
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
147156
logger: mockLogger,
157+
loggerWithContext: mockLoggerWithContext,
148158
trackEvent: mockTrackEvent,
149159
db: mockDb,
150160
})
@@ -180,6 +190,7 @@ describe('agentRunsStepsPost', () => {
180190
runId: 'run-123',
181191
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
182192
logger: mockLogger,
193+
loggerWithContext: mockLoggerWithContext,
183194
trackEvent: mockTrackEvent,
184195
db: dbWithNoRun,
185196
})
@@ -215,6 +226,7 @@ describe('agentRunsStepsPost', () => {
215226
runId: 'run-123',
216227
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
217228
logger: mockLogger,
229+
loggerWithContext: mockLoggerWithContext,
218230
trackEvent: mockTrackEvent,
219231
db: dbWithDifferentUser,
220232
})
@@ -239,6 +251,7 @@ describe('agentRunsStepsPost', () => {
239251
runId: 'run-123',
240252
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
241253
logger: mockLogger,
254+
loggerWithContext: mockLoggerWithContext,
242255
trackEvent: mockTrackEvent,
243256
db: mockDb,
244257
})
@@ -269,6 +282,7 @@ describe('agentRunsStepsPost', () => {
269282
runId: 'run-123',
270283
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
271284
logger: mockLogger,
285+
loggerWithContext: mockLoggerWithContext,
272286
trackEvent: mockTrackEvent,
273287
db: mockDb,
274288
})
@@ -310,6 +324,7 @@ describe('agentRunsStepsPost', () => {
310324
runId: 'run-123',
311325
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
312326
logger: mockLogger,
327+
loggerWithContext: mockLoggerWithContext,
313328
trackEvent: mockTrackEvent,
314329
db: dbWithError,
315330
})

‎web/src/app/api/v1/agent-runs/__tests__/agent-runs.test.ts‎

Lines changed: 29 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,10 @@ import type {
1010
GetUserInfoFromApiKeyFn,
1111
GetUserInfoFromApiKeyOutput,
1212
} from '@codebuff/common/types/contracts/database'
13-
import type { Logger } from '@codebuff/common/types/contracts/logger'
13+
import type {
14+
Logger,
15+
LoggerWithContextFn,
16+
} from '@codebuff/common/types/contracts/logger'
1417

1518
describe('/api/v1/agent-runs POST endpoint', () => {
1619
const mockUserData: Record<
@@ -39,6 +42,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
3942
}
4043

4144
let mockLogger: Logger
45+
let mockLoggerWithContext: LoggerWithContextFn
4246
let mockTrackEvent: TrackEventFn
4347
let mockDb: any
4448

@@ -49,6 +53,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
4953
info: mock(() => {}),
5054
debug: mock(() => {}),
5155
}
56+
mockLoggerWithContext = mock(() => mockLogger)
5257

5358
mockTrackEvent = mock(() => {})
5459

@@ -82,6 +87,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
8287
req,
8388
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
8489
logger: mockLogger,
90+
loggerWithContext: mockLoggerWithContext,
8591
trackEvent: mockTrackEvent,
8692
db: mockDb,
8793
})
@@ -105,6 +111,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
105111
req,
106112
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
107113
logger: mockLogger,
114+
loggerWithContext: mockLoggerWithContext,
108115
trackEvent: mockTrackEvent,
109116
db: mockDb,
110117
})
@@ -128,6 +135,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
128135
req,
129136
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
130137
logger: mockLogger,
138+
loggerWithContext: mockLoggerWithContext,
131139
trackEvent: mockTrackEvent,
132140
db: mockDb,
133141
})
@@ -150,6 +158,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
150158
req,
151159
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
152160
logger: mockLogger,
161+
loggerWithContext: mockLoggerWithContext,
153162
trackEvent: mockTrackEvent,
154163
db: mockDb,
155164
})
@@ -172,6 +181,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
172181
req,
173182
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
174183
logger: mockLogger,
184+
loggerWithContext: mockLoggerWithContext,
175185
trackEvent: mockTrackEvent,
176186
db: mockDb,
177187
})
@@ -194,6 +204,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
194204
req,
195205
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
196206
logger: mockLogger,
207+
loggerWithContext: mockLoggerWithContext,
197208
trackEvent: mockTrackEvent,
198209
db: mockDb,
199210
})
@@ -216,6 +227,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
216227
req,
217228
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
218229
logger: mockLogger,
230+
loggerWithContext: mockLoggerWithContext,
219231
trackEvent: mockTrackEvent,
220232
db: mockDb,
221233
})
@@ -239,6 +251,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
239251
req,
240252
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
241253
logger: mockLogger,
254+
loggerWithContext: mockLoggerWithContext,
242255
trackEvent: mockTrackEvent,
243256
db: mockDb,
244257
})
@@ -263,6 +276,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
263276
req,
264277
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
265278
logger: mockLogger,
279+
loggerWithContext: mockLoggerWithContext,
266280
trackEvent: mockTrackEvent,
267281
db: mockDb,
268282
})
@@ -286,6 +300,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
286300
req,
287301
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
288302
logger: mockLogger,
303+
loggerWithContext: mockLoggerWithContext,
289304
trackEvent: mockTrackEvent,
290305
db: mockDb,
291306
})
@@ -324,6 +339,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
324339
req,
325340
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
326341
logger: mockLogger,
342+
loggerWithContext: mockLoggerWithContext,
327343
trackEvent: mockTrackEvent,
328344
db: mockDb,
329345
})
@@ -358,6 +374,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
358374
req,
359375
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
360376
logger: mockLogger,
377+
loggerWithContext: mockLoggerWithContext,
361378
trackEvent: mockTrackEvent,
362379
db: mockDb,
363380
})
@@ -394,6 +411,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
394411
req,
395412
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
396413
logger: mockLogger,
414+
loggerWithContext: mockLoggerWithContext,
397415
trackEvent: mockTrackEvent,
398416
db: mockDb,
399417
})
@@ -431,6 +449,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
431449
req,
432450
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
433451
logger: mockLogger,
452+
loggerWithContext: mockLoggerWithContext,
434453
trackEvent: mockTrackEvent,
435454
db: mockDb,
436455
})
@@ -458,6 +477,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
458477
req,
459478
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
460479
logger: mockLogger,
480+
loggerWithContext: mockLoggerWithContext,
461481
trackEvent: mockTrackEvent,
462482
db: mockDb,
463483
})
@@ -485,6 +505,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
485505
req,
486506
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
487507
logger: mockLogger,
508+
loggerWithContext: mockLoggerWithContext,
488509
trackEvent: mockTrackEvent,
489510
db: mockDb,
490511
})
@@ -512,6 +533,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
512533
req,
513534
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
514535
logger: mockLogger,
536+
loggerWithContext: mockLoggerWithContext,
515537
trackEvent: mockTrackEvent,
516538
db: mockDb,
517539
})
@@ -539,6 +561,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
539561
req,
540562
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
541563
logger: mockLogger,
564+
loggerWithContext: mockLoggerWithContext,
542565
trackEvent: mockTrackEvent,
543566
db: mockDb,
544567
})
@@ -582,6 +605,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
582605
req,
583606
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
584607
logger: mockLogger,
608+
loggerWithContext: mockLoggerWithContext,
585609
trackEvent: mockTrackEvent,
586610
db: mockDb,
587611
})
@@ -623,6 +647,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
623647
req,
624648
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
625649
logger: mockLogger,
650+
loggerWithContext: mockLoggerWithContext,
626651
trackEvent: mockTrackEvent,
627652
db: mockDb,
628653
})
@@ -664,6 +689,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
664689
req,
665690
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
666691
logger: mockLogger,
692+
loggerWithContext: mockLoggerWithContext,
667693
trackEvent: mockTrackEvent,
668694
db: mockDb,
669695
})
@@ -698,6 +724,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
698724
req,
699725
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
700726
logger: mockLogger,
727+
loggerWithContext: mockLoggerWithContext,
701728
trackEvent: mockTrackEvent,
702729
db: mockDb,
703730
})
@@ -736,6 +763,7 @@ describe('/api/v1/agent-runs POST endpoint', () => {
736763
req,
737764
getUserInfoFromApiKey: mockGetUserInfoFromApiKey,
738765
logger: mockLogger,
766+
loggerWithContext: mockLoggerWithContext,
739767
trackEvent: mockTrackEvent,
740768
db: mockDb,
741769
})

0 commit comments

Comments
 (0)