Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(72)

Unified Diff: google_apis/gcm/base/socket_stream_unittest.cc

Issue 29273008: Revert r229533 "[GCM] Initial work to set up directory structure and introduce socket integration" (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 7 years, 2 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « google_apis/gcm/base/socket_stream.cc ('k') | google_apis/gcm/gcm.gyp » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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
« no previous file with comments | « google_apis/gcm/base/socket_stream.cc ('k') | google_apis/gcm/gcm.gyp » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698