cc/test/scheduler_test_common.h

   1 // Copyright 2011 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 CC_TEST_SCHEDULER_TEST_COMMON_H_
   6 #define CC_TEST_SCHEDULER_TEST_COMMON_H_
   7
   8 #include "base/memory/scoped_ptr.h"
   9 #include "cc/delay_based_time_source.h"
  10 #include "cc/frame_rate_controller.h"
  11 #include "cc/thread.h"
  12 #include "testing/gtest/include/gtest/gtest.h"
  13
  14 namespace cc {
  15
  16 class FakeTimeSourceClient : public cc::TimeSourceClient {
  17 public:
  18     FakeTimeSourceClient() { reset(); }
  19     void reset() { m_tickCalled = false; }
  20     bool tickCalled() const { return m_tickCalled; }
  21
  22     virtual void onTimerTick() OVERRIDE;
  23
  24 protected:
  25     bool m_tickCalled;
  26 };
  27
  28 class FakeThread : public cc::Thread {
  29 public:
  30     FakeThread();
  31     virtual ~FakeThread();
  32
  33     void reset()
  34     {
  35         m_pendingTaskDelay = 0;
  36         m_pendingTask.reset();
  37         m_runPendingTaskOnOverwrite = false;
  38     }
  39
  40     void runPendingTaskOnOverwrite(bool enable)
  41     {
  42         m_runPendingTaskOnOverwrite = enable;
  43     }
  44
  45     bool hasPendingTask() const { return m_pendingTask; }
  46     void runPendingTask();
  47
  48     long long pendingDelayMs() const
  49     {
  50         EXPECT_TRUE(hasPendingTask());
  51         return m_pendingTaskDelay;
  52     }
  53
  54     virtual void postTask(base::Closure cb) OVERRIDE;
  55     virtual void postDelayedTask(base::Closure cb, long long delay) OVERRIDE;
  56     virtual bool belongsToCurrentThread() const OVERRIDE;
  57
  58 protected:
  59     scoped_ptr<base::Closure> m_pendingTask;
  60     long long m_pendingTaskDelay;
  61     bool m_runPendingTaskOnOverwrite;
  62 };
  63
  64 class FakeTimeSource : public cc::TimeSource {
  65 public:
  66     FakeTimeSource()
  67         : m_active(false)
  68         , m_client(0)
  69     {
  70     }
  71
  72     virtual void setClient(cc::TimeSourceClient* client) OVERRIDE;
  73     virtual void setActive(bool b) OVERRIDE;
  74     virtual bool active() const OVERRIDE;
  75     virtual void setTimebaseAndInterval(base::TimeTicks timebase, base::TimeDelta interval) OVERRIDE { }
  76     virtual base::TimeTicks lastTickTime() OVERRIDE;
  77     virtual base::TimeTicks nextTickTime() OVERRIDE;
  78
  79     void tick()
  80     {
  81         ASSERT_TRUE(m_active);
  82         if (m_client)
  83             m_client->onTimerTick();
  84     }
  85
  86     void setNextTickTime(base::TimeTicks nextTickTime) { m_nextTickTime = nextTickTime; }
  87
  88 protected:
  89     virtual ~FakeTimeSource() { }
  90
  91     bool m_active;
  92     base::TimeTicks m_nextTickTime;
  93     cc::TimeSourceClient* m_client;
  94 };
  95
  96 class FakeDelayBasedTimeSource : public cc::DelayBasedTimeSource {
  97 public:
  98     static scoped_refptr<FakeDelayBasedTimeSource> create(base::TimeDelta interval, cc::Thread* thread)
  99     {
 100         return make_scoped_refptr(new FakeDelayBasedTimeSource(interval, thread));
 101     }
 102
 103     void setNow(base::TimeTicks time) { m_now = time; }
 104     virtual base::TimeTicks now() const OVERRIDE;
 105
 106 protected:
 107     FakeDelayBasedTimeSource(base::TimeDelta interval, cc::Thread* thread)
 108         : DelayBasedTimeSource(interval, thread)
 109     {
 110     }
 111     virtual ~FakeDelayBasedTimeSource() { }
 112
 113     base::TimeTicks m_now;
 114 };
 115
 116 class FakeFrameRateController : public cc::FrameRateController {
 117 public:
 118     FakeFrameRateController(scoped_refptr<cc::TimeSource> timer) : cc::FrameRateController(timer) { }
 119
 120     int numFramesPending() const { return m_numFramesPending; }
 121 };
 122
 123 }  // namespace cc
 124
 125 #endif  // CC_TEST_SCHEDULER_TEST_COMMON_H_