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
|
// Copyright 2014 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 EXTENSIONS_BROWSER_API_CAST_CHANNEL_CAST_TEST_UTIL_H_
#define EXTENSIONS_BROWSER_API_CAST_CHANNEL_CAST_TEST_UTIL_H_
#include <string>
#include <utility>
#include "base/macros.h"
#include "base/message_loop/message_loop.h"
#include "extensions/browser/api/cast_channel/cast_socket.h"
#include "extensions/browser/api/cast_channel/cast_transport.h"
#include "extensions/common/api/cast_channel/cast_channel.pb.h"
#include "net/base/ip_endpoint.h"
#include "testing/gmock/include/gmock/gmock.h"
namespace extensions {
namespace api {
namespace cast_channel {
extern const char kTestExtensionId[];
class MockCastTransport : public extensions::api::cast_channel::CastTransport {
public:
MockCastTransport();
~MockCastTransport() override;
void SetReadDelegate(scoped_ptr<CastTransport::Delegate> delegate) override;
MOCK_METHOD2(SendMessage,
void(const extensions::api::cast_channel::CastMessage& message,
const net::CompletionCallback& callback));
MOCK_METHOD0(Start, void(void));
// Gets the read delegate that is currently active for this transport.
CastTransport::Delegate* current_delegate() const;
private:
scoped_ptr<CastTransport::Delegate> delegate_;
DISALLOW_COPY_AND_ASSIGN(MockCastTransport);
};
class MockCastTransportDelegate : public CastTransport::Delegate {
public:
MockCastTransportDelegate();
~MockCastTransportDelegate() override;
MOCK_METHOD1(OnError, void(ChannelError error));
MOCK_METHOD1(OnMessage, void(const CastMessage& message));
MOCK_METHOD0(Start, void(void));
private:
DISALLOW_COPY_AND_ASSIGN(MockCastTransportDelegate);
};
class MockCastSocket : public CastSocket {
public:
MockCastSocket();
~MockCastSocket() override;
// Mockable version of Connect. Accepts a bare pointer to a mock object.
// (GMock won't compile with scoped_ptr method parameters.)
MOCK_METHOD2(ConnectRawPtr,
void(CastTransport::Delegate* delegate,
base::Callback<void(ChannelError)> callback));
// Proxy for ConnectRawPtr. Unpacks scoped_ptr into a GMock-friendly bare
// ptr.
void Connect(scoped_ptr<CastTransport::Delegate> delegate,
base::Callback<void(ChannelError)> callback) override {
delegate_ = std::move(delegate);
ConnectRawPtr(delegate_.get(), callback);
}
MOCK_METHOD1(Close, void(const net::CompletionCallback& callback));
MOCK_CONST_METHOD0(ip_endpoint, const net::IPEndPoint&());
MOCK_CONST_METHOD0(id, int());
MOCK_METHOD1(set_id, void(int id));
MOCK_CONST_METHOD0(channel_auth, ChannelAuthType());
MOCK_CONST_METHOD0(ready_state, ReadyState());
MOCK_CONST_METHOD0(error_state, ChannelError());
MOCK_CONST_METHOD0(keep_alive, bool(void));
MOCK_CONST_METHOD0(audio_only, bool(void));
MOCK_METHOD1(SetErrorState, void(ChannelError error_state));
CastTransport* transport() const override { return mock_transport_.get(); }
MockCastTransport* mock_transport() const { return mock_transport_.get(); }
private:
scoped_ptr<MockCastTransport> mock_transport_;
scoped_ptr<CastTransport::Delegate> delegate_;
DISALLOW_COPY_AND_ASSIGN(MockCastSocket);
};
// Creates the IPEndpoint 192.168.1.1.
net::IPEndPoint CreateIPEndPointForTest();
// Checks if two proto messages are the same.
// From
// third_party/cacheinvalidation/overrides/google/cacheinvalidation/deps/gmock.h
// TODO(kmarshall): promote to a shared testing library.
MATCHER_P(EqualsProto, message, "") {
std::string expected_serialized, actual_serialized;
message.SerializeToString(&expected_serialized);
arg.SerializeToString(&actual_serialized);
return expected_serialized == actual_serialized;
}
ACTION_TEMPLATE(PostCompletionCallbackTask,
HAS_1_TEMPLATE_PARAMS(int, cb_idx),
AND_1_VALUE_PARAMS(rv)) {
base::MessageLoop::current()->PostTask(
FROM_HERE, base::Bind(testing::get<cb_idx>(args), rv));
}
} // namespace cast_channel
} // namespace api
} // namespace extensions
#endif // EXTENSIONS_BROWSER_API_CAST_CHANNEL_CAST_TEST_UTIL_H_
|
