blob: c4243065deaad798834ca042608ba7f2c3794e30 (
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
|
// Copyright (c) 2009 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 WEBKIT_GLUE_DEVTOOLS_DEVTOOLS_MOCK_RPC_H_
#define WEBKIT_GLUE_DEVTOOLS_DEVTOOLS_MOCK_RPC_H_
#include <string>
#include "base/string_util.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "webkit/glue/devtools/devtools_rpc.h"
#include "webkit/glue/glue_util.h"
// Universal mock delegate for DevToolsRpc. Typical usage of the mock is:
// mock->Method1(); // Set expectation.
// mock->Replay();
// // Do Something here;
// mock->Verify(); // Verify.
class DevToolsMockRpc : public DevToolsRpc::Delegate {
public:
DevToolsMockRpc() {}
~DevToolsMockRpc() {}
virtual void SendRpcMessage(const std::string& msg) {
if (!log_.length()) {
log_ = msg;
} else {
log_ = StringPrintf("%s\n%s", log_.c_str(), msg.c_str());
}
}
void Replay() {
ref_log_ = log_;
log_ = "";
}
void Verify() {
ASSERT_EQ(ref_log_, log_);
}
void Reset() {
ref_log_ = "";
log_ = "";
}
const std::string &get_log() { return log_; }
private:
std::string log_;
std::string ref_log_;
DISALLOW_COPY_AND_ASSIGN(DevToolsMockRpc);
};
#endif // WEBKIT_GLUE_DEVTOOLS_DEVTOOLS_MOCK_RPC_H_
|