Index: google_apis/gcm/base/socket_stream_unittest.cc |
diff --git a/google_apis/gcm/base/socket_stream_unittest.cc b/google_apis/gcm/base/socket_stream_unittest.cc |
deleted file mode 100644 |
index d7ba6793bf9c1010a5b4c7a19f36334b990430a9..0000000000000000000000000000000000000000 |
--- a/google_apis/gcm/base/socket_stream_unittest.cc |
+++ /dev/null |
@@ -1,406 +0,0 @@ |
-// Copyright 2013 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. |
- |
-#include "google_apis/gcm/base/socket_stream.h" |
- |
-#include "base/basictypes.h" |
-#include "base/bind.h" |
-#include "base/memory/scoped_ptr.h" |
-#include "base/run_loop.h" |
-#include "base/stl_util.h" |
-#include "base/strings/string_piece.h" |
-#include "net/socket/socket_test_util.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-namespace gcm { |
-namespace { |
- |
-typedef std::vector<net::MockRead> ReadList; |
-typedef std::vector<net::MockWrite> WriteList; |
- |
-const char kReadData[] = "read_data"; |
-const uint64 kReadDataSize = arraysize(kReadData) - 1; |
-const char kReadData2[] = "read_alternate_data"; |
-const uint64 kReadData2Size = arraysize(kReadData2) - 1; |
-const char kWriteData[] = "write_data"; |
-const uint64 kWriteDataSize = arraysize(kWriteData) - 1; |
- |
-class GCMSocketStreamTest : public testing::Test { |
- public: |
- GCMSocketStreamTest(); |
- virtual ~GCMSocketStreamTest(); |
- |
- // Build a socket with the expected reads and writes. |
- void BuildSocket(const ReadList& read_list, const WriteList& write_list); |
- |
- // Pump the message loop until idle. |
- void PumpLoop(); |
- |
- // Simulates a google::protobuf::io::CodedInputStream read. |
- base::StringPiece DoInputStreamRead(uint64 bytes); |
- // Simulates a google::protobuf::io::CodedOutputStream write. |
- uint64 DoOutputStreamWrite(const base::StringPiece& write_src); |
- |
- // Synchronous Refresh wrapper. |
- void WaitForData(size_t msg_size); |
- |
- base::MessageLoop* message_loop() { return &message_loop_; }; |
- net::DelayedSocketData* data_provider() { return data_provider_.get(); } |
- SocketInputStream* input_stream() { return socket_input_stream_.get(); } |
- SocketOutputStream* output_stream() { return socket_output_stream_.get(); } |
- net::StreamSocket* socket() { return socket_.get(); } |
- |
- private: |
- void OpenConnection(); |
- void ResetInputStream(); |
- void ResetOutputStream(); |
- |
- void ConnectCallback(int result); |
- |
- // SocketStreams and their data providers. |
- ReadList mock_reads_; |
- WriteList mock_writes_; |
- scoped_ptr<net::DelayedSocketData> data_provider_; |
- scoped_ptr<SocketInputStream> socket_input_stream_; |
- scoped_ptr<SocketOutputStream> socket_output_stream_; |
- |
- // net:: components. |
- scoped_ptr<net::StreamSocket> socket_; |
- net::MockClientSocketFactory socket_factory_; |
- net::AddressList address_list_; |
- |
- base::MessageLoopForIO message_loop_; |
-}; |
- |
-GCMSocketStreamTest::GCMSocketStreamTest() { |
- net::IPAddressNumber ip_number; |
- net::ParseIPLiteralToNumber("127.0.0.1", &ip_number); |
- address_list_ = net::AddressList::CreateFromIPAddress(ip_number, 5228); |
-} |
- |
-GCMSocketStreamTest::~GCMSocketStreamTest() {} |
- |
-void GCMSocketStreamTest::BuildSocket(const ReadList& read_list, |
- const WriteList& write_list) { |
- mock_reads_ = read_list; |
- mock_writes_ = write_list; |
- data_provider_.reset( |
- new net::DelayedSocketData( |
- 0, |
- vector_as_array(&mock_reads_), mock_reads_.size(), |
- vector_as_array(&mock_writes_), mock_writes_.size())); |
- socket_factory_.AddSocketDataProvider(data_provider_.get()); |
- OpenConnection(); |
- ResetInputStream(); |
- ResetOutputStream(); |
-} |
- |
-void GCMSocketStreamTest::PumpLoop() { |
- base::RunLoop run_loop; |
- run_loop.RunUntilIdle(); |
-} |
- |
-base::StringPiece GCMSocketStreamTest::DoInputStreamRead(uint64 bytes) { |
- uint64 total_bytes_read = 0; |
- const void* initial_buffer = NULL; |
- const void* buffer = NULL; |
- int size = 0; |
- |
- do { |
- DCHECK(socket_input_stream_->GetState() == SocketInputStream::EMPTY || |
- socket_input_stream_->GetState() == SocketInputStream::READY); |
- if (!socket_input_stream_->Next(&buffer, &size)) |
- break; |
- total_bytes_read += size; |
- if (initial_buffer) { // Verify the buffer doesn't skip data. |
- EXPECT_EQ(static_cast<const uint8*>(initial_buffer) + total_bytes_read, |
- static_cast<const uint8*>(buffer) + size); |
- } else { |
- initial_buffer = buffer; |
- } |
- } while (total_bytes_read < bytes); |
- |
- if (total_bytes_read > bytes) { |
- socket_input_stream_->BackUp(total_bytes_read - bytes); |
- total_bytes_read = bytes; |
- } |
- |
- return base::StringPiece(static_cast<const char*>(initial_buffer), |
- total_bytes_read); |
-} |
- |
-uint64 GCMSocketStreamTest::DoOutputStreamWrite( |
- const base::StringPiece& write_src) { |
- DCHECK_EQ(socket_output_stream_->GetState(), SocketOutputStream::EMPTY); |
- uint64 total_bytes_written = 0; |
- void* buffer = NULL; |
- int size = 0; |
- size_t bytes = write_src.size(); |
- |
- do { |
- if (!socket_output_stream_->Next(&buffer, &size)) |
- break; |
- uint64 bytes_to_write = (static_cast<uint64>(size) < bytes ? size : bytes); |
- memcpy(buffer, |
- write_src.data() + total_bytes_written, |
- bytes_to_write); |
- if (bytes_to_write < static_cast<uint64>(size)) |
- socket_output_stream_->BackUp(size - bytes_to_write); |
- total_bytes_written += bytes_to_write; |
- } while (total_bytes_written < bytes); |
- |
- base::RunLoop run_loop; |
- if (socket_output_stream_->Flush(run_loop.QuitClosure()) == |
- net::ERR_IO_PENDING) { |
- run_loop.Run(); |
- } |
- |
- return total_bytes_written; |
-} |
- |
-void GCMSocketStreamTest::WaitForData(size_t msg_size) { |
- while (input_stream()->UnreadByteCount() < msg_size) { |
- base::RunLoop run_loop; |
- if (input_stream()->Refresh(run_loop.QuitClosure(), |
- msg_size - input_stream()->UnreadByteCount()) == |
- net::ERR_IO_PENDING) { |
- run_loop.Run(); |
- } |
- if (input_stream()->GetState() == SocketInputStream::CLOSED) |
- return; |
- } |
-} |
- |
-void GCMSocketStreamTest::OpenConnection() { |
- socket_ = socket_factory_.CreateTransportClientSocket( |
- address_list_, NULL, net::NetLog::Source()); |
- socket_->Connect( |
- base::Bind(&GCMSocketStreamTest::ConnectCallback, |
- base::Unretained(this))); |
- PumpLoop(); |
-} |
- |
-void GCMSocketStreamTest::ConnectCallback(int result) {} |
- |
-void GCMSocketStreamTest::ResetInputStream() { |
- DCHECK(socket_.get()); |
- socket_input_stream_.reset(new SocketInputStream(socket_.get())); |
-} |
- |
-void GCMSocketStreamTest::ResetOutputStream() { |
- DCHECK(socket_.get()); |
- socket_output_stream_.reset(new SocketOutputStream(socket_.get())); |
-} |
- |
-// A read where all data is already available. |
-TEST_F(GCMSocketStreamTest, ReadDataSync) { |
- BuildSocket(ReadList(1, net::MockRead(net::SYNCHRONOUS, |
- kReadData, |
- kReadDataSize)), |
- WriteList()); |
- |
- WaitForData(kReadDataSize); |
- ASSERT_EQ(std::string(kReadData, kReadDataSize), |
- DoInputStreamRead(kReadDataSize)); |
-} |
- |
-// A read that comes in two parts. |
-TEST_F(GCMSocketStreamTest, ReadPartialDataSync) { |
- size_t first_read_len = kReadDataSize / 2; |
- size_t second_read_len = kReadDataSize - first_read_len; |
- ReadList read_list; |
- read_list.push_back( |
- net::MockRead(net::SYNCHRONOUS, |
- kReadData, |
- first_read_len)); |
- read_list.push_back( |
- net::MockRead(net::SYNCHRONOUS, |
- &kReadData[first_read_len], |
- second_read_len)); |
- BuildSocket(read_list, WriteList()); |
- |
- WaitForData(kReadDataSize); |
- ASSERT_EQ(std::string(kReadData, kReadDataSize), |
- DoInputStreamRead(kReadDataSize)); |
-} |
- |
-// A read where no data is available at first (IO_PENDING will be returned). |
-TEST_F(GCMSocketStreamTest, ReadAsync) { |
- size_t first_read_len = kReadDataSize / 2; |
- size_t second_read_len = kReadDataSize - first_read_len; |
- ReadList read_list; |
- read_list.push_back( |
- net::MockRead(net::SYNCHRONOUS, net::ERR_IO_PENDING)); |
- read_list.push_back( |
- net::MockRead(net::ASYNC, kReadData, first_read_len)); |
- read_list.push_back( |
- net::MockRead(net::ASYNC, &kReadData[first_read_len], second_read_len)); |
- BuildSocket(read_list, WriteList()); |
- |
- base::MessageLoop::current()->PostTask( |
- FROM_HERE, |
- base::Bind(&net::DelayedSocketData::ForceNextRead, |
- base::Unretained(data_provider()))); |
- WaitForData(kReadDataSize); |
- ASSERT_EQ(std::string(kReadData, kReadDataSize), |
- DoInputStreamRead(kReadDataSize)); |
-} |
- |
-// Simulate two packets arriving at once. Read them in two separate calls. |
-TEST_F(GCMSocketStreamTest, TwoReadsAtOnce) { |
- std::string long_data = std::string(kReadData, kReadDataSize) + |
- std::string(kReadData2, kReadData2Size); |
- BuildSocket(ReadList(1, net::MockRead(net::SYNCHRONOUS, |
- long_data.c_str(), |
- long_data.size())), |
- WriteList()); |
- |
- WaitForData(kReadDataSize); |
- ASSERT_EQ(std::string(kReadData, kReadDataSize), |
- DoInputStreamRead(kReadDataSize)); |
- |
- WaitForData(kReadData2Size); |
- ASSERT_EQ(std::string(kReadData2, kReadData2Size), |
- DoInputStreamRead(kReadData2Size)); |
-} |
- |
-// Simulate two packets arriving at once. Read them in two calls separated |
-// by a Rebuild. |
-TEST_F(GCMSocketStreamTest, TwoReadsAtOnceWithRebuild) { |
- std::string long_data = std::string(kReadData, kReadDataSize) + |
- std::string(kReadData2, kReadData2Size); |
- BuildSocket(ReadList(1, net::MockRead(net::SYNCHRONOUS, |
- long_data.c_str(), |
- long_data.size())), |
- WriteList()); |
- |
- WaitForData(kReadDataSize); |
- ASSERT_EQ(std::string(kReadData, kReadDataSize), |
- DoInputStreamRead(kReadDataSize)); |
- |
- input_stream()->RebuildBuffer(); |
- WaitForData(kReadData2Size); |
- ASSERT_EQ(std::string(kReadData2, kReadData2Size), |
- DoInputStreamRead(kReadData2Size)); |
-} |
- |
-// Simulate a read that is aborted. |
-TEST_F(GCMSocketStreamTest, ReadError) { |
- int result = net::ERR_ABORTED; |
- BuildSocket(ReadList(1, net::MockRead(net::SYNCHRONOUS, result)), |
- WriteList()); |
- |
- WaitForData(kReadDataSize); |
- ASSERT_EQ(SocketInputStream::CLOSED, input_stream()->GetState()); |
- ASSERT_EQ(result, input_stream()->last_error()); |
-} |
- |
-// Simulate a read after the connection is closed. |
-TEST_F(GCMSocketStreamTest, ReadDisconnected) { |
- BuildSocket(ReadList(), WriteList()); |
- socket()->Disconnect(); |
- WaitForData(kReadDataSize); |
- ASSERT_EQ(SocketInputStream::CLOSED, input_stream()->GetState()); |
- ASSERT_EQ(net::ERR_CONNECTION_CLOSED, input_stream()->last_error()); |
-} |
- |
-// Write a full message in one go. |
-TEST_F(GCMSocketStreamTest, WriteFull) { |
- BuildSocket(ReadList(), |
- WriteList(1, net::MockWrite(net::SYNCHRONOUS, |
- kWriteData, |
- kWriteDataSize))); |
- ASSERT_EQ(kWriteDataSize, |
- DoOutputStreamWrite(base::StringPiece(kWriteData, |
- kWriteDataSize))); |
-} |
- |
-// Write a message in two go's. |
-TEST_F(GCMSocketStreamTest, WritePartial) { |
- WriteList write_list; |
- write_list.push_back(net::MockWrite(net::SYNCHRONOUS, |
- kWriteData, |
- kWriteDataSize / 2)); |
- write_list.push_back(net::MockWrite(net::SYNCHRONOUS, |
- kWriteData + kWriteDataSize / 2, |
- kWriteDataSize / 2)); |
- BuildSocket(ReadList(), write_list); |
- ASSERT_EQ(kWriteDataSize, |
- DoOutputStreamWrite(base::StringPiece(kWriteData, |
- kWriteDataSize))); |
-} |
- |
-// Write a message completely asynchronously (returns IO_PENDING before |
-// finishing the write in two go's). |
-TEST_F(GCMSocketStreamTest, WriteNone) { |
- WriteList write_list; |
- write_list.push_back(net::MockWrite(net::SYNCHRONOUS, |
- kWriteData, |
- kWriteDataSize / 2)); |
- write_list.push_back(net::MockWrite(net::SYNCHRONOUS, |
- kWriteData + kWriteDataSize / 2, |
- kWriteDataSize / 2)); |
- BuildSocket(ReadList(), write_list); |
- ASSERT_EQ(kWriteDataSize, |
- DoOutputStreamWrite(base::StringPiece(kWriteData, |
- kWriteDataSize))); |
-} |
- |
-// Write a message then read a message. |
-TEST_F(GCMSocketStreamTest, WriteThenRead) { |
- BuildSocket(ReadList(1, net::MockRead(net::SYNCHRONOUS, |
- kReadData, |
- kReadDataSize)), |
- WriteList(1, net::MockWrite(net::SYNCHRONOUS, |
- kWriteData, |
- kWriteDataSize))); |
- |
- ASSERT_EQ(kWriteDataSize, |
- DoOutputStreamWrite(base::StringPiece(kWriteData, |
- kWriteDataSize))); |
- |
- WaitForData(kReadDataSize); |
- ASSERT_EQ(std::string(kReadData, kReadDataSize), |
- DoInputStreamRead(kReadDataSize)); |
-} |
- |
-// Read a message then write a message. |
-TEST_F(GCMSocketStreamTest, ReadThenWrite) { |
- BuildSocket(ReadList(1, net::MockRead(net::SYNCHRONOUS, |
- kReadData, |
- kReadDataSize)), |
- WriteList(1, net::MockWrite(net::SYNCHRONOUS, |
- kWriteData, |
- kWriteDataSize))); |
- |
- WaitForData(kReadDataSize); |
- ASSERT_EQ(std::string(kReadData, kReadDataSize), |
- DoInputStreamRead(kReadDataSize)); |
- |
- ASSERT_EQ(kWriteDataSize, |
- DoOutputStreamWrite(base::StringPiece(kWriteData, |
- kWriteDataSize))); |
-} |
- |
-// Simulate a write that gets aborted. |
-TEST_F(GCMSocketStreamTest, WriteError) { |
- int result = net::ERR_ABORTED; |
- BuildSocket(ReadList(), |
- WriteList(1, net::MockWrite(net::SYNCHRONOUS, result))); |
- DoOutputStreamWrite(base::StringPiece(kWriteData, kWriteDataSize)); |
- ASSERT_EQ(SocketOutputStream::CLOSED, output_stream()->GetState()); |
- ASSERT_EQ(result, output_stream()->last_error()); |
-} |
- |
-// Simulate a write after the connection is closed. |
-TEST_F(GCMSocketStreamTest, WriteDisconnected) { |
- BuildSocket(ReadList(), WriteList()); |
- socket()->Disconnect(); |
- DoOutputStreamWrite(base::StringPiece(kWriteData, kWriteDataSize)); |
- ASSERT_EQ(SocketOutputStream::CLOSED, output_stream()->GetState()); |
- ASSERT_EQ(net::ERR_CONNECTION_CLOSED, output_stream()->last_error()); |
-} |
- |
-} // namespace |
-} // namespace gcm |