Move card-controller out of utils.

2023-10-03 20:14:08 -07:00
parent 913355ad1d
commit ed91c87899
45 changed files with 8008 additions and 0 deletions
@@ -0,0 +1,94 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+import { mock } from 'vitest-mock-extended';
+import { FrigateCardMediaPlayer } from '../../src/types';
+import { DownloadManager } from '../../src/card-controller/download-manager';
+import { downloadMedia, downloadURL } from '../../src/utils/download.js';
+import {
+  createCardAPI,
+  createHASS,
+  createMediaLoadedInfo,
+  createViewWithMedia,
+} from '../test-utils';
+
+vi.mock('../../src/utils/download.js');
+
+describe('DownloadManager.downloadViewerMedia', () => {
+  beforeEach(() => {
+    vi.resetAllMocks();
+  });
+
+  it('should download', async () => {
+    const api = createCardAPI();
+    vi.mocked(api.getViewManager().getView).mockReturnValue(createViewWithMedia());
+    vi.mocked(api.getHASSManager().getHASS).mockReturnValue(createHASS());
+    const manager = new DownloadManager(api);
+
+    expect(await manager.downloadViewerMedia()).toBeTruthy();
+    expect(downloadMedia).toBeCalledWith(
+      api.getHASSManager().getHASS(),
+      api.getCameraManager(),
+      api.getViewManager().getView()?.queryResults?.getResult(0),
+    );
+  });
+
+  it('should not download due to exception thrown', async () => {
+    const api = createCardAPI();
+    vi.mocked(api.getViewManager().getView).mockReturnValue(createViewWithMedia());
+    vi.mocked(api.getHASSManager().getHASS).mockReturnValue(createHASS());
+    const manager = new DownloadManager(api);
+
+    const error = new Error();
+    vi.mocked(downloadMedia).mockRejectedValue(error);
+
+    expect(await manager.downloadViewerMedia()).toBeFalsy();
+    expect(api.getMessageManager().setErrorIfHigherPriority).toBeCalledWith(error);
+  });
+
+  it('should not download without hass', async () => {
+    const api = createCardAPI();
+    vi.mocked(api.getViewManager().getView).mockReturnValue(createViewWithMedia());
+    vi.mocked(api.getHASSManager().getHASS).mockReturnValue(null);
+    const manager = new DownloadManager(api);
+
+    expect(await manager.downloadViewerMedia()).toBeFalsy();
+    expect(downloadMedia).not.toBeCalled();
+  });
+});
+
+describe('DownloadManager.downloadScreenshot', () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it('with url', async () => {
+    const api = createCardAPI();
+    const player = mock<FrigateCardMediaPlayer>();
+    player.getScreenshotURL.mockResolvedValue('http://screenshot');
+
+    vi.mocked(api.getMediaLoadedInfoManager().get).mockReturnValue(
+      createMediaLoadedInfo({
+        player: player,
+      }),
+    );
+    const manager = new DownloadManager(api);
+    await manager.downloadScreenshot();
+
+    expect(downloadURL).toBeCalledWith('http://screenshot', 'screenshot.jpg');
+  });
+
+  it('without url', async () => {
+    const api = createCardAPI();
+    const player = mock<FrigateCardMediaPlayer>();
+    player.getScreenshotURL.mockResolvedValue(null);
+
+    vi.mocked(api.getMediaLoadedInfoManager().get).mockReturnValue(
+      createMediaLoadedInfo({
+        player: player,
+      }),
+    );
+    const manager = new DownloadManager(api);
+    await manager.downloadScreenshot();
+
+    expect(downloadURL).not.toBeCalled();
+  });
+});