11import { dbChainMockFns , resetDbChainMock } from '@sim/testing'
22import { billingStorageMock , billingStorageMockFns } from '@sim/testing/mocks/billing-storage.mock'
3- import { storageServiceMock , storageServiceMockFns } from '@sim/testing/mocks/storage-service.mock'
3+ import { storageServiceMock } from '@sim/testing/mocks/storage-service.mock'
44import {
55 uploadsMetadataMock ,
66 uploadsMetadataMockFns ,
@@ -22,11 +22,7 @@ vi.mock('@/lib/uploads/core/storage-service', () => storageServiceMock)
2222
2323vi . mock ( '@/lib/uploads/server/metadata' , ( ) => uploadsMetadataMock )
2424
25- import {
26- createDocumentRecords ,
27- createSingleDocument ,
28- deleteDocumentStorageFiles ,
29- } from '@/lib/knowledge/documents/service'
25+ import { createDocumentRecords , createSingleDocument } from '@/lib/knowledge/documents/service'
3026
3127const mockCheckStorageQuotaForBillingContext =
3228 billingStorageMockFns . mockCheckStorageQuotaForBillingContext
@@ -36,8 +32,6 @@ const mockMaybeNotifyStorageLimitForBillingContext =
3632 billingStorageMockFns . mockMaybeNotifyStorageLimitForBillingContext
3733const mockResolveStorageBillingContext = billingStorageMockFns . mockResolveStorageBillingContext
3834
39- const mockDeleteFile = storageServiceMockFns . mockDeleteFile
40- const mockDeleteFileMetadataByIdentity = uploadsMetadataMockFns . mockDeleteFileMetadataByIdentity
4135const mockGetFileMetadataByKeys = uploadsMetadataMockFns . mockGetFileMetadataByKeys
4236const mockGetBoundWorkspaceFileSecretProvenanceByMetadata =
4337 workspaceFileSecretProvenanceMockFns . mockGetBoundWorkspaceFileSecretProvenanceByMetadata
@@ -228,50 +222,4 @@ describe('knowledge workspace source provenance', () => {
228222 } )
229223 }
230224 } )
231-
232- it ( 'never deletes a referenced workspace source as knowledge-base storage' , async ( ) => {
233- await deleteDocumentStorageFiles (
234- [ { id : 'document-1' , fileUrl : SOURCE_URL , workspaceId : WORKSPACE_ID } ] ,
235- 'request-1'
236- )
237-
238- expect ( mockGetFileMetadataByKeys ) . not . toHaveBeenCalled ( )
239- expect ( mockDeleteFile ) . not . toHaveBeenCalled ( )
240- expect ( mockDeleteFileMetadataByIdentity ) . not . toHaveBeenCalled ( )
241- } )
242-
243- it . each ( [ 'org-1' , 'org-2' , null ] ) (
244- 'only queues an organization cache for its exact owner: %s' ,
245- async ( organizationId ) => {
246- const storageKey = 'kb/org-source.pdf'
247- mockGetFileMetadataByKeys . mockResolvedValue ( [
248- {
249- ...SOURCE_BINDING ,
250- key : storageKey ,
251- context : 'knowledge-base' ,
252- workspaceId : null ,
253- organizationId : 'org-1' ,
254- } ,
255- ] )
256- const cleanup = deleteDocumentStorageFiles (
257- [
258- {
259- id : 'org-doc' ,
260- fileUrl : `/api/files/serve/${ encodeURIComponent ( storageKey ) } ` ,
261- workspaceId : null ,
262- organizationId,
263- } ,
264- ] ,
265- 'request-1'
266- )
267- if ( organizationId === 'org-1' ) {
268- await expect ( cleanup ) . resolves . toBeUndefined ( )
269- expect ( dbChainMockFns . values ) . toHaveBeenCalledOnce ( )
270- } else {
271- await expect ( cleanup ) . rejects . toThrow ( )
272- expect ( dbChainMockFns . values ) . not . toHaveBeenCalled ( )
273- }
274- expect ( mockDeleteFile ) . not . toHaveBeenCalled ( )
275- }
276- )
277225} )
0 commit comments