summaryrefslogtreecommitdiffstats
path: root/cc/test/scheduler_test_common.h
blob: 521def9b1a4220ec9db532e6af4c9afe0b682c77 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
// Copyright 2011 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 CC_TEST_SCHEDULER_TEST_COMMON_H_
#define CC_TEST_SCHEDULER_TEST_COMMON_H_

#include "base/memory/scoped_ptr.h"
#include "base/time.h"
#include "cc/base/thread.h"
#include "cc/scheduler/delay_based_time_source.h"
#include "cc/scheduler/frame_rate_controller.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace cc {

class FakeTimeSourceClient : public cc::TimeSourceClient {
 public:
  FakeTimeSourceClient() { Reset(); }
  void Reset() { tick_called_ = false; }
  bool TickCalled() const { return tick_called_; }

  // TimeSourceClient implementation.
  virtual void OnTimerTick() OVERRIDE;

 protected:
  bool tick_called_;
};

class FakeThread : public cc::Thread {
 public:
  FakeThread();
  virtual ~FakeThread();

  void Reset() {
    pending_task_delay_ = 0;
    pending_task_.reset();
    run_pending_task_on_overwrite_ = false;
  }

  void RunPendingTaskOnOverwrite(bool enable) {
    run_pending_task_on_overwrite_ = enable;
  }

  bool HasPendingTask() const { return pending_task_; }
  void RunPendingTask();

  long long PendingDelayMs() const {
    EXPECT_TRUE(HasPendingTask());
    return pending_task_delay_;
  }

  virtual void PostTask(base::Closure cb) OVERRIDE;
  virtual void PostDelayedTask(base::Closure cb, base::TimeDelta delay)
      OVERRIDE;
  virtual bool BelongsToCurrentThread() const OVERRIDE;

 protected:
  scoped_ptr<base::Closure> pending_task_;
  long long pending_task_delay_;
  bool run_pending_task_on_overwrite_;
};

class FakeTimeSource : public cc::TimeSource {
 public:
  FakeTimeSource() : active_(false), client_(0) {}

  virtual void SetClient(cc::TimeSourceClient* client) OVERRIDE;
  virtual void SetActive(bool b) OVERRIDE;
  virtual bool Active() const OVERRIDE;
  virtual void SetTimebaseAndInterval(base::TimeTicks timebase,
                                      base::TimeDelta interval) OVERRIDE {}
  virtual base::TimeTicks LastTickTime() OVERRIDE;
  virtual base::TimeTicks NextTickTime() OVERRIDE;

  void Tick() {
    ASSERT_TRUE(active_);
    if (client_)
      client_->OnTimerTick();
  }

  void SetNextTickTime(base::TimeTicks next_tick_time) {
    next_tick_time_ = next_tick_time;
  }

 protected:
  virtual ~FakeTimeSource() {}

  bool active_;
  base::TimeTicks next_tick_time_;
  cc::TimeSourceClient* client_;
};

class FakeDelayBasedTimeSource : public cc::DelayBasedTimeSource {
 public:
  static scoped_refptr<FakeDelayBasedTimeSource> Create(
      base::TimeDelta interval, cc::Thread* thread) {
    return make_scoped_refptr(new FakeDelayBasedTimeSource(interval, thread));
  }

  void SetNow(base::TimeTicks time) { now_ = time; }
  virtual base::TimeTicks Now() const OVERRIDE;

 protected:
  FakeDelayBasedTimeSource(base::TimeDelta interval, cc::Thread* thread)
      : DelayBasedTimeSource(interval, thread) {}
  virtual ~FakeDelayBasedTimeSource() {}

  base::TimeTicks now_;
};

class FakeFrameRateController : public cc::FrameRateController {
 public:
  FakeFrameRateController(scoped_refptr<cc::TimeSource> timer)
      : cc::FrameRateController(timer) {}

  int NumFramesPending() const { return num_frames_pending_; }
};

}  // namespace cc

#endif  // CC_TEST_SCHEDULER_TEST_COMMON_H_