Skip to content

Commit d5af5f9

Browse files
committed
fix(tests): replace banned Function type with explicit callback signature
1 parent c00a52d commit d5af5f9

File tree

2 files changed

+44
-42
lines changed

2 files changed

+44
-42
lines changed

apps/sim/app/api/chat/[identifier]/otp/route.test.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ vi.mock('@sim/db', () => ({
8080
select: mockDbSelect,
8181
insert: mockDbInsert,
8282
delete: mockDbDelete,
83-
transaction: vi.fn(async (callback: Function) => {
83+
transaction: vi.fn(async (callback: (tx: Record<string, unknown>) => unknown) => {
8484
return callback({
8585
select: mockDbSelect,
8686
insert: mockDbInsert,

apps/sim/app/api/tools/custom/route.test.ts

Lines changed: 43 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -117,47 +117,49 @@ vi.mock('@sim/db', () => ({
117117
insert: (...args: unknown[]) => mockInsert(...args),
118118
update: (...args: unknown[]) => mockUpdate(...args),
119119
delete: (...args: unknown[]) => mockDelete(...args),
120-
transaction: vi.fn().mockImplementation(async (callback: Function) => {
121-
const txMockSelect = vi.fn().mockReturnValue({ from: mockFrom })
122-
const txMockInsert = vi.fn().mockReturnValue({ values: mockValues })
123-
const txMockUpdate = vi.fn().mockReturnValue({ set: mockSet })
124-
const txMockDelete = vi.fn().mockReturnValue({ where: mockWhere })
125-
126-
const txMockOrderBy = vi.fn().mockImplementation(() => {
127-
const queryBuilder = {
128-
limit: mockLimit,
129-
then: (resolve: (value: typeof sampleTools) => void) => {
130-
resolve(sampleTools)
131-
return queryBuilder
132-
},
133-
catch: (_reject: (error: Error) => void) => queryBuilder,
134-
}
135-
return queryBuilder
136-
})
137-
138-
const txMockWhere = vi.fn().mockImplementation(() => {
139-
const queryBuilder = {
140-
orderBy: txMockOrderBy,
141-
limit: mockLimit,
142-
then: (resolve: (value: typeof sampleTools) => void) => {
143-
resolve(sampleTools)
144-
return queryBuilder
145-
},
146-
catch: (_reject: (error: Error) => void) => queryBuilder,
147-
}
148-
return queryBuilder
149-
})
150-
151-
const txMockFrom = vi.fn().mockReturnValue({ where: txMockWhere })
152-
txMockSelect.mockReturnValue({ from: txMockFrom })
153-
154-
return await callback({
155-
select: txMockSelect,
156-
insert: txMockInsert,
157-
update: txMockUpdate,
158-
delete: txMockDelete,
159-
})
160-
}),
120+
transaction: vi
121+
.fn()
122+
.mockImplementation(async (callback: (tx: Record<string, unknown>) => unknown) => {
123+
const txMockSelect = vi.fn().mockReturnValue({ from: mockFrom })
124+
const txMockInsert = vi.fn().mockReturnValue({ values: mockValues })
125+
const txMockUpdate = vi.fn().mockReturnValue({ set: mockSet })
126+
const txMockDelete = vi.fn().mockReturnValue({ where: mockWhere })
127+
128+
const txMockOrderBy = vi.fn().mockImplementation(() => {
129+
const queryBuilder = {
130+
limit: mockLimit,
131+
then: (resolve: (value: typeof sampleTools) => void) => {
132+
resolve(sampleTools)
133+
return queryBuilder
134+
},
135+
catch: (_reject: (error: Error) => void) => queryBuilder,
136+
}
137+
return queryBuilder
138+
})
139+
140+
const txMockWhere = vi.fn().mockImplementation(() => {
141+
const queryBuilder = {
142+
orderBy: txMockOrderBy,
143+
limit: mockLimit,
144+
then: (resolve: (value: typeof sampleTools) => void) => {
145+
resolve(sampleTools)
146+
return queryBuilder
147+
},
148+
catch: (_reject: (error: Error) => void) => queryBuilder,
149+
}
150+
return queryBuilder
151+
})
152+
153+
const txMockFrom = vi.fn().mockReturnValue({ where: txMockWhere })
154+
txMockSelect.mockReturnValue({ from: txMockFrom })
155+
156+
return await callback({
157+
select: txMockSelect,
158+
insert: txMockInsert,
159+
update: txMockUpdate,
160+
delete: txMockDelete,
161+
})
162+
}),
161163
},
162164
}))
163165

0 commit comments

Comments
 (0)