50 lines
1.4 KiB
C++
50 lines
1.4 KiB
C++
// Copyright 2014 Google Inc. All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file or at
|
|
// https://developers.google.com/open-source/licenses/bsd
|
|
|
|
#include "packager/mpd/base/bandwidth_estimator.h"
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
namespace shaka {
|
|
|
|
namespace {
|
|
const uint64_t kBitsInByte = 8;
|
|
} // namespace
|
|
|
|
TEST(BandwidthEstimatorTest, AllBlocks) {
|
|
const double kDuration = 1.0;
|
|
BandwidthEstimator be(kDuration);
|
|
const uint64_t kNumBlocksToAdd = 100;
|
|
uint64_t total_bytes = 0;
|
|
for (uint64_t i = 1; i <= kNumBlocksToAdd; ++i) {
|
|
be.AddBlock(i, kDuration);
|
|
total_bytes += i;
|
|
}
|
|
|
|
const uint64_t kExptectedEstimate =
|
|
total_bytes * kBitsInByte / kNumBlocksToAdd;
|
|
EXPECT_EQ(kExptectedEstimate, be.Estimate());
|
|
const uint64_t kMax = kNumBlocksToAdd * kBitsInByte;
|
|
EXPECT_EQ(kMax, be.Max());
|
|
}
|
|
|
|
TEST(BandwidthEstimatorTest, ExcludeShortSegments) {
|
|
const double kDuration = 1.0;
|
|
BandwidthEstimator be(kDuration);
|
|
|
|
// Add 4 blocks with duration 0.1, 0.8, 1.8 and 0.2 respectively. The first
|
|
// and the last blocks are excluded as they are too short.
|
|
be.AddBlock(1, 0.1 * kDuration);
|
|
be.AddBlock(1, 0.8 * kDuration);
|
|
be.AddBlock(1, 1.8 * kDuration);
|
|
be.AddBlock(1, 0.2 * kDuration);
|
|
|
|
const uint64_t kExpectedMax = 1 / 0.8 * kBitsInByte;
|
|
EXPECT_EQ(kExpectedMax, be.Max());
|
|
}
|
|
|
|
} // namespace shaka
|