Add better control over substreams.

2023-05-06 10:42:17 -07:00
parent fd74c1c855
commit 867a118a16
9 changed files with 405 additions and 26 deletions
@@ -0,0 +1,144 @@
+import { describe, expect, it, vi } from 'vitest';
+import { mock } from 'vitest-mock-extended';
+import { CameraManager } from '../../src/camera-manager/manager';
+import { getAllDependentCameras } from '../../src/utils/camera';
+import {
+  createViewWithNextStream,
+  createViewWithSelectedSubstream,
+  createViewWithoutSubstream,
+  hasSubstream,
+} from '../../src/utils/substream';
+import { View } from '../../src/view/view';
+
+vi.mock('../../src/utils/camera');
+
+describe('createViewWithSelectedSubstream', () => {
+  it('should create view with selected substream', () => {
+    const view = new View({ view: 'live', camera: 'camera' });
+    const newView = createViewWithSelectedSubstream(view, 'substream');
+    expect(newView?.context?.live?.overrides).toEqual(
+      new Map([['camera', 'substream']]),
+    );
+  });
+
+  it('should create view with selected substream with existing overrides', () => {
+    const view = new View({
+      view: 'live',
+      camera: 'camera',
+      context: {
+        live: {
+          overrides: new Map([['camera', 'camera']]),
+        },
+      },
+    });
+    const newView = createViewWithSelectedSubstream(view, 'substream');
+    expect(newView?.context?.live?.overrides).toEqual(
+      new Map([['camera', 'substream']]),
+    );
+  });
+});
+
+describe('createViewWithoutSubstream', () => {
+  it('should create view without substream', () => {
+    const view = new View({
+      view: 'live',
+      camera: 'camera',
+      context: {
+        live: {
+          overrides: new Map([['camera', 'camera']]),
+        },
+      },
+    });
+    const newView = createViewWithoutSubstream(view);
+    expect(newView?.context?.live?.overrides).toEqual(new Map());
+  });
+});
+
+describe('hasSubstream', () => {
+  it('should detect substream', () => {
+    const view = new View({
+      view: 'live',
+      camera: 'camera',
+      context: {
+        live: {
+          overrides: new Map([['camera', 'camera2']]),
+        },
+      },
+    });
+    expect(hasSubstream(view)).toBeTruthy();
+  });
+  it('should not detect substream when absent', () => {
+    const view = new View({
+      view: 'live',
+      camera: 'camera',
+    });
+    expect(hasSubstream(view)).toBeFalsy();
+  });
+  it('should not detect substream when main stream', () => {
+    const view = new View({
+      view: 'live',
+      camera: 'camera',
+      context: {
+        live: {
+          overrides: new Map([['camera', 'camera']]),
+        },
+      },
+    });
+    expect(hasSubstream(view)).toBeFalsy();
+  });
+});
+
+describe('createViewWithNextStream', () => {
+  it('should create new equal view with no dependencies', () => {
+    const view = new View({
+      view: 'live',
+      camera: 'camera',
+    });
+    vi.mocked(getAllDependentCameras).mockReturnValue(new Set(['camera']));
+    const cameraManager = mock<CameraManager>();
+    const newView = createViewWithNextStream(cameraManager, view);
+    expect(newView.camera).toBe(view.camera);
+    expect(newView.view).toBe(view.view);
+    expect(newView.context).toEqual(view.context);
+  });
+  it('should create new view with next stream', () => {
+    const view = new View({
+      view: 'live',
+      camera: 'camera',
+    });
+    vi.mocked(getAllDependentCameras).mockReturnValue(new Set(['camera', 'camera2']));
+    const cameraManager = mock<CameraManager>();
+    const newView = createViewWithNextStream(cameraManager, view);
+    expect(newView.context?.live?.overrides).toEqual(new Map([['camera', 'camera2']]));
+  });
+  it('should create new view with next stream that cycles back', () => {
+    const view = new View({
+      view: 'live',
+      camera: 'camera',
+      context: {
+        live: {
+          overrides: new Map([['camera', 'camera2']]),
+        },
+      },
+    });
+    vi.mocked(getAllDependentCameras).mockReturnValue(new Set(['camera', 'camera2']));
+    const cameraManager = mock<CameraManager>();
+    const newView = createViewWithNextStream(cameraManager, view);
+    expect(newView.context?.live?.overrides).toEqual(new Map([['camera', 'camera']]));
+  });
+  it('should create new view with first stream with invalid substream', () => {
+    const view = new View({
+      view: 'live',
+      camera: 'camera',
+      context: {
+        live: {
+          overrides: new Map([['camera', 'camera-that-does-not-exist']]),
+        },
+      },
+    });
+    vi.mocked(getAllDependentCameras).mockReturnValue(new Set(['camera', 'camera2']));
+    const cameraManager = mock<CameraManager>();
+    const newView = createViewWithNextStream(cameraManager, view);
+    expect(newView.context?.live?.overrides).toEqual(new Map([['camera', 'camera']]));
+  });
+});