You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
1.8 KiB
57 lines
1.8 KiB
// 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_METHOD(bool, SendHeaders, (const HeaderList&, ErrorPtr*), (override));
|
|
MOCK_METHOD(bool, MockSetRequestData, (Stream*, ErrorPtr*));
|
|
MOCK_METHOD(void, MockSetResponseData, (Stream*));
|
|
MOCK_METHOD(bool, FinishRequest, (ErrorPtr*), (override));
|
|
MOCK_METHOD(RequestID,
|
|
FinishRequestAsync,
|
|
(const SuccessCallback&, const ErrorCallback&),
|
|
(override));
|
|
MOCK_METHOD(int, GetResponseStatusCode, (), (const, override));
|
|
MOCK_METHOD(std::string, GetResponseStatusText, (), (const, override));
|
|
MOCK_METHOD(std::string, GetProtocolVersion, (), (const, override));
|
|
MOCK_METHOD(std::string,
|
|
GetResponseHeader,
|
|
(const std::string&),
|
|
(const, override));
|
|
MOCK_METHOD(Stream*, MockExtractDataStream, (brillo::ErrorPtr*), (const));
|
|
|
|
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_
|