51 lines
1.6 KiB
C++
51 lines
1.6 KiB
C++
// 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 LIBBRILLO_BRILLO_HTTP_MOCK_CONNECTION_H_
|
|
#define LIBBRILLO_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 // LIBBRILLO_BRILLO_HTTP_MOCK_CONNECTION_H_
|