| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <list> | 5 #include <list> |
| 6 #include <vector> | 6 #include <vector> |
| 7 | 7 |
| 8 #include "base/basictypes.h" | 8 #include "base/basictypes.h" |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/memory/ref_counted.h" | 10 #include "base/memory/ref_counted.h" |
| 11 #include "base/memory/scoped_ptr.h" | 11 #include "base/memory/scoped_ptr.h" |
| 12 #include "base/thread_task_runner_handle.h" |
| 12 #include "base/threading/thread.h" | 13 #include "base/threading/thread.h" |
| 13 #include "base/time/time.h" | 14 #include "base/time/time.h" |
| 14 #include "chromecast/media/cma/base/balanced_media_task_runner_factory.h" | 15 #include "chromecast/media/cma/base/balanced_media_task_runner_factory.h" |
| 15 #include "chromecast/media/cma/base/media_task_runner.h" | 16 #include "chromecast/media/cma/base/media_task_runner.h" |
| 16 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
| 17 | 18 |
| 18 namespace chromecast { | 19 namespace chromecast { |
| 19 namespace media { | 20 namespace media { |
| 20 | 21 |
| 21 namespace { | 22 namespace { |
| (...skipping 71 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 93 | 94 |
| 94 scheduling_pattern_ = pattern; | 95 scheduling_pattern_ = pattern; |
| 95 pattern_index_ = 0; | 96 pattern_index_ = 0; |
| 96 | 97 |
| 97 // Setup each task runner. | 98 // Setup each task runner. |
| 98 size_t n = timestamps_in_ms.size(); | 99 size_t n = timestamps_in_ms.size(); |
| 99 contexts_.resize(n); | 100 contexts_.resize(n); |
| 100 for (size_t k = 0; k < n; k++) { | 101 for (size_t k = 0; k < n; k++) { |
| 101 contexts_[k].media_task_runner = | 102 contexts_[k].media_task_runner = |
| 102 media_task_runner_factory_->CreateMediaTaskRunner( | 103 media_task_runner_factory_->CreateMediaTaskRunner( |
| 103 base::MessageLoopProxy::current()); | 104 base::ThreadTaskRunnerHandle::Get()); |
| 104 contexts_[k].is_pending_task = false; | 105 contexts_[k].is_pending_task = false; |
| 105 contexts_[k].task_index = 0; | 106 contexts_[k].task_index = 0; |
| 106 contexts_[k].task_timestamp_list.resize( | 107 contexts_[k].task_timestamp_list.resize( |
| 107 timestamps_in_ms[k].size()); | 108 timestamps_in_ms[k].size()); |
| 108 for (size_t i = 0; i < timestamps_in_ms[k].size(); i++) { | 109 for (size_t i = 0; i < timestamps_in_ms[k].size(); i++) { |
| 109 contexts_[k].task_timestamp_list[i] = | 110 contexts_[k].task_timestamp_list[i] = |
| 110 base::TimeDelta::FromMilliseconds(timestamps_in_ms[k][i]); | 111 base::TimeDelta::FromMilliseconds(timestamps_in_ms[k][i]); |
| 111 } | 112 } |
| 112 } | 113 } |
| 113 | 114 |
| (...skipping 28 matching lines...) Expand all Loading... |
| 142 (pattern_index_ + 1) % scheduling_pattern_.size(); | 143 (pattern_index_ + 1) % scheduling_pattern_.size(); |
| 143 | 144 |
| 144 size_t task_runner_id = scheduling_pattern_[pattern_index_]; | 145 size_t task_runner_id = scheduling_pattern_[pattern_index_]; |
| 145 MediaTaskRunnerTestContext& context = contexts_[task_runner_id]; | 146 MediaTaskRunnerTestContext& context = contexts_[task_runner_id]; |
| 146 | 147 |
| 147 // Check whether all tasks have been scheduled for that task runner | 148 // Check whether all tasks have been scheduled for that task runner |
| 148 // or if there is already one pending task. | 149 // or if there is already one pending task. |
| 149 if (context.task_index >= context.task_timestamp_list.size() || | 150 if (context.task_index >= context.task_timestamp_list.size() || |
| 150 context.is_pending_task) { | 151 context.is_pending_task) { |
| 151 pattern_index_ = next_pattern_index; | 152 pattern_index_ = next_pattern_index; |
| 152 base::MessageLoopProxy::current()->PostTask( | 153 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 153 FROM_HERE, | 154 FROM_HERE, base::Bind(&BalancedMediaTaskRunnerTest::ScheduleTask, |
| 154 base::Bind(&BalancedMediaTaskRunnerTest::ScheduleTask, | 155 base::Unretained(this))); |
| 155 base::Unretained(this))); | |
| 156 return; | 156 return; |
| 157 } | 157 } |
| 158 | 158 |
| 159 bool expected_may_run = false; | 159 bool expected_may_run = false; |
| 160 if (context.task_timestamp_list[context.task_index] >= | 160 if (context.task_timestamp_list[context.task_index] >= |
| 161 context.max_timestamp) { | 161 context.max_timestamp) { |
| 162 expected_may_run = true; | 162 expected_may_run = true; |
| 163 context.max_timestamp = context.task_timestamp_list[context.task_index]; | 163 context.max_timestamp = context.task_timestamp_list[context.task_index]; |
| 164 } | 164 } |
| 165 | 165 |
| 166 bool may_run = context.media_task_runner->PostMediaTask( | 166 bool may_run = context.media_task_runner->PostMediaTask( |
| 167 FROM_HERE, | 167 FROM_HERE, |
| 168 base::Bind(&BalancedMediaTaskRunnerTest::Task, | 168 base::Bind(&BalancedMediaTaskRunnerTest::Task, |
| 169 base::Unretained(this), | 169 base::Unretained(this), |
| 170 task_runner_id, | 170 task_runner_id, |
| 171 context.task_timestamp_list[context.task_index]), | 171 context.task_timestamp_list[context.task_index]), |
| 172 context.task_timestamp_list[context.task_index]); | 172 context.task_timestamp_list[context.task_index]); |
| 173 EXPECT_EQ(may_run, expected_may_run); | 173 EXPECT_EQ(may_run, expected_may_run); |
| 174 | 174 |
| 175 if (may_run) | 175 if (may_run) |
| 176 context.is_pending_task = true; | 176 context.is_pending_task = true; |
| 177 | 177 |
| 178 context.task_index++; | 178 context.task_index++; |
| 179 pattern_index_ = next_pattern_index; | 179 pattern_index_ = next_pattern_index; |
| 180 base::MessageLoopProxy::current()->PostTask( | 180 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 181 FROM_HERE, | 181 FROM_HERE, base::Bind(&BalancedMediaTaskRunnerTest::ScheduleTask, |
| 182 base::Bind(&BalancedMediaTaskRunnerTest::ScheduleTask, | 182 base::Unretained(this))); |
| 183 base::Unretained(this))); | |
| 184 } | 183 } |
| 185 | 184 |
| 186 void BalancedMediaTaskRunnerTest::Task( | 185 void BalancedMediaTaskRunnerTest::Task( |
| 187 size_t task_runner_id, base::TimeDelta timestamp) { | 186 size_t task_runner_id, base::TimeDelta timestamp) { |
| 188 ASSERT_FALSE(expected_task_timestamps_.empty()); | 187 ASSERT_FALSE(expected_task_timestamps_.empty()); |
| 189 EXPECT_EQ(timestamp, expected_task_timestamps_.front()); | 188 EXPECT_EQ(timestamp, expected_task_timestamps_.front()); |
| 190 expected_task_timestamps_.pop_front(); | 189 expected_task_timestamps_.pop_front(); |
| 191 | 190 |
| 192 contexts_[task_runner_id].is_pending_task = false; | 191 contexts_[task_runner_id].is_pending_task = false; |
| 193 } | 192 } |
| (...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 254 timestamps_ms, | 253 timestamps_ms, |
| 255 scheduling_pattern, | 254 scheduling_pattern, |
| 256 expected_timestamps_ms); | 255 expected_timestamps_ms); |
| 257 ProcessAllTasks(); | 256 ProcessAllTasks(); |
| 258 message_loop->Run(); | 257 message_loop->Run(); |
| 259 EXPECT_TRUE(expected_task_timestamps_.empty()); | 258 EXPECT_TRUE(expected_task_timestamps_.empty()); |
| 260 } | 259 } |
| 261 | 260 |
| 262 } // namespace media | 261 } // namespace media |
| 263 } // namespace chromecast | 262 } // namespace chromecast |
| OLD | NEW |