Fix chunk count bug in FakeAdaptiveDataSet.

-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=162331762

testutils/src/main/java/com/google/android/exoplayer2/testutil/FakeAdaptiveDataSet.java

@@ -42,6 +42,7 @@ public final class FakeAdaptiveDataSet extends FakeDataSet {
 
 
   }
   }
 
 
+  private final long chunkCount;
   private final long chunkDurationUs;
   private final long chunkDurationUs;
   private final long lastChunkDurationUs;
   private final long lastChunkDurationUs;
 
 
@@ -66,6 +67,11 @@ public final class FakeAdaptiveDataSet extends FakeDataSet {
       }
       }
     }
     }
     this.lastChunkDurationUs = lastChunkDurationUs == 0 ? chunkDurationUs : lastChunkDurationUs;
     this.lastChunkDurationUs = lastChunkDurationUs == 0 ? chunkDurationUs : lastChunkDurationUs;
+    this.chunkCount = lastChunkDurationUs == 0 ? fullChunks : fullChunks + 1;
+  }
+
+  public long getChunkCount() {
+    return chunkCount;
   }
   }
 
 
   public String getUri(int trackSelectionIndex) {
   public String getUri(int trackSelectionIndex) {
@@ -73,7 +79,7 @@ public final class FakeAdaptiveDataSet extends FakeDataSet {
   }
   }
 
 
   public long getChunkDuration(int chunkIndex) {
   public long getChunkDuration(int chunkIndex) {
-    return chunkIndex == getAllData().size() - 1 ? lastChunkDurationUs : chunkDurationUs;
+    return chunkIndex == getChunkCount() - 1 ? lastChunkDurationUs : chunkDurationUs;
   }
   }
 
 
   public long getStartTime(int chunkIndex) {
   public long getStartTime(int chunkIndex) {