1+ import { schemaFromCode } from '@immich/sql-tools' ;
12import { Kysely } from 'kysely' ;
23import { DateTime } from 'luxon' ;
34import { AssetMetadataKey , UserMetadataKey } from 'src/enum' ;
@@ -6,7 +7,6 @@ import { LoggingRepository } from 'src/repositories/logging.repository';
67import { BaseSync , SyncRepository } from 'src/repositories/sync.repository' ;
78import { DB } from 'src/schema' ;
89import { SyncService } from 'src/services/sync.service' ;
9- import { getRegisteredItems } from 'src/sql-tools/register' ;
1010import { newMediumService } from 'test/medium.factory' ;
1111import { getKyselyDB } from 'test/utils' ;
1212import { v4 } from 'uuid' ;
@@ -227,10 +227,9 @@ describe(SyncService.name, () => {
227227 it ( 'should cleanup every table' , async ( ) => {
228228 const { sut } = setup ( ) ;
229229
230- const tables = getRegisteredItems ( ) . filter ( ( t ) => t . type === 'table' ) ;
231- const auditTables = tables
232- . filter ( ( t ) => t . item . options . name ?. endsWith ( '_audit' ) )
233- . map ( ( t ) => t . item . options . name ! ) as ( keyof DB ) [ ] ;
230+ const auditTables = schemaFromCode ( )
231+ . tables . filter ( ( table ) => table . name . endsWith ( '_audit' ) )
232+ . map ( ( { name } ) => name ) ;
234233
235234 const auditCleanupSpy = vi . spyOn ( BaseSync . prototype as any , 'auditCleanup' ) ;
236235 await expect ( sut . onAuditTableCleanup ( ) ) . resolves . toBeUndefined ( ) ;
0 commit comments