import { vi, describe, expect, it, beforeAll, afterAll } from 'vitest'; import axios from 'axios'; import { downloadFile } from 'src/composables/downloadFile'; import { useSession } from 'src/composables/useSession'; const session = useSession(); const token = session.getToken(); describe('downloadFile', () => { let defaulCreateObjectURL; beforeAll(() => { vi.mock('src/composables/getUrl', () => ({ getUrl: vi.fn().mockResolvedValue(''), })); defaulCreateObjectURL = window.URL.createObjectURL; window.URL.createObjectURL = vi.fn(() => 'blob:http://localhost:9000/blob-id'); }); afterAll(() => (window.URL.createObjectURL = defaulCreateObjectURL)); it('should open a new window to download the file', async () => { const res = { data: new Blob(['file content'], { type: 'application/octet-stream' }), headers: { 'content-disposition': 'attachment; filename="test-file.txt"' }, }; vi.spyOn(axios, 'get').mockImplementation((url) => { if (url.includes('downloadFile')) return Promise.resolve(res); }); await downloadFile(1); expect(axios.get).toHaveBeenCalledWith( `/api/dms/1/downloadFile?access_token=${token}`, { responseType: 'blob' }, ); }); });