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
|
// Copyright 2014 The Chromium OS 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 LIBCHROMEOS_BRILLO_HTTP_MOCK_CONNECTION_H_
#define LIBCHROMEOS_BRILLO_HTTP_MOCK_CONNECTION_H_
#include <memory>
#include <string>
#include <base/macros.h>
#include <brillo/http/http_connection.h>
#include <gmock/gmock.h>
namespace brillo {
namespace http {
class MockConnection : public Connection {
public:
using Connection::Connection;
MOCK_METHOD2(SendHeaders, bool(const HeaderList&, ErrorPtr*));
MOCK_METHOD2(MockSetRequestData, bool(Stream*, ErrorPtr*));
MOCK_METHOD1(MockSetResponseData, void(Stream*));
MOCK_METHOD1(FinishRequest, bool(ErrorPtr*));
MOCK_METHOD2(FinishRequestAsync,
RequestID(const SuccessCallback&, const ErrorCallback&));
MOCK_CONST_METHOD0(GetResponseStatusCode, int());
MOCK_CONST_METHOD0(GetResponseStatusText, std::string());
MOCK_CONST_METHOD0(GetProtocolVersion, std::string());
MOCK_CONST_METHOD1(GetResponseHeader, std::string(const std::string&));
MOCK_CONST_METHOD1(MockExtractDataStream, Stream*(brillo::ErrorPtr*));
private:
bool SetRequestData(StreamPtr stream, brillo::ErrorPtr* error) override {
return MockSetRequestData(stream.get(), error);
}
void SetResponseData(StreamPtr stream) override {
MockSetResponseData(stream.get());
}
StreamPtr ExtractDataStream(brillo::ErrorPtr* error) override {
return StreamPtr{MockExtractDataStream(error)};
}
DISALLOW_COPY_AND_ASSIGN(MockConnection);
};
} // namespace http
} // namespace brillo
#endif // LIBCHROMEOS_BRILLO_HTTP_MOCK_CONNECTION_H_
|
