@@ -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