122 lines
4.0 KiB
TypeScript
122 lines
4.0 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from 'vitest';
|
|
import { mock } from 'vitest-mock-extended';
|
|
import { downloadMedia, downloadURL } from '../../src/utils/download';
|
|
import { homeAssistantSignPath } from '../../src/utils/ha';
|
|
import { ViewMedia } from '../../src/view/media';
|
|
import { createCameraManager, createHASS } from '../test-utils';
|
|
|
|
vi.mock('../../src/utils/ha');
|
|
|
|
const media = new ViewMedia('clip', 'camera-1');
|
|
|
|
// @vitest-environment jsdom
|
|
describe('downloadURL', () => {
|
|
afterEach(() => {
|
|
vi.restoreAllMocks();
|
|
global.window.location = mock<Location>();
|
|
});
|
|
|
|
it('should download same origin via link', () => {
|
|
const location: Location & { origin: string } = mock<Location>();
|
|
location.origin = 'http://foo';
|
|
global.window.location = location;
|
|
|
|
const link = document.createElement('a');
|
|
link.click = vi.fn();
|
|
link.setAttribute = vi.fn();
|
|
vi.spyOn(document, 'createElement').mockReturnValue(link);
|
|
|
|
downloadURL('http://foo/url.mp4');
|
|
|
|
expect(link.href).toBe('http://foo/url.mp4');
|
|
expect(link.setAttribute).toBeCalledWith('download', 'download');
|
|
expect(link.click).toBeCalled();
|
|
});
|
|
|
|
it('should download data URL via link', () => {
|
|
const link = document.createElement('a');
|
|
link.click = vi.fn();
|
|
link.setAttribute = vi.fn();
|
|
vi.spyOn(document, 'createElement').mockReturnValue(link);
|
|
|
|
downloadURL('data:text/plain;charset=utf-8;base64,VEhJUyBJUyBEQVRB');
|
|
|
|
expect(link.href).toBe('data:text/plain;charset=utf-8;base64,VEhJUyBJUyBEQVRB');
|
|
expect(link.setAttribute).toBeCalledWith('download', 'download');
|
|
expect(link.click).toBeCalled();
|
|
});
|
|
|
|
it('should download in apps via window.open', () => {
|
|
// Set the origin to the same.
|
|
const location: Location & { origin: string } = mock<Location>();
|
|
location.origin = 'http://foo';
|
|
global.window.location = location;
|
|
|
|
vi.stubGlobal('navigator', {
|
|
userAgent: 'Home Assistant/2023.3.0-3260 (Android 13; Pixel 7 Pro)',
|
|
});
|
|
|
|
const windowSpy = vi.spyOn(window, 'open').mockReturnValue(null);
|
|
|
|
downloadURL('http://foo/url.mp4');
|
|
expect(windowSpy).toBeCalledWith('http://foo/url.mp4', '_blank');
|
|
});
|
|
});
|
|
|
|
describe('downloadMedia', () => {
|
|
afterEach(() => {
|
|
vi.resetAllMocks();
|
|
global.window.location = mock<Location>();
|
|
});
|
|
|
|
it('should throw error when no media', () => {
|
|
const cameraManager = createCameraManager();
|
|
vi.mocked(cameraManager.getMediaDownloadPath).mockResolvedValue(null);
|
|
|
|
expect(downloadMedia(createHASS(), cameraManager, media)).rejects.toThrow(
|
|
/No media to download/,
|
|
);
|
|
});
|
|
|
|
it('should throw error when signing fails', () => {
|
|
vi.spyOn(global.console, 'warn').mockReturnValue(undefined);
|
|
|
|
const cameraManager = createCameraManager();
|
|
vi.mocked(cameraManager).getMediaDownloadPath.mockResolvedValue({
|
|
sign: true,
|
|
endpoint: 'foo',
|
|
});
|
|
const signError = new Error('sign-error');
|
|
vi.mocked(homeAssistantSignPath).mockRejectedValue(signError);
|
|
|
|
expect(downloadMedia(createHASS(), cameraManager, media)).rejects.toThrow(
|
|
/Could not sign media URL for download/,
|
|
);
|
|
});
|
|
|
|
it('should download media', async () => {
|
|
const cameraManager = createCameraManager();
|
|
vi.mocked(cameraManager).getMediaDownloadPath.mockResolvedValue({
|
|
sign: true,
|
|
endpoint: 'foo',
|
|
});
|
|
vi.mocked(homeAssistantSignPath).mockResolvedValue('http://foo/signed-url');
|
|
const windowSpy = vi.spyOn(window, 'open').mockReturnValue(null);
|
|
|
|
await downloadMedia(createHASS(), cameraManager, media);
|
|
expect(windowSpy).toBeCalledWith('http://foo/signed-url', '_blank');
|
|
});
|
|
|
|
it('should download media without signing', async () => {
|
|
const cameraManager = createCameraManager();
|
|
vi.mocked(cameraManager).getMediaDownloadPath.mockResolvedValue({
|
|
sign: false,
|
|
endpoint: 'https://foo/',
|
|
});
|
|
const windowSpy = vi.spyOn(window, 'open').mockReturnValue(null);
|
|
|
|
await downloadMedia(createHASS(), cameraManager, media);
|
|
expect(windowSpy).toBeCalledWith('https://foo/', '_blank');
|
|
});
|
|
});
|