38 lines
1.3 KiB
JavaScript
38 lines
1.3 KiB
JavaScript
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' },
|
|
);
|
|
});
|
|
});
|