src/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   virtual bool PostDelayedTask(const tracked_objects::Location& from_here,
  32                                const base::Closure& task,
  33                                base::TimeDelta delay) OVERRIDE;
  34
  35   virtual bool RunsTasksOnCurrentThread() const OVERRIDE;
  36
  37   // This function is currently not used, and will return false.
  38   virtual bool PostNonNestableDelayedTask(
  39       const tracked_objects::Location& from_here,
  40       const base::Closure& task,
  41       base::TimeDelta delay) OVERRIDE;
  42
  43  protected:
  44   virtual ~FakeSingleThreadTaskRunner();
  45
  46  private:
  47   base::SimpleTestTickClock* const clock_;
  48   std::multimap<base::TimeTicks, PostedTask> tasks_;
  49   bool fail_on_next_task_;
  50
  51   DISALLOW_COPY_AND_ASSIGN(FakeSingleThreadTaskRunner);
  52 };
  53
  54 }  // namespace test
  55 }  // namespace cast
  56 }  // namespace media
  57
  58 #endif  // MEDIA_CAST_TEST_FAKE_TASK_RUNNER_H_