Index: cc/test/scheduler_test_common.cc |
diff --git a/cc/test/scheduler_test_common.cc b/cc/test/scheduler_test_common.cc |
index 7001b59ddeadd00e0c4563aa4acdb5f4fceed335..5d81f64306ee617111e24c0e4a45238a40a022ce 100644 |
--- a/cc/test/scheduler_test_common.cc |
+++ b/cc/test/scheduler_test_common.cc |
@@ -34,6 +34,20 @@ std::string TestDelayBasedTimeSource::TypeString() const { |
TestDelayBasedTimeSource::~TestDelayBasedTimeSource() { |
} |
+bool FakeBeginFrameSource::NeedsBeginFrames() { |
+ return needs_begin_frames_; |
+} |
+void FakeBeginFrameSource::SetNeedsBeginFrames(bool needs_begin_frames) { |
+ needs_begin_frames_ = needs_begin_frames; |
+} |
+void FakeBeginFrameSource::FinishedFrame(size_t remaining_frames) { |
+ remaining_frames_ = remaining_frames; |
+} |
+void FakeBeginFrameSource::AsValueInto(base::debug::TracedValue* dict) const { |
+ dict->SetString("type", "FakeBeginFrameSource"); |
+ BeginFrameSource::AsValueInto(dict); |
+} |
+ |
TestScheduler::TestScheduler( |
scoped_refptr<TestNowSource> now_src, |
SchedulerClient* client, |
@@ -46,12 +60,20 @@ TestScheduler::TestScheduler( |
test_task_runner), |
now_src_(now_src), |
test_task_runner_(test_task_runner.get()) { |
+} |
+ |
+void TestScheduler::SetUpBackgroundFrameSource() { |
+ background_frame_source_store_ = TestSyntheticBeginFrameSource::Create( |
+ now_src_, test_task_runner_, base::TimeDelta::FromSeconds(1)); |
+} |
+ |
+void TestScheduler::SetUpPrimaryFrameSource() { |
if (!settings_.begin_frame_scheduling_enabled) { |
- scoped_refptr<DelayBasedTimeSource> time_source = |
- TestDelayBasedTimeSource::Create( |
- now_src, VSyncInterval(), test_task_runner_); |
- synthetic_begin_frame_source_.reset( |
- new SyntheticBeginFrameSource(this, time_source)); |
+ primary_frame_source_store_ = TestSyntheticBeginFrameSource::Create( |
+ now_src_, test_task_runner_, BeginFrameArgs::DefaultInterval()); |
+ primary_frame_source_ = primary_frame_source_store_.get(); |
+ } else { |
+ Scheduler::SetUpPrimaryFrameSource(); |
} |
} |