OLD | NEW |
| (Empty) |
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 "components/scheduler/renderer/task_cost_estimator.h" | |
6 | |
7 #include <memory> | |
8 | |
9 #include "base/test/simple_test_tick_clock.h" | |
10 #include "components/scheduler/base/test_time_source.h" | |
11 #include "testing/gmock/include/gmock/gmock.h" | |
12 #include "testing/gtest/include/gtest/gtest.h" | |
13 | |
14 namespace scheduler { | |
15 | |
16 class TaskCostEstimatorTest : public testing::Test { | |
17 public: | |
18 TaskCostEstimatorTest() {} | |
19 ~TaskCostEstimatorTest() override {} | |
20 | |
21 void SetUp() override { | |
22 test_time_source_.reset(new TestTimeSource(&clock_)); | |
23 } | |
24 | |
25 base::SimpleTestTickClock clock_; | |
26 std::unique_ptr<TestTimeSource> test_time_source_; | |
27 }; | |
28 | |
29 class TaskCostEstimatorForTest : public TaskCostEstimator { | |
30 public: | |
31 TaskCostEstimatorForTest(TestTimeSource* test_time_source, | |
32 int sample_count, | |
33 double estimation_percentile) | |
34 : TaskCostEstimator(test_time_source, | |
35 sample_count, | |
36 estimation_percentile) {} | |
37 }; | |
38 | |
39 TEST_F(TaskCostEstimatorTest, BasicEstimation) { | |
40 TaskCostEstimatorForTest estimator(test_time_source_.get(), 1, 100); | |
41 base::PendingTask task(FROM_HERE, base::Closure()); | |
42 | |
43 estimator.WillProcessTask(task); | |
44 clock_.Advance(base::TimeDelta::FromMilliseconds(500)); | |
45 estimator.DidProcessTask(task); | |
46 | |
47 EXPECT_EQ(base::TimeDelta::FromMilliseconds(500), | |
48 estimator.expected_task_duration()); | |
49 } | |
50 | |
51 TEST_F(TaskCostEstimatorTest, Clear) { | |
52 TaskCostEstimatorForTest estimator(test_time_source_.get(), 1, 100); | |
53 base::PendingTask task(FROM_HERE, base::Closure()); | |
54 | |
55 estimator.WillProcessTask(task); | |
56 clock_.Advance(base::TimeDelta::FromMilliseconds(500)); | |
57 estimator.DidProcessTask(task); | |
58 | |
59 estimator.Clear(); | |
60 | |
61 EXPECT_EQ(base::TimeDelta(), estimator.expected_task_duration()); | |
62 } | |
63 | |
64 TEST_F(TaskCostEstimatorTest, NestedRunLoop) { | |
65 TaskCostEstimatorForTest estimator(test_time_source_.get(), 1, 100); | |
66 base::PendingTask task(FROM_HERE, base::Closure()); | |
67 | |
68 // Make sure we ignore the tasks inside the nested run loop. | |
69 estimator.WillProcessTask(task); | |
70 estimator.WillProcessTask(task); | |
71 clock_.Advance(base::TimeDelta::FromMilliseconds(500)); | |
72 estimator.DidProcessTask(task); | |
73 clock_.Advance(base::TimeDelta::FromMilliseconds(500)); | |
74 estimator.DidProcessTask(task); | |
75 | |
76 EXPECT_EQ(base::TimeDelta::FromMilliseconds(1000), | |
77 estimator.expected_task_duration()); | |
78 } | |
79 | |
80 } // namespace scheduler | |
OLD | NEW |