From 4205e7ca20e59eaabc2c09ed77cf396fdfb83056 Mon Sep 17 00:00:00 2001
From: Nicolas ANGOT <nicolas.angot@orange.com>
Date: Tue, 22 Aug 2017 17:59:53 +0200
Subject: [PATCH] small update on ScheduleController unit test : add some
 missing mocks

---
 test/unit/mocks/StreamProcessorMock.js         |  2 +-
 ...streaming.controllers.ScheduleController.js | 18 +++++++++++++++++-
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/test/unit/mocks/StreamProcessorMock.js b/test/unit/mocks/StreamProcessorMock.js
index 7eb1114a90..1877a0d42e 100644
--- 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
index 3cab25b386..8694acd355 100644
--- 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();