Add unit tests for getAverageThroughput

Dash-Industry-Forum · Jun 28, 2017 · 6a3973c · 6a3973c
1 parent bb9dea9
commit 6a3973c
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/test/unit/mocks/AbrControllerMock.js b/test/unit/mocks/AbrControllerMock.js
@@ -26,6 +26,7 @@ class AbrControllerMock{
         this.limitBitrateByPortal = false;
         this.usePixelRatioInLimitBitrateByPortal = false;
         this.autoSwitchBitrate = {video: true, audio: true};
+        this.throughputHistory = undefined;
     }
 
     initialize() {}
@@ -164,6 +165,10 @@ class AbrControllerMock{
 
     }
 
+    getThroughputHistory() {
+        return this.throughputHistory;
+    }
+
     updateTopQualityIndex() {}
 
     isPlayingAtTopQuality() {}

diff --git a/test/unit/streaming.MediaPlayerSpec.js b/test/unit/streaming.MediaPlayerSpec.js
@@ -327,7 +327,7 @@ describe("MediaPlayer", function () {
 
     });
 
-    describe("AutoBitrate Functions", function () {
+    describe("AbrController Functions", function () {
         afterEach(function () {
             abrControllerMock.reset();
         })
@@ -457,6 +457,22 @@ describe("MediaPlayer", function () {
             expect(AutoSwitchBitrateFor).to.be.false;
         });
 
+        it("Method getAverageThroughput should return 0 when throughputHistory is not set up", function() {
+            const averageThroughput = player.getAverageThroughput('video');
+            expect(averageThroughput).to.equal(0);
+        });
+
+        it("Method getAverageThroughput should value computed from ThroughputHistory", function() {
+            const AVERAGE_THROUGHPUT = 2000;
+            abrControllerMock.throughputHistory = {
+                getAverageThroughput: function() {
+                    return AVERAGE_THROUGHPUT;
+                }
+            };
+            const averageThroughput = player.getAverageThroughput('video');
+            expect(averageThroughput).to.equal(AVERAGE_THROUGHPUT);
+        });
+
         describe("When it is not initialized", function () {
             it("Method getQualityFor should throw an exception", function () {
                 expect(player.getQualityFor).to.throw(PLAYBACK_NOT_INITIALIZED_ERROR);