summaryrefslogtreecommitdiffstats
path: root/cc/test/scheduler_test_common.h
blob: 3b73d0dcb2b8578e646fa808f2c3dd7ebde92e52 (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
123
124
125
126
127
128
129
130
131
// 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 CCSchedulerTestCommon_h
#define CCSchedulerTestCommon_h

#include "CCDelayBasedTimeSource.h"
#include "CCFrameRateController.h"
#include "CCThread.h"
#include "base/threading/platform_thread.h"
#include "testing/gtest/include/gtest/gtest.h"
#include <wtf/OwnPtr.h>

namespace WebKitTests {

class FakeCCTimeSourceClient : public cc::CCTimeSourceClient {
public:
    FakeCCTimeSourceClient() { reset(); }
    void reset() { m_tickCalled = false; }
    bool tickCalled() const { return m_tickCalled; }

    virtual void onTimerTick() OVERRIDE;

protected:
    bool m_tickCalled;
};

class FakeCCThread : public cc::CCThread {
public:
    FakeCCThread();
    virtual ~FakeCCThread();

    void reset()
    {
        m_pendingTaskDelay = 0;
        m_pendingTask.clear();
        m_runPendingTaskOnOverwrite = false;
    }

    void runPendingTaskOnOverwrite(bool enable)
    {
        m_runPendingTaskOnOverwrite = enable;
    }

    bool hasPendingTask() const { return m_pendingTask; }
    void runPendingTask()
    {
        ASSERT_TRUE(m_pendingTask);
        OwnPtr<Task> task = m_pendingTask.release();
        task->performTask();
    }

    long long pendingDelayMs() const
    {
        EXPECT_TRUE(hasPendingTask());
        return m_pendingTaskDelay;
    }

    virtual void postTask(PassOwnPtr<Task>) OVERRIDE;
    virtual void postDelayedTask(PassOwnPtr<Task> task, long long delay) OVERRIDE;
    virtual base::PlatformThreadId threadID() const OVERRIDE;

protected:
    OwnPtr<Task> m_pendingTask;
    long long m_pendingTaskDelay;
    bool m_runPendingTaskOnOverwrite;
};

class FakeCCTimeSource : public cc::CCTimeSource {
public:
    FakeCCTimeSource()
        : m_active(false)
        , m_client(0)
    {
        turnOffVerifier();
    }

    virtual ~FakeCCTimeSource() { }

    virtual void setClient(cc::CCTimeSourceClient* 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(m_active);
        if (m_client)
            m_client->onTimerTick();
    }

    void setNextTickTime(base::TimeTicks nextTickTime) { m_nextTickTime = nextTickTime; }

protected:
    bool m_active;
    base::TimeTicks m_nextTickTime;
    cc::CCTimeSourceClient* m_client;
};

class FakeCCDelayBasedTimeSource : public cc::CCDelayBasedTimeSource {
public:
    static PassRefPtr<FakeCCDelayBasedTimeSource> create(base::TimeDelta interval, cc::CCThread* thread)
    {
        return adoptRef(new FakeCCDelayBasedTimeSource(interval, thread));
    }

    void setNow(base::TimeTicks time) { m_now = time; }
    virtual base::TimeTicks now() const OVERRIDE;

protected:
    FakeCCDelayBasedTimeSource(base::TimeDelta interval, cc::CCThread* thread)
        : CCDelayBasedTimeSource(interval, thread)
    {
    }

    base::TimeTicks m_now;
};

class FakeCCFrameRateController : public cc::CCFrameRateController {
public:
    FakeCCFrameRateController(PassRefPtr<cc::CCTimeSource> timer) : cc::CCFrameRateController(timer) { }

    int numFramesPending() const { return m_numFramesPending; }
};

}

#endif // CCSchedulerTestCommon_h