small update on ScheduleController unit test : add some missing mocks

Dash-Industry-Forum · Aug 22, 2017 · 4205e7c · 4205e7c
1 parent c1c38c7
commit 4205e7c
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/test/unit/mocks/StreamProcessorMock.js b/test/unit/mocks/StreamProcessorMock.js
@@ -90,7 +90,7 @@ class StreamProcessorMock {
     }
 
     getCurrentRepresentationInfo() {
-        return {};
+        return {mediaInfo: {type : this.type}};
     }
 
     isBufferingCompleted() {

diff --git a/test/unit/streaming.controllers.ScheduleController.js b/test/unit/streaming.controllers.ScheduleController.js
@@ -6,6 +6,10 @@ import PlaybackControllerMock from './mocks/PlaybackControllerMock';
 import StreamProcessorMock from './mocks/StreamProcessorMock';
 import MediaPlayerModelMock from './mocks/MediaPlayerModelMock';
 import DashManifestModelMock from './mocks/DashManifestModelMock';
+import AbrControllerMock from './mocks/AbrControllerMock';
+import StreamControllerMock from './mocks/StreamControllerMock';
+import DashMetricsMock from './mocks/DashMetricsMock';
+import MetricsModelMock from './mocks/MetricsModelMock';
 
 const expect = require('chai').expect;
 const context = {};
@@ -24,19 +28,31 @@ describe('ScheduleController', function () {
     let streamProcessorMock;
     let dashManifestModelMock;
     let playbackControllerMock;
+    let abrControllerMock;
+    let streamControllerMock;
+    let dashMetricsMock;
+    let metricsModelMock;
 
     beforeEach(function () {
         mediaPlayerModelMock = new MediaPlayerModelMock();
         streamProcessorMock = new StreamProcessorMock(testType, streamInfo);
         dashManifestModelMock = new DashManifestModelMock();
         playbackControllerMock = new PlaybackControllerMock();
+        abrControllerMock = new AbrControllerMock();
+        streamControllerMock = new StreamControllerMock();
+        dashMetricsMock = new DashMetricsMock();
+        metricsModelMock = new MetricsModelMock();
 
         scheduleController = ScheduleController(context).create({
             type: testType,
             mediaPlayerModel: mediaPlayerModelMock,
             streamProcessor: streamProcessorMock,
             dashManifestModel: dashManifestModelMock,
-            playbackController: playbackControllerMock
+            playbackController: playbackControllerMock,
+            abrController: abrControllerMock,
+            streamController: streamControllerMock,
+            dashMetrics: dashMetricsMock,
+            metricsModel: metricsModelMock
         });
 
         scheduleController.initialize();