Index: blimp/net/test_common.h |
diff --git a/blimp/net/test_common.h b/blimp/net/test_common.h |
deleted file mode 100644 |
index 8a88cb806005ff54f593dbe4561c27ce83bd13ee..0000000000000000000000000000000000000000 |
--- a/blimp/net/test_common.h |
+++ /dev/null |
@@ -1,230 +0,0 @@ |
-// Copyright 2015 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 BLIMP_NET_TEST_COMMON_H_ |
-#define BLIMP_NET_TEST_COMMON_H_ |
- |
-#include <stddef.h> |
-#include <stdint.h> |
- |
-#include <memory> |
-#include <string> |
- |
-#include "blimp/common/proto/blimp_message.pb.h" |
-#include "blimp/net/blimp_connection.h" |
-#include "blimp/net/blimp_message_processor.h" |
-#include "blimp/net/blimp_transport.h" |
-#include "blimp/net/connection_error_observer.h" |
-#include "blimp/net/connection_handler.h" |
-#include "blimp/net/message_port.h" |
-#include "blimp/net/packet_reader.h" |
-#include "blimp/net/packet_writer.h" |
-#include "net/log/net_log_with_source.h" |
-#include "net/socket/stream_socket.h" |
-#include "testing/gmock/include/gmock/gmock.h" |
- |
-namespace net { |
-class GrowableIOBuffer; |
-} // namespace net |
- |
-namespace blimp { |
- |
-// Checks if the contents of a buffer are an exact match with std::string. |
-// Using this matcher for inequality checks will result in undefined behavior, |
-// due to IOBuffer's lack of a size field. |
-// |
-// arg (type: IOBuffer*) The buffer to check. |
-// data (type: std::string) The string to compare with |arg|. |
-MATCHER_P(BufferEquals, expected, "") { |
- return expected == std::string(arg->data(), expected.size()); |
-} |
- |
-// Checks if two proto messages are the same. |
-MATCHER_P(EqualsProto, message, "") { |
- std::string expected_serialized; |
- std::string actual_serialized; |
- message.SerializeToString(&expected_serialized); |
- arg.SerializeToString(&actual_serialized); |
- return expected_serialized == actual_serialized; |
-} |
- |
-// Checks if the contents of a buffer are an exact match with BlimpMessage. |
-// arg (type: net::DrainableIOBuffer*) The buffer to check. |
-// message (type: BlimpMessage) The message to compare with |arg|. |
-MATCHER_P(BufferEqualsProto, message, "") { |
- BlimpMessage actual_message; |
- actual_message.ParseFromArray(arg->data(), message.ByteSize()); |
- std::string expected_serialized; |
- std::string actual_serialized; |
- message.SerializeToString(&expected_serialized); |
- actual_message.SerializeToString(&actual_serialized); |
- return expected_serialized == actual_serialized; |
-} |
- |
-// Checks if the contents of a BlobDataPtr match the string |expected|. |
-MATCHER_P(BlobDataPtrEqualsString, expected, "") { |
- return expected == arg->data; |
-} |
- |
-// GMock action that writes data from a string to an IOBuffer. |
-// |
-// buf_idx (template parameter 0): 0-based index of the IOBuffer arg. |
-// str: the string containing data to be written to the IOBuffer. |
-ACTION_TEMPLATE(FillBufferFromString, |
- HAS_1_TEMPLATE_PARAMS(int, buf_idx), |
- AND_1_VALUE_PARAMS(str)) { |
- memcpy(testing::get<buf_idx>(args)->data(), str.data(), str.size()); |
-} |
- |
-// Returns true if |buf| is prefixed by |str|. |
-bool BufferStartsWith(net::GrowableIOBuffer* buf, |
- size_t buf_size, |
- const std::string& str); |
- |
-// GMock action that writes data from a BlimpMessage to a GrowableIOBuffer. |
-// Advances the buffer's |offset| to the end of the message. |
-// |
-// buf_idx (template parameter 0): 0-based index of the IOBuffer arg. |
-// message: the blimp message containing data to be written to the IOBuffer |
-ACTION_TEMPLATE(FillBufferFromMessage, |
- HAS_1_TEMPLATE_PARAMS(int, buf_idx), |
- AND_1_VALUE_PARAMS(message)) { |
- message->SerializeToArray(testing::get<buf_idx>(args)->data(), |
- message->ByteSize()); |
-} |
- |
-// Calls |set_offset()| for a GrowableIOBuffer. |
-ACTION_TEMPLATE(SetBufferOffset, |
- HAS_1_TEMPLATE_PARAMS(int, buf_idx), |
- AND_1_VALUE_PARAMS(offset)) { |
- testing::get<buf_idx>(args)->set_offset(offset); |
-} |
- |
-// Formats a string-based representation of a BlimpMessage header. |
-std::string EncodeHeader(size_t size); |
- |
-class MockStreamSocket : public net::StreamSocket { |
- public: |
- MockStreamSocket(); |
- virtual ~MockStreamSocket(); |
- |
- MOCK_METHOD3(Read, int(net::IOBuffer*, int, const net::CompletionCallback&)); |
- MOCK_METHOD3(Write, int(net::IOBuffer*, int, const net::CompletionCallback&)); |
- MOCK_METHOD1(SetReceiveBufferSize, int(int32_t)); |
- MOCK_METHOD1(SetSendBufferSize, int(int32_t)); |
- MOCK_METHOD1(Connect, int(const net::CompletionCallback&)); |
- MOCK_METHOD0(Disconnect, void()); |
- MOCK_CONST_METHOD0(IsConnected, bool()); |
- MOCK_CONST_METHOD0(IsConnectedAndIdle, bool()); |
- MOCK_CONST_METHOD1(GetPeerAddress, int(net::IPEndPoint*)); |
- MOCK_CONST_METHOD1(GetLocalAddress, int(net::IPEndPoint*)); |
- MOCK_CONST_METHOD0(NetLog, const net::NetLogWithSource&()); |
- MOCK_METHOD0(SetSubresourceSpeculation, void()); |
- MOCK_METHOD0(SetOmniboxSpeculation, void()); |
- MOCK_CONST_METHOD0(WasEverUsed, bool()); |
- MOCK_CONST_METHOD0(UsingTCPFastOpen, bool()); |
- MOCK_CONST_METHOD0(NumBytesRead, int64_t()); |
- MOCK_CONST_METHOD0(GetConnectTimeMicros, base::TimeDelta()); |
- MOCK_CONST_METHOD0(WasAlpnNegotiated, bool()); |
- MOCK_CONST_METHOD0(GetNegotiatedProtocol, net::NextProto()); |
- MOCK_METHOD1(GetSSLInfo, bool(net::SSLInfo*)); |
- MOCK_CONST_METHOD1(GetConnectionAttempts, void(net::ConnectionAttempts*)); |
- MOCK_METHOD0(ClearConnectionAttempts, void()); |
- MOCK_METHOD1(AddConnectionAttempts, void(const net::ConnectionAttempts&)); |
- MOCK_CONST_METHOD0(GetTotalReceivedBytes, int64_t()); |
-}; |
- |
-class MockBlimpConnection; |
- |
-class MockTransport : public BlimpTransport { |
- public: |
- MockTransport(); |
- ~MockTransport() override; |
- |
- MOCK_METHOD1(Connect, void(const net::CompletionCallback& callback)); |
- MOCK_METHOD0(TakeMessagePortPtr, MessagePort*()); |
- std::unique_ptr<BlimpConnection> MakeConnection() override; |
- |
- const char* GetName() const override; |
- |
- void SetMockConnection(std::unique_ptr<MockBlimpConnection> connection); |
- |
- private: |
- std::unique_ptr<MockBlimpConnection> connection_; |
-}; |
- |
-class MockConnectionHandler : public ConnectionHandler { |
- public: |
- MockConnectionHandler(); |
- ~MockConnectionHandler() override; |
- |
- MOCK_METHOD1(HandleConnectionPtr, void(BlimpConnection* connection)); |
- void HandleConnection(std::unique_ptr<BlimpConnection> connection) override; |
-}; |
- |
-class MockPacketReader : public PacketReader { |
- public: |
- MockPacketReader(); |
- ~MockPacketReader() override; |
- |
- MOCK_METHOD2(ReadPacket, |
- void(const scoped_refptr<net::GrowableIOBuffer>&, |
- const net::CompletionCallback&)); |
-}; |
- |
-class MockPacketWriter : public PacketWriter { |
- public: |
- MockPacketWriter(); |
- ~MockPacketWriter() override; |
- |
- MOCK_METHOD2(WritePacket, |
- void(const scoped_refptr<net::DrainableIOBuffer>&, |
- const net::CompletionCallback&)); |
-}; |
- |
-class MockBlimpConnection : public BlimpConnection { |
- public: |
- MockBlimpConnection(); |
- ~MockBlimpConnection() override; |
- |
- MOCK_METHOD1(SetConnectionErrorObserver, |
- void(ConnectionErrorObserver* observer)); |
- |
- MOCK_METHOD1(SetIncomingMessageProcessor, |
- void(BlimpMessageProcessor* processor)); |
- |
- MOCK_METHOD1(AddConnectionErrorObserver, void(ConnectionErrorObserver*)); |
- |
- MOCK_METHOD1(RemoveConnectionErrorObserver, void(ConnectionErrorObserver*)); |
- |
- MOCK_METHOD0(GetOutgoingMessageProcessor, BlimpMessageProcessor*(void)); |
-}; |
- |
-class MockConnectionErrorObserver : public ConnectionErrorObserver { |
- public: |
- MockConnectionErrorObserver(); |
- ~MockConnectionErrorObserver() override; |
- |
- MOCK_METHOD1(OnConnectionError, void(int error)); |
-}; |
- |
-class MockBlimpMessageProcessor : public BlimpMessageProcessor { |
- public: |
- MockBlimpMessageProcessor(); |
- |
- ~MockBlimpMessageProcessor() override; |
- |
- // Adapts calls from ProcessMessage to MockableProcessMessage by |
- // unboxing the |message| std::unique_ptr for GMock compatibility. |
- void ProcessMessage(std::unique_ptr<BlimpMessage> message, |
- const net::CompletionCallback& callback) override; |
- |
- MOCK_METHOD2(MockableProcessMessage, |
- void(const BlimpMessage& message, |
- const net::CompletionCallback& callback)); |
-}; |
- |
-} // namespace blimp |
- |
-#endif // BLIMP_NET_TEST_COMMON_H_ |