components/scheduler/renderer/task_cost_estimator_unittest.cc

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/test/simple_test_tick_clock.h"
   6 #include "components/scheduler/renderer/task_cost_estimator.h"
   7
   8 #include "testing/gmock/include/gmock/gmock.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 namespace scheduler {
  12
  13 class TaskCostEstimatorTest : public testing::Test {
  14  public:
  15   TaskCostEstimatorTest() {}
  16   ~TaskCostEstimatorTest() override {}
  17
  18   void SetUp() override {}
  19
  20   base::SimpleTestTickClock clock_;
  21 };
  22
  23 class TaskCostEstimatorForTest : public TaskCostEstimator {
  24  public:
  25   TaskCostEstimatorForTest(base::SimpleTestTickClock* clock,
  26                            int sample_count,
  27                            double estimation_percentile)
  28       : TaskCostEstimator(sample_count, estimation_percentile), clock_(clock) {}
  29
  30  protected:
  31   base::TimeTicks Now() override { return clock_->NowTicks(); }
  32
  33  private:
  34   base::SimpleTestTickClock* clock_;
  35 };
  36
  37 TEST_F(TaskCostEstimatorTest, BasicEstimation) {
  38   TaskCostEstimatorForTest estimator(&clock_, 1, 100);
  39   base::PendingTask task(FROM_HERE, base::Closure());
  40
  41   estimator.WillProcessTask(task);
  42   clock_.Advance(base::TimeDelta::FromMilliseconds(500));
  43   estimator.DidProcessTask(task);
  44
  45   EXPECT_EQ(base::TimeDelta::FromMilliseconds(500),
  46             estimator.expected_task_duration());
  47 };
  48
  49 TEST_F(TaskCostEstimatorTest, NestedRunLoop) {
  50   TaskCostEstimatorForTest estimator(&clock_, 1, 100);
  51   base::PendingTask task(FROM_HERE, base::Closure());
  52
  53   // Make sure we ignore the tasks inside the nested run loop.
  54   estimator.WillProcessTask(task);
  55   estimator.WillProcessTask(task);
  56   clock_.Advance(base::TimeDelta::FromMilliseconds(500));
  57   estimator.DidProcessTask(task);
  58   clock_.Advance(base::TimeDelta::FromMilliseconds(500));
  59   estimator.DidProcessTask(task);
  60
  61   EXPECT_EQ(base::TimeDelta::FromMilliseconds(1000),
  62             estimator.expected_task_duration());
  63 };
  64
  65 }  // namespace scheduler