media/cast/test/fake_single_thread_task_runner.h

   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
   3 // found in the LICENSE file.
   4
   5 #ifndef MEDIA_CAST_TEST_FAKE_TASK_RUNNER_H_
   6 #define MEDIA_CAST_TEST_FAKE_TASK_RUNNER_H_
   7
   8 #include <map>
   9
  10 #include "base/basictypes.h"
  11 #include "base/single_thread_task_runner.h"
  12 #include "base/test/simple_test_tick_clock.h"
  13 #include "base/test/test_pending_task.h"
  14
  15 namespace media {
  16 namespace cast {
  17 namespace test {
  18
  19 typedef base::TestPendingTask PostedTask;
  20
  21 class FakeSingleThreadTaskRunner : public base::SingleThreadTaskRunner {
  22  public:
  23   explicit FakeSingleThreadTaskRunner(base::SimpleTestTickClock* clock);
  24
  25   void RunTasks();
  26
  27   // Note: Advances |clock_|.
  28   void Sleep(base::TimeDelta t);
  29
  30   // base::SingleThreadTaskRunner implementation.
  31   bool PostDelayedTask(const tracked_objects::Location& from_here,
  32                        const base::Closure& task,
  33                        base::TimeDelta delay) final;
  34
  35   bool RunsTasksOnCurrentThread() const final;
  36
  37   // This function is currently not used, and will return false.
  38   bool PostNonNestableDelayedTask(const tracked_objects::Location& from_here,
  39                                   const base::Closure& task,
  40                                   base::TimeDelta delay) final;
  41
  42  protected:
  43   ~FakeSingleThreadTaskRunner() final;
  44
  45  private:
  46   base::SimpleTestTickClock* const clock_;
  47   std::multimap<base::TimeTicks, PostedTask> tasks_;
  48   bool fail_on_next_task_;
  49
  50   DISALLOW_COPY_AND_ASSIGN(FakeSingleThreadTaskRunner);
  51 };
  52
  53 }  // namespace test
  54 }  // namespace cast
  55 }  // namespace media
  56
  57 #endif  // MEDIA_CAST_TEST_FAKE_TASK_RUNNER_H_