[chromedriver] Implement connecting to devtools and loading a page.

chrome/test/chromedriver/net/net_util_unittest.cc

+// 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 <string>
+
+#include "base/bind.h"
+#include "base/compiler_specific.h"
+#include "base/location.h"
+#include "base/memory/ref_counted.h"
+#include "base/message_loop.h"
+#include "base/message_loop_proxy.h"
+#include "base/single_thread_task_runner.h"
+#include "base/stringprintf.h"
+#include "base/synchronization/waitable_event.h"
+#include "base/threading/thread.h"
+#include "chrome/test/chromedriver/net/net_util.h"
+#include "chrome/test/chromedriver/net/url_request_context_getter.h"
+#include "googleurl/src/gurl.h"
+#include "net/base/ip_endpoint.h"
+#include "net/base/net_errors.h"
+#include "net/base/tcp_listen_socket.h"
+#include "net/server/http_server.h"
+#include "net/server/http_server_request_info.h"
+#include "net/url_request/url_request_context_getter.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace {
+
+class FetchUrlTest : public testing::Test,
+                     public net::HttpServer::Delegate {
+ public:
+  FetchUrlTest()
+      : io_thread_("io"),
+        response_(kSendHello) {
+    base::Thread::Options options(MessageLoop::TYPE_IO, 0);
+    CHECK(io_thread_.StartWithOptions(options));
+    context_getter_ = new URLRequestContextGetter(
+        io_thread_.message_loop_proxy());
+    base::WaitableEvent event(false, false);
+    io_thread_.message_loop_proxy()->PostTask(
+        FROM_HERE,
+        base::Bind(&FetchUrlTest::InitOnIO,
+                   base::Unretained(this), &event));
+    event.Wait();
+  }
+
+  virtual ~FetchUrlTest() {
+    base::WaitableEvent event(false, false);
+    io_thread_.message_loop_proxy()->PostTask(
+        FROM_HERE,
+        base::Bind(&FetchUrlTest::DestroyServerOnIO,
+                   base::Unretained(this), &event));
+    event.Wait();
+  }
+
+  void InitOnIO(base::WaitableEvent* event) {
+    net::TCPListenSocketFactory factory("127.0.0.1", 0);
+    server_ = new net::HttpServer(factory, this);
+    net::IPEndPoint address;
+    CHECK_EQ(net::OK, server_->GetLocalAddress(&address));
+    server_url_ = GURL(
+        base::StringPrintf("http://127.0.0.1:%d", address.port()));
+    event->Signal();
+  }
+
+  void DestroyServerOnIO(base::WaitableEvent* event) {
+    server_ = NULL;
+    event->Signal();
+  }
+
+  // Overridden from net::HttpServer::Delegate:
+  virtual void OnHttpRequest(int connection_id,
+                             const net::HttpServerRequestInfo& info) OVERRIDE {
+    switch (response_) {
+      case kSendHello:
+        server_->Send200(connection_id, "hello", "text/plain");
+        break;
+      case kSend404:
+        server_->Send404(connection_id);
+        break;
+      case kClose:
+        // net::HttpServer doesn't allow us to close connection during callback.
+        MessageLoop::current()->PostTask(
+            FROM_HERE,
+            base::Bind(&net::HttpServer::Close, server_, connection_id));
+        break;
+      default:
+        break;
+    }
+  }
+
+  virtual void OnWebSocketRequest(
+      int connection_id,
+      const net::HttpServerRequestInfo& info) OVERRIDE {}
+  virtual void OnWebSocketMessage(int connection_id,
+                                  const std::string& data) OVERRIDE {}
+  virtual void OnClose(int connection_id) OVERRIDE {}
+
+ protected:
+  enum ServerResponse {
+    kSendHello = 0,
+    kSend404,
+    kClose,
+  };
+
+  base::Thread io_thread_;
+  ServerResponse response_;
+  scoped_refptr<net::HttpServer> server_;
+  scoped_refptr<URLRequestContextGetter> context_getter_;
+  GURL server_url_;
+};
+
+}  // namespace
+
+TEST_F(FetchUrlTest, Http200) {
+  std::string response("stuff");
+  ASSERT_TRUE(FetchUrl(server_url_, context_getter_, &response));
+  ASSERT_STREQ("hello", response.c_str());
+}
+
+TEST_F(FetchUrlTest, HttpNon200) {
+  response_ = kSend404;
+  std::string response("stuff");
+  ASSERT_FALSE(FetchUrl(server_url_, context_getter_, &response));
+  ASSERT_STREQ("stuff", response.c_str());
+}
+
+TEST_F(FetchUrlTest, ConnectionClose) {
+  response_ = kClose;
+  std::string response("stuff");
+  ASSERT_FALSE(FetchUrl(server_url_, context_getter_, &response));
+  ASSERT_STREQ("stuff", response.c_str());
+}
+
+TEST_F(FetchUrlTest, NoServer) {
+  std::string response("stuff");
+  GURL bogus_url("http://localhost:33333");
+  ASSERT_FALSE(FetchUrl(bogus_url, context_getter_, &response));
+  ASSERT_STREQ("stuff", response.c_str());
+}