chrome/test/chromedriver/net/net_util_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include <string>
   6
   7 #include "base/bind.h"
   8 #include "base/compiler_specific.h"
   9 #include "base/location.h"
  10 #include "base/memory/ref_counted.h"
  11 #include "base/message_loop/message_loop.h"
  12 #include "base/message_loop/message_loop_proxy.h"
  13 #include "base/single_thread_task_runner.h"
  14 #include "base/strings/stringprintf.h"
  15 #include "base/synchronization/waitable_event.h"
  16 #include "base/threading/thread.h"
  17 #include "chrome/test/chromedriver/net/net_util.h"
  18 #include "chrome/test/chromedriver/net/url_request_context_getter.h"
  19 #include "net/base/ip_endpoint.h"
  20 #include "net/base/net_errors.h"
  21 #include "net/server/http_server.h"
  22 #include "net/server/http_server_request_info.h"
  23 #include "net/socket/tcp_server_socket.h"
  24 #include "net/url_request/url_request_context_getter.h"
  25 #include "testing/gtest/include/gtest/gtest.h"
  26
  27 namespace {
  28
  29 class FetchUrlTest : public testing::Test,
  30                      public net::HttpServer::Delegate {
  31  public:
  32   FetchUrlTest()
  33       : io_thread_("io"),
  34         response_(kSendHello) {
  35     base::Thread::Options options(base::MessageLoop::TYPE_IO, 0);
  36     CHECK(io_thread_.StartWithOptions(options));
  37     context_getter_ = new URLRequestContextGetter(
  38         io_thread_.message_loop_proxy());
  39     base::WaitableEvent event(false, false);
  40     io_thread_.message_loop_proxy()->PostTask(
  41         FROM_HERE,
  42         base::Bind(&FetchUrlTest::InitOnIO,
  43                    base::Unretained(this), &event));
  44     event.Wait();
  45   }
  46
  47   virtual ~FetchUrlTest() {
  48     base::WaitableEvent event(false, false);
  49     io_thread_.message_loop_proxy()->PostTask(
  50         FROM_HERE,
  51         base::Bind(&FetchUrlTest::DestroyServerOnIO,
  52                    base::Unretained(this), &event));
  53     event.Wait();
  54   }
  55
  56   void InitOnIO(base::WaitableEvent* event) {
  57     scoped_ptr<net::ServerSocket> server_socket(
  58         new net::TCPServerSocket(NULL, net::NetLog::Source()));
  59     server_socket->ListenWithAddressAndPort("127.0.0.1", 0, 1);
  60     server_.reset(new net::HttpServer(server_socket.Pass(), this));
  61     net::IPEndPoint address;
  62     CHECK_EQ(net::OK, server_->GetLocalAddress(&address));
  63     server_url_ = base::StringPrintf("http://127.0.0.1:%d", address.port());
  64     event->Signal();
  65   }
  66
  67   void DestroyServerOnIO(base::WaitableEvent* event) {
  68     server_.reset(NULL);
  69     event->Signal();
  70   }
  71
  72   // Overridden from net::HttpServer::Delegate:
  73   virtual void OnConnect(int connection_id) override {}
  74
  75   virtual void OnHttpRequest(int connection_id,
  76                              const net::HttpServerRequestInfo& info) override {
  77     switch (response_) {
  78       case kSendHello:
  79         server_->Send200(connection_id, "hello", "text/plain");
  80         break;
  81       case kSend404:
  82         server_->Send404(connection_id);
  83         break;
  84       case kClose:
  85         server_->Close(connection_id);
  86         break;
  87       default:
  88         break;
  89     }
  90   }
  91
  92   virtual void OnWebSocketRequest(
  93       int connection_id,
  94       const net::HttpServerRequestInfo& info) override {}
  95   virtual void OnWebSocketMessage(int connection_id,
  96                                   const std::string& data) override {}
  97   virtual void OnClose(int connection_id) override {}
  98
  99  protected:
 100   enum ServerResponse {
 101     kSendHello = 0,
 102     kSend404,
 103     kClose,
 104   };
 105
 106   base::Thread io_thread_;
 107   ServerResponse response_;
 108   scoped_ptr<net::HttpServer> server_;
 109   scoped_refptr<URLRequestContextGetter> context_getter_;
 110   std::string server_url_;
 111 };
 112
 113 }  // namespace
 114
 115 TEST_F(FetchUrlTest, Http200) {
 116   std::string response("stuff");
 117   ASSERT_TRUE(FetchUrl(server_url_, context_getter_.get(), &response));
 118   ASSERT_STREQ("hello", response.c_str());
 119 }
 120
 121 TEST_F(FetchUrlTest, HttpNon200) {
 122   response_ = kSend404;
 123   std::string response("stuff");
 124   ASSERT_FALSE(FetchUrl(server_url_, context_getter_.get(), &response));
 125   ASSERT_STREQ("stuff", response.c_str());
 126 }
 127
 128 TEST_F(FetchUrlTest, ConnectionClose) {
 129   response_ = kClose;
 130   std::string response("stuff");
 131   ASSERT_FALSE(FetchUrl(server_url_, context_getter_.get(), &response));
 132   ASSERT_STREQ("stuff", response.c_str());
 133 }
 134
 135 TEST_F(FetchUrlTest, NoServer) {
 136   std::string response("stuff");
 137   ASSERT_FALSE(
 138       FetchUrl("http://localhost:33333", context_getter_.get(), &response));
 139   ASSERT_STREQ("stuff", response.c_str());
 140 }