77
88import * as versionUtils from '../version-utils'
99
10+ import type { CodebuffPgDatabase } from '../../db/types'
11+
1012const {
1113 versionOne,
1214 parseVersion,
@@ -127,7 +129,7 @@ describe('version-utils', () => {
127129
128130 describe ( 'getLatestAgentVersion' , ( ) => {
129131 it ( 'should return version 0.0.0 when no agent exists' , async ( ) => {
130- const mockDb = createVersionQueryDbMock ( [ ] )
132+ const mockDb = createVersionQueryDbMock ( [ ] ) as unknown as CodebuffPgDatabase
131133
132134 const result = await getLatestAgentVersion ( {
133135 agentId : 'test-agent' ,
@@ -138,7 +140,9 @@ describe('version-utils', () => {
138140 } )
139141
140142 it ( 'should return latest version when agent exists' , async ( ) => {
141- const mockDb = createVersionQueryDbMock ( [ { major : 1 , minor : 2 , patch : 3 } ] )
143+ const mockDb = createVersionQueryDbMock ( [
144+ { major : 1 , minor : 2 , patch : 3 } ,
145+ ] ) as unknown as CodebuffPgDatabase
142146
143147 const result = await getLatestAgentVersion ( {
144148 agentId : 'test-agent' ,
@@ -149,7 +153,9 @@ describe('version-utils', () => {
149153 } )
150154
151155 it ( 'should handle null values in database response' , async ( ) => {
152- const mockDb = createVersionQueryDbMock ( [ { major : null , minor : null , patch : null } ] )
156+ const mockDb = createVersionQueryDbMock ( [
157+ { major : null , minor : null , patch : null } ,
158+ ] ) as unknown as CodebuffPgDatabase
153159
154160 const result = await getLatestAgentVersion ( {
155161 agentId : 'test-agent' ,
@@ -162,7 +168,9 @@ describe('version-utils', () => {
162168
163169 describe ( 'determineNextVersion' , ( ) => {
164170 it ( 'should increment patch of latest version when no version provided' , async ( ) => {
165- const mockDb = createVersionQueryDbMock ( [ { major : 1 , minor : 2 , patch : 3 } ] )
171+ const mockDb = createVersionQueryDbMock ( [
172+ { major : 1 , minor : 2 , patch : 3 } ,
173+ ] ) as unknown as CodebuffPgDatabase
166174
167175 const result = await determineNextVersion ( {
168176 agentId : 'test-agent' ,
@@ -173,7 +181,7 @@ describe('version-utils', () => {
173181 } )
174182
175183 it ( 'should use provided version when higher than latest' , async ( ) => {
176- const mockDb = createVersionQueryDbMock ( [ ] )
184+ const mockDb = createVersionQueryDbMock ( [ ] ) as unknown as CodebuffPgDatabase
177185
178186 const result = await determineNextVersion ( {
179187 agentId : 'test-agent' ,
@@ -185,7 +193,9 @@ describe('version-utils', () => {
185193 } )
186194
187195 it ( 'should throw error when provided version is not greater than latest' , async ( ) => {
188- const mockDb = createVersionQueryDbMock ( [ { major : 2 , minor : 0 , patch : 0 } ] )
196+ const mockDb = createVersionQueryDbMock ( [
197+ { major : 2 , minor : 0 , patch : 0 } ,
198+ ] ) as unknown as CodebuffPgDatabase
189199
190200 await expect (
191201 determineNextVersion ( {
@@ -200,7 +210,9 @@ describe('version-utils', () => {
200210 } )
201211
202212 it ( 'should throw error when provided version equals latest' , async ( ) => {
203- const mockDb = createVersionQueryDbMock ( [ { major : 1 , minor : 5 , patch : 0 } ] )
213+ const mockDb = createVersionQueryDbMock ( [
214+ { major : 1 , minor : 5 , patch : 0 } ,
215+ ] ) as unknown as CodebuffPgDatabase
204216
205217 await expect (
206218 determineNextVersion ( {
@@ -215,7 +227,7 @@ describe('version-utils', () => {
215227 } )
216228
217229 it ( 'should throw error for invalid provided version' , async ( ) => {
218- const mockDb = createVersionQueryDbMock ( [ ] )
230+ const mockDb = createVersionQueryDbMock ( [ ] ) as unknown as CodebuffPgDatabase
219231
220232 await expect (
221233 determineNextVersion ( {
@@ -232,7 +244,9 @@ describe('version-utils', () => {
232244
233245 describe ( 'versionExists' , ( ) => {
234246 it ( 'should return true when version exists' , async ( ) => {
235- const mockDb = createExistsQueryDbMock ( [ { id : 'test-agent' } ] )
247+ const mockDb = createExistsQueryDbMock ( [
248+ { id : 'test-agent' } ,
249+ ] ) as unknown as CodebuffPgDatabase
236250
237251 const result = await versionExists ( {
238252 agentId : 'test-agent' ,
@@ -244,7 +258,7 @@ describe('version-utils', () => {
244258 } )
245259
246260 it ( 'should return false when version does not exist' , async ( ) => {
247- const mockDb = createExistsQueryDbMock ( [ ] )
261+ const mockDb = createExistsQueryDbMock ( [ ] ) as unknown as CodebuffPgDatabase
248262
249263 const result = await versionExists ( {
250264 agentId : 'test-agent' ,
0 commit comments