blob: 6aa650b95604a2bc0e0362c6e9d5709f22b6b817 [file] [log] [blame]
// Copyright 2017 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef THIRD_PARTY_BLINK_RENDERER_PLATFORM_TESTING_TESTING_PLATFORM_SUPPORT_WITH_MOCK_SCHEDULER_H_
#define THIRD_PARTY_BLINK_RENDERER_PLATFORM_TESTING_TESTING_PLATFORM_SUPPORT_WITH_MOCK_SCHEDULER_H_
#include <memory>
#include "base/macros.h"
#include "base/test/test_mock_time_task_runner.h"
#include "third_party/blink/renderer/platform/scheduler/public/thread.h"
#include "third_party/blink/renderer/platform/testing/scoped_main_thread_overrider.h"
#include "third_party/blink/renderer/platform/testing/testing_platform_support.h"
namespace base {
class TestMockTimeTaskRunner;
namespace sequence_manager {
class SequenceManager;
} // namespace sequence_manager
} // namespace base
namespace blink {
namespace scheduler {
class MainThreadSchedulerImpl;
}
// This class adds scheduler and threading support to TestingPlatformSupport.
// See also ScopedTestingPlatformSupport to use this class correctly.
class TestingPlatformSupportWithMockScheduler : public TestingPlatformSupport {
public:
TestingPlatformSupportWithMockScheduler();
~TestingPlatformSupportWithMockScheduler() override;
scoped_refptr<base::TestMockTimeTaskRunner> test_task_runner() {
return test_task_runner_;
}
// Runs a single task.
void RunSingleTask();
// Runs all currently queued immediate tasks and delayed tasks whose delay has
// expired plus any immediate tasks that are posted as a result of running
// those tasks.
//
// This function ignores future delayed tasks when deciding if the system is
// idle. If you need to ensure delayed tasks run, try runForPeriodSeconds()
// instead.
void RunUntilIdle() override;
// Runs for |seconds| the testing clock is advanced by |seconds|. Note real
// time elapsed will typically much less than |seconds| because delays between
// timers are fast forwarded.
void RunForPeriodSeconds(double seconds);
void RunForPeriod(base::TimeDelta period);
// Advances |clock_| by |seconds|.
void AdvanceClockSeconds(double seconds);
void AdvanceClock(base::TimeDelta duration);
scheduler::MainThreadSchedulerImpl* GetMainThreadScheduler() const;
// Controls the behavior of |m_mockTaskRunner| if true, then |m_clock| will
// be advanced to the next timer when there's no more immediate work to do.
void SetAutoAdvanceNowToPendingTasks(bool);
// Returns the current mock time.
base::TimeTicks NowTicks() const;
protected:
scoped_refptr<base::TestMockTimeTaskRunner> test_task_runner_;
bool auto_advance_ = true;
std::unique_ptr<scheduler::MainThreadSchedulerImpl> scheduler_;
base::sequence_manager::SequenceManager*
sequence_manager_; // Owned by scheduler_.
std::unique_ptr<ScopedMainThreadOverrider> main_thread_overrider_;
private:
DISALLOW_COPY_AND_ASSIGN(TestingPlatformSupportWithMockScheduler);
};
} // namespace blink
#endif // THIRD_PARTY_BLINK_RENDERER_PLATFORM_TESTING_TESTING_PLATFORM_SUPPORT_WITH_MOCK_SCHEDULER_H_