components/devtools_http_handler/devtools_http_handler_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 "base/files/file_util.h"
   6 #include "base/files/scoped_temp_dir.h"
   7 #include "base/location.h"
   8 #include "base/run_loop.h"
   9 #include "base/single_thread_task_runner.h"
  10 #include "base/strings/string_number_conversions.h"
  11 #include "base/thread_task_runner_handle.h"
  12 #include "base/values.h"
  13 #include "components/devtools_http_handler/devtools_http_handler.h"
  14 #include "components/devtools_http_handler/devtools_http_handler_delegate.h"
  15 #include "content/public/test/test_browser_thread_bundle.h"
  16 #include "content/public/test/test_utils.h"
  17 #include "net/base/ip_endpoint.h"
  18 #include "net/base/net_errors.h"
  19 #include "net/socket/server_socket.h"
  20 #include "testing/gtest/include/gtest/gtest.h"
  21
  22 using content::BrowserThread;
  23
  24 namespace devtools_http_handler {
  25 namespace {
  26
  27 const uint16 kDummyPort = 4321;
  28 const base::FilePath::CharType kDevToolsActivePortFileName[] =
  29     FILE_PATH_LITERAL("DevToolsActivePort");
  30
  31 class DummyServerSocket : public net::ServerSocket {
  32  public:
  33   DummyServerSocket() {}
  34
  35   // net::ServerSocket "implementation"
  36   int Listen(const net::IPEndPoint& address, int backlog) override {
  37     return net::OK;
  38   }
  39
  40   int GetLocalAddress(net::IPEndPoint* address) const override {
  41     net::IPAddressNumber number;
  42     EXPECT_TRUE(net::ParseIPLiteralToNumber("127.0.0.1", &number));
  43     *address = net::IPEndPoint(number, kDummyPort);
  44     return net::OK;
  45   }
  46
  47   int Accept(scoped_ptr<net::StreamSocket>* socket,
  48              const net::CompletionCallback& callback) override {
  49     return net::ERR_IO_PENDING;
  50   }
  51 };
  52
  53 void QuitFromHandlerThread(const base::Closure& quit_closure) {
  54   BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, quit_closure);
  55 }
  56
  57 class DummyServerSocketFactory
  58     : public DevToolsHttpHandler::ServerSocketFactory {
  59  public:
  60   DummyServerSocketFactory(base::Closure quit_closure_1,
  61                            base::Closure quit_closure_2)
  62       : quit_closure_1_(quit_closure_1),
  63         quit_closure_2_(quit_closure_2) {}
  64
  65   ~DummyServerSocketFactory() override {
  66     BrowserThread::PostTask(
  67         BrowserThread::UI, FROM_HERE, quit_closure_2_);
  68   }
  69
  70  protected:
  71   scoped_ptr<net::ServerSocket> CreateForHttpServer() override {
  72     base::ThreadTaskRunnerHandle::Get()->PostTask(
  73         FROM_HERE, base::Bind(&QuitFromHandlerThread, quit_closure_1_));
  74     return scoped_ptr<net::ServerSocket>(new DummyServerSocket());
  75   }
  76
  77   base::Closure quit_closure_1_;
  78   base::Closure quit_closure_2_;
  79 };
  80
  81 class FailingServerSocketFactory : public DummyServerSocketFactory {
  82  public:
  83   FailingServerSocketFactory(const base::Closure& quit_closure_1,
  84                              const base::Closure& quit_closure_2)
  85       : DummyServerSocketFactory(quit_closure_1, quit_closure_2) {
  86   }
  87
  88  private:
  89   scoped_ptr<net::ServerSocket> CreateForHttpServer() override {
  90     base::ThreadTaskRunnerHandle::Get()->PostTask(
  91         FROM_HERE, base::Bind(&QuitFromHandlerThread, quit_closure_1_));
  92     return scoped_ptr<net::ServerSocket>();
  93   }
  94 };
  95
  96 class DummyDelegate : public DevToolsHttpHandlerDelegate {
  97  public:
  98   std::string GetDiscoveryPageHTML() override { return std::string(); }
  99
 100   std::string GetFrontendResource(const std::string& path) override {
 101     return std::string();
 102   }
 103
 104   std::string GetPageThumbnailData(const GURL& url) override {
 105     return std::string();
 106   }
 107
 108   content::DevToolsExternalAgentProxyDelegate*
 109       HandleWebSocketConnection(const std::string& path) override {
 110     return nullptr;
 111   }
 112 };
 113
 114 }
 115
 116 class DevToolsHttpHandlerTest : public testing::Test {
 117  public:
 118   DevToolsHttpHandlerTest() : testing::Test() { }
 119
 120  private:
 121   content::TestBrowserThreadBundle thread_bundle_;
 122 };
 123
 124 TEST_F(DevToolsHttpHandlerTest, TestStartStop) {
 125   base::RunLoop run_loop, run_loop_2;
 126   scoped_ptr<DevToolsHttpHandler::ServerSocketFactory> factory(
 127       new DummyServerSocketFactory(run_loop.QuitClosure(),
 128                                    run_loop_2.QuitClosure()));
 129   scoped_ptr<DevToolsHttpHandler> devtools_http_handler(
 130       new DevToolsHttpHandler(factory.Pass(),
 131                               std::string(),
 132                               new DummyDelegate(),
 133                               base::FilePath(),
 134                               base::FilePath(),
 135                               std::string(),
 136                               std::string()));
 137   // Our dummy socket factory will post a quit message once the server will
 138   // become ready.
 139   run_loop.Run();
 140   devtools_http_handler.reset();
 141   // Make sure the handler actually stops.
 142   run_loop_2.Run();
 143 }
 144
 145 TEST_F(DevToolsHttpHandlerTest, TestServerSocketFailed) {
 146   base::RunLoop run_loop, run_loop_2;
 147   scoped_ptr<DevToolsHttpHandler::ServerSocketFactory> factory(
 148       new FailingServerSocketFactory(run_loop.QuitClosure(),
 149                                      run_loop_2.QuitClosure()));
 150   scoped_ptr<DevToolsHttpHandler> devtools_http_handler(
 151       new DevToolsHttpHandler(factory.Pass(),
 152                               std::string(),
 153                               new DummyDelegate(),
 154                               base::FilePath(),
 155                               base::FilePath(),
 156                               std::string(),
 157                               std::string()));
 158   // Our dummy socket factory will post a quit message once the server will
 159   // become ready.
 160   run_loop.Run();
 161   for (int i = 0; i < 5; i++) {
 162     RunAllPendingInMessageLoop(BrowserThread::UI);
 163     RunAllPendingInMessageLoop(BrowserThread::FILE);
 164   }
 165   devtools_http_handler.reset();
 166   // Make sure the handler actually stops.
 167   run_loop_2.Run();
 168 }
 169
 170
 171 TEST_F(DevToolsHttpHandlerTest, TestDevToolsActivePort) {
 172   base::RunLoop run_loop, run_loop_2;
 173   base::ScopedTempDir temp_dir;
 174   EXPECT_TRUE(temp_dir.CreateUniqueTempDir());
 175   scoped_ptr<DevToolsHttpHandler::ServerSocketFactory> factory(
 176       new DummyServerSocketFactory(run_loop.QuitClosure(),
 177                                    run_loop_2.QuitClosure()));
 178   scoped_ptr<DevToolsHttpHandler> devtools_http_handler(
 179       new DevToolsHttpHandler(factory.Pass(),
 180                               std::string(),
 181                               new DummyDelegate(),
 182                               temp_dir.path(),
 183                               base::FilePath(),
 184                               std::string(),
 185                               std::string()));
 186   // Our dummy socket factory will post a quit message once the server will
 187   // become ready.
 188   run_loop.Run();
 189   devtools_http_handler.reset();
 190   // Make sure the handler actually stops.
 191   run_loop_2.Run();
 192
 193   // Now make sure the DevToolsActivePort was written into the
 194   // temporary directory and its contents are as expected.
 195   base::FilePath active_port_file = temp_dir.path().Append(
 196       kDevToolsActivePortFileName);
 197   EXPECT_TRUE(base::PathExists(active_port_file));
 198   std::string file_contents;
 199   EXPECT_TRUE(base::ReadFileToString(active_port_file, &file_contents));
 200   int port = 0;
 201   EXPECT_TRUE(base::StringToInt(file_contents, &port));
 202   EXPECT_EQ(static_cast<int>(kDummyPort), port);
 203 }
 204
 205 }  // namespace devtools_http_handler