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

Unified Diff: net/base/listen_socket_unittest.cc

Issue 10108015: Upstream changes making ListenSocket an abstract class. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Add listen_socket.cc. Created 8 years, 8 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 | « net/base/listen_socket_unittest.h ('k') | net/base/tcp_listen_socket.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: net/base/listen_socket_unittest.cc
diff --git a/net/base/listen_socket_unittest.cc b/net/base/listen_socket_unittest.cc
deleted file mode 100644
index 5e3811e1d991f425f290cf6432729b702396ea99..0000000000000000000000000000000000000000
--- a/net/base/listen_socket_unittest.cc
+++ /dev/null
@@ -1,248 +0,0 @@
-// Copyright (c) 2012 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 "net/base/listen_socket_unittest.h"
-
-#include <fcntl.h>
-#include <sys/types.h>
-
-#include "base/bind.h"
-#include "base/eintr_wrapper.h"
-#include "base/sys_byteorder.h"
-#include "net/base/net_util.h"
-#include "testing/platform_test.h"
-
-namespace net {
-
-const int ListenSocketTester::kTestPort = 9999;
-
-static const int kReadBufSize = 1024;
-static const char kHelloWorld[] = "HELLO, WORLD";
-static const int kMaxQueueSize = 20;
-static const char kLoopback[] = "127.0.0.1";
-static const int kDefaultTimeoutMs = 5000;
-
-ListenSocketTester::ListenSocketTester()
- : thread_(NULL),
- loop_(NULL),
- server_(NULL),
- connection_(NULL),
- cv_(&lock_) {
-}
-
-void ListenSocketTester::SetUp() {
- base::Thread::Options options;
- options.message_loop_type = MessageLoop::TYPE_IO;
- thread_.reset(new base::Thread("socketio_test"));
- thread_->StartWithOptions(options);
- loop_ = reinterpret_cast<MessageLoopForIO*>(thread_->message_loop());
-
- loop_->PostTask(FROM_HERE, base::Bind(&ListenSocketTester::Listen, this));
-
- // verify Listen succeeded
- NextAction();
- ASSERT_FALSE(server_ == NULL);
- ASSERT_EQ(ACTION_LISTEN, last_action_.type());
-
- // verify the connect/accept and setup test_socket_
- test_socket_ = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
- ASSERT_NE(INVALID_SOCKET, test_socket_);
- struct sockaddr_in client;
- client.sin_family = AF_INET;
- client.sin_addr.s_addr = inet_addr(kLoopback);
- client.sin_port = base::HostToNet16(kTestPort);
- int ret = HANDLE_EINTR(
- connect(test_socket_, reinterpret_cast<sockaddr*>(&client),
- sizeof(client)));
- ASSERT_NE(ret, SOCKET_ERROR);
-
- NextAction();
- ASSERT_EQ(ACTION_ACCEPT, last_action_.type());
-}
-
-void ListenSocketTester::TearDown() {
-#if defined(OS_WIN)
- ASSERT_EQ(0, closesocket(test_socket_));
-#elif defined(OS_POSIX)
- ASSERT_EQ(0, HANDLE_EINTR(close(test_socket_)));
-#endif
- NextAction();
- ASSERT_EQ(ACTION_CLOSE, last_action_.type());
-
- loop_->PostTask(FROM_HERE, base::Bind(&ListenSocketTester::Shutdown, this));
- NextAction();
- ASSERT_EQ(ACTION_SHUTDOWN, last_action_.type());
-
- thread_.reset();
- loop_ = NULL;
-}
-
-void ListenSocketTester::ReportAction(const ListenSocketTestAction& action) {
- base::AutoLock locked(lock_);
- queue_.push_back(action);
- cv_.Broadcast();
-}
-
-void ListenSocketTester::NextAction() {
- base::AutoLock locked(lock_);
- while (queue_.empty())
- cv_.Wait();
- last_action_ = queue_.front();
- queue_.pop_front();
-}
-
-int ListenSocketTester::ClearTestSocket() {
- char buf[kReadBufSize];
- int len_ret = 0;
- do {
- int len = HANDLE_EINTR(recv(test_socket_, buf, kReadBufSize, 0));
- if (len == SOCKET_ERROR || len == 0) {
- break;
- } else {
- len_ret += len;
- }
- } while (true);
- return len_ret;
-}
-
-void ListenSocketTester::Shutdown() {
- connection_->Release();
- connection_ = NULL;
- server_->Release();
- server_ = NULL;
- ReportAction(ListenSocketTestAction(ACTION_SHUTDOWN));
-}
-
-void ListenSocketTester::Listen() {
- server_ = DoListen();
- if (server_) {
- server_->AddRef();
- ReportAction(ListenSocketTestAction(ACTION_LISTEN));
- }
-}
-
-void ListenSocketTester::SendFromTester() {
- connection_->Send(kHelloWorld);
- ReportAction(ListenSocketTestAction(ACTION_SEND));
-}
-
-void ListenSocketTester::TestClientSend() {
- ASSERT_TRUE(Send(test_socket_, kHelloWorld));
- NextAction();
- ASSERT_EQ(ACTION_READ, last_action_.type());
- ASSERT_EQ(last_action_.data(), kHelloWorld);
-}
-
-void ListenSocketTester::TestClientSendLong() {
- size_t hello_len = strlen(kHelloWorld);
- std::string long_string;
- size_t long_len = 0;
- for (int i = 0; i < 200; i++) {
- long_string += kHelloWorld;
- long_len += hello_len;
- }
- ASSERT_TRUE(Send(test_socket_, long_string));
- size_t read_len = 0;
- while (read_len < long_len) {
- NextAction();
- ASSERT_EQ(ACTION_READ, last_action_.type());
- std::string last_data = last_action_.data();
- size_t len = last_data.length();
- if (long_string.compare(read_len, len, last_data)) {
- ASSERT_EQ(long_string.compare(read_len, len, last_data), 0);
- }
- read_len += last_data.length();
- }
- ASSERT_EQ(read_len, long_len);
-}
-
-void ListenSocketTester::TestServerSend() {
- loop_->PostTask(FROM_HERE, base::Bind(
- &ListenSocketTester::SendFromTester, this));
- NextAction();
- ASSERT_EQ(ACTION_SEND, last_action_.type());
- const int buf_len = 200;
- char buf[buf_len+1];
- unsigned recv_len = 0;
- while (recv_len < strlen(kHelloWorld)) {
- int r = HANDLE_EINTR(recv(test_socket_, buf, buf_len, 0));
- ASSERT_GE(r, 0);
- recv_len += static_cast<unsigned>(r);
- if (!r)
- break;
- }
- buf[recv_len] = 0;
- ASSERT_STREQ(buf, kHelloWorld);
-}
-
-bool ListenSocketTester::Send(SOCKET sock, const std::string& str) {
- int len = static_cast<int>(str.length());
- int send_len = HANDLE_EINTR(send(sock, str.data(), len, 0));
- if (send_len == SOCKET_ERROR) {
- LOG(ERROR) << "send failed: " << errno;
- return false;
- } else if (send_len != len) {
- return false;
- }
- return true;
-}
-
-void ListenSocketTester::DidAccept(ListenSocket *server,
- ListenSocket *connection) {
- connection_ = connection;
- connection_->AddRef();
- ReportAction(ListenSocketTestAction(ACTION_ACCEPT));
-}
-
-void ListenSocketTester::DidRead(ListenSocket *connection,
- const char* data,
- int len) {
- std::string str(data, len);
- ReportAction(ListenSocketTestAction(ACTION_READ, str));
-}
-
-void ListenSocketTester::DidClose(ListenSocket *sock) {
- ReportAction(ListenSocketTestAction(ACTION_CLOSE));
-}
-
-ListenSocketTester::~ListenSocketTester() {}
-
-ListenSocket* ListenSocketTester::DoListen() {
- return ListenSocket::Listen(kLoopback, kTestPort, this);
-}
-
-class ListenSocketTest: public PlatformTest {
- public:
- ListenSocketTest() {
- tester_ = NULL;
- }
-
- virtual void SetUp() {
- PlatformTest::SetUp();
- tester_ = new ListenSocketTester();
- tester_->SetUp();
- }
-
- virtual void TearDown() {
- PlatformTest::TearDown();
- tester_->TearDown();
- tester_ = NULL;
- }
-
- scoped_refptr<ListenSocketTester> tester_;
-};
-
-TEST_F(ListenSocketTest, ClientSend) {
- tester_->TestClientSend();
-}
-
-TEST_F(ListenSocketTest, ClientSendLong) {
- tester_->TestClientSendLong();
-}
-
-TEST_F(ListenSocketTest, ServerSend) {
- tester_->TestServerSend();
-}
-
-} // namespace net
« no previous file with comments | « net/base/listen_socket_unittest.h ('k') | net/base/tcp_listen_socket.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698