| Index: blimp/net/blimp_connection_unittest.cc
|
| diff --git a/blimp/net/blimp_connection_unittest.cc b/blimp/net/blimp_connection_unittest.cc
|
| index d8a0ec7576795e7b4dd7a0a7ee9ec6a99089854d..681ea2b77be4ab9030c39962aeb3343bcd665bcf 100644
|
| --- a/blimp/net/blimp_connection_unittest.cc
|
| +++ b/blimp/net/blimp_connection_unittest.cc
|
| @@ -7,6 +7,7 @@
|
| #include <stddef.h>
|
|
|
| #include <string>
|
| +#include <utility>
|
|
|
| #include "base/callback_helpers.h"
|
| #include "base/memory/ptr_util.h"
|
| @@ -15,6 +16,7 @@
|
| #include "blimp/common/proto/blimp_message.pb.h"
|
| #include "blimp/net/common.h"
|
| #include "blimp/net/connection_error_observer.h"
|
| +#include "blimp/net/message_port.h"
|
| #include "blimp/net/test_common.h"
|
| #include "net/base/completion_callback.h"
|
| #include "net/base/io_buffer.h"
|
| @@ -34,14 +36,14 @@ namespace {
|
| class BlimpConnectionTest : public testing::Test {
|
| public:
|
| BlimpConnectionTest() {
|
| - std::unique_ptr<testing::StrictMock<MockPacketWriter>> writer(
|
| - new testing::StrictMock<MockPacketWriter>);
|
| - writer_ = writer.get();
|
| - std::unique_ptr<testing::StrictMock<MockPacketReader>> reader(
|
| - new testing::StrictMock<MockPacketReader>);
|
| - reader_ = reader.get();
|
| - connection_.reset(
|
| - new BlimpConnection(std::move(reader), std::move(writer)));
|
| + std::unique_ptr<MockPacketReader> mock_reader(new MockPacketReader);
|
| + std::unique_ptr<MockPacketWriter> mock_writer(new MockPacketWriter);
|
| + mock_reader_ = mock_reader.get();
|
| + mock_writer_ = mock_writer.get();
|
| + connection_ =
|
| + base::MakeUnique<BlimpConnection>(base::MakeUnique<MessagePort>(
|
| + std::move(mock_reader), std::move(mock_writer)));
|
| +
|
| connection_->AddConnectionErrorObserver(&error_observer1_);
|
| connection_->AddConnectionErrorObserver(&error_observer2_);
|
| connection_->AddConnectionErrorObserver(&error_observer3_);
|
| @@ -64,8 +66,8 @@ class BlimpConnectionTest : public testing::Test {
|
| }
|
|
|
| base::MessageLoop message_loop_;
|
| - testing::StrictMock<MockPacketReader>* reader_;
|
| - testing::StrictMock<MockPacketWriter>* writer_;
|
| + MockPacketReader* mock_reader_;
|
| + MockPacketWriter* mock_writer_;
|
| testing::StrictMock<MockConnectionErrorObserver> error_observer1_;
|
| testing::StrictMock<MockConnectionErrorObserver> error_observer2_;
|
|
|
| @@ -82,11 +84,11 @@ TEST_F(BlimpConnectionTest, AsyncTwoPacketsWrite) {
|
| net::CompletionCallback write_packet_cb;
|
|
|
| InSequence s;
|
| - EXPECT_CALL(*writer_,
|
| + EXPECT_CALL(*mock_writer_,
|
| WritePacket(BufferEqualsProto(*CreateInputMessage()), _))
|
| .WillOnce(SaveArg<1>(&write_packet_cb))
|
| .RetiresOnSaturation();
|
| - EXPECT_CALL(*writer_,
|
| + EXPECT_CALL(*mock_writer_,
|
| WritePacket(BufferEqualsProto(*CreateControlMessage()), _))
|
| .WillOnce(SaveArg<1>(&write_packet_cb))
|
| .RetiresOnSaturation();
|
| @@ -118,11 +120,11 @@ TEST_F(BlimpConnectionTest, AsyncTwoPacketsWriteWithError) {
|
| net::CompletionCallback write_packet_cb;
|
|
|
| InSequence s;
|
| - EXPECT_CALL(*writer_,
|
| + EXPECT_CALL(*mock_writer_,
|
| WritePacket(BufferEqualsProto(*CreateInputMessage()), _))
|
| .WillOnce(SaveArg<1>(&write_packet_cb))
|
| .RetiresOnSaturation();
|
| - EXPECT_CALL(*writer_,
|
| + EXPECT_CALL(*mock_writer_,
|
| WritePacket(BufferEqualsProto(*CreateControlMessage()), _))
|
| .WillOnce(SaveArg<1>(&write_packet_cb))
|
| .RetiresOnSaturation();
|
| @@ -148,7 +150,7 @@ TEST_F(BlimpConnectionTest, DeleteHappyObserversAreOK) {
|
| net::CompletionCallback write_packet_cb;
|
|
|
| InSequence s;
|
| - EXPECT_CALL(*writer_,
|
| + EXPECT_CALL(*mock_writer_,
|
| WritePacket(BufferEqualsProto(*CreateInputMessage()), _))
|
| .WillOnce(SaveArg<1>(&write_packet_cb))
|
| .RetiresOnSaturation();
|
| @@ -168,7 +170,7 @@ TEST_F(BlimpConnectionTest, ReadPacketErrorInvokesErrorObservers) {
|
| scoped_refptr<net::GrowableIOBuffer> read_packet_buffer;
|
| net::CompletionCallback read_packet_cb;
|
|
|
| - EXPECT_CALL(*reader_, ReadPacket(_, _))
|
| + EXPECT_CALL(*mock_reader_, ReadPacket(_, _))
|
| .WillOnce(
|
| DoAll(SaveArg<0>(&read_packet_buffer), SaveArg<1>(&read_packet_cb)))
|
| .RetiresOnSaturation();
|
| @@ -194,7 +196,7 @@ TEST_F(BlimpConnectionTest, EndConnectionInvokesErrorObservers) {
|
| scoped_refptr<net::GrowableIOBuffer> read_packet_buffer;
|
| net::CompletionCallback read_packet_cb;
|
|
|
| - EXPECT_CALL(*reader_, ReadPacket(_, _))
|
| + EXPECT_CALL(*mock_reader_, ReadPacket(_, _))
|
| .WillOnce(
|
| DoAll(SaveArg<0>(&read_packet_buffer), SaveArg<1>(&read_packet_cb)))
|
| .WillOnce(Return())
|
|
|