1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124
|
// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "cc/test/task_graph_runner_test_template.h"
namespace cc {
TaskGraphRunnerTestBase::TaskGraphRunnerTestBase() = default;
TaskGraphRunnerTestBase::~TaskGraphRunnerTestBase() = default;
void TaskGraphRunnerTestBase::SetTaskGraphRunner(
TaskGraphRunner* task_graph_runner) {
task_graph_runner_ = task_graph_runner;
}
void TaskGraphRunnerTestBase::ResetIds(int namespace_index) {
run_task_ids_[namespace_index].clear();
on_task_completed_ids_[namespace_index].clear();
}
void TaskGraphRunnerTestBase::RunAllTasks(int namespace_index) {
task_graph_runner_->WaitForTasksToFinishRunning(
namespace_token_[namespace_index]);
Task::Vector completed_tasks;
task_graph_runner_->CollectCompletedTasks(namespace_token_[namespace_index],
&completed_tasks);
for (Task::Vector::const_iterator it = completed_tasks.begin();
it != completed_tasks.end(); ++it) {
FakeTaskImpl* task = static_cast<FakeTaskImpl*>(it->get());
task->OnTaskCompleted();
}
}
void TaskGraphRunnerTestBase::RunUntilIdle() {
task_graph_runner_->RunTasksUntilIdleForTest();
for (const NamespaceToken& token : namespace_token_) {
Task::Vector completed_tasks;
task_graph_runner_->CollectCompletedTasks(token, &completed_tasks);
for (scoped_refptr<Task>& task : completed_tasks) {
static_cast<FakeTaskImpl*>(task.get())->OnTaskCompleted();
}
}
}
void TaskGraphRunnerTestBase::RunTaskOnWorkerThread(int namespace_index,
unsigned id) {
base::AutoLock lock(run_task_ids_lock_);
run_task_ids_[namespace_index].push_back(id);
}
void TaskGraphRunnerTestBase::OnTaskCompleted(int namespace_index,
unsigned id) {
on_task_completed_ids_[namespace_index].push_back(id);
}
const std::vector<unsigned>& TaskGraphRunnerTestBase::run_task_ids(
int namespace_index) {
return run_task_ids_[namespace_index];
}
const std::vector<unsigned>& TaskGraphRunnerTestBase::on_task_completed_ids(
int namespace_index) {
return on_task_completed_ids_[namespace_index];
}
void TaskGraphRunnerTestBase::ScheduleTasks(
int namespace_index,
const std::vector<TaskInfo>& tasks) {
Task::Vector new_tasks;
Task::Vector new_dependents;
TaskGraph new_graph;
for (auto it = tasks.begin(); it != tasks.end(); ++it) {
scoped_refptr<FakeTaskImpl> new_task(
new FakeTaskImpl(this, it->namespace_index, it->id));
new_graph.nodes.emplace_back(new_task.get(), it->category, it->priority,
it->has_external_dependency ? 1u : 0u,
it->has_external_dependency);
for (unsigned i = 0; i < it->dependent_count; ++i) {
scoped_refptr<FakeDependentTaskImpl> new_dependent_task(
new FakeDependentTaskImpl(this, it->namespace_index,
it->dependent_id));
new_graph.nodes.push_back(TaskGraph::Node(
new_dependent_task.get(), it->category, it->priority, 1u));
new_graph.edges.push_back(
TaskGraph::Edge(new_task.get(), new_dependent_task.get()));
new_dependents.push_back(new_dependent_task.get());
}
new_tasks.push_back(new_task.get());
}
task_graph_runner_->ScheduleTasks(namespace_token_[namespace_index],
&new_graph);
dependents_[namespace_index].swap(new_dependents);
tasks_[namespace_index].swap(new_tasks);
}
void TaskGraphRunnerTestBase::ExternalDependencyCompletedForTask(
int namespace_index,
int task_index) {
task_graph_runner_->ExternalDependencyCompletedForTask(
namespace_token_[namespace_index], tasks_[namespace_index][task_index]);
}
void TaskGraphRunnerTestBase::FakeTaskImpl::RunOnWorkerThread() {
test_->RunTaskOnWorkerThread(namespace_index_, id_);
}
void TaskGraphRunnerTestBase::FakeTaskImpl::OnTaskCompleted() {
test_->OnTaskCompleted(namespace_index_, id_);
}
// These suites are instantiated in binaries that use //cc:test_support.
GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(TaskGraphRunnerTest);
GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(SingleThreadTaskGraphRunnerTest);
} // namespace cc
|