src/chrome/browser/extensions/api/sockets_tcp/sockets_tcp_apitest.cc

   1 // Copyright 2013 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/memory/ref_counted.h"
   6 #include "base/strings/stringprintf.h"
   7 #include "chrome/browser/extensions/extension_apitest.h"
   8 #include "chrome/browser/extensions/extension_service.h"
   9 #include "chrome/browser/ui/browser.h"
  10 #include "chrome/browser/ui/extensions/application_launch.h"
  11 #include "chrome/common/chrome_paths.h"
  12 #include "chrome/test/base/in_process_browser_test.h"
  13 #include "extensions/browser/api/dns/host_resolver_wrapper.h"
  14 #include "extensions/browser/api/dns/mock_host_resolver_creator.h"
  15 #include "extensions/browser/api/sockets_tcp/sockets_tcp_api.h"
  16 #include "extensions/test/extension_test_message_listener.h"
  17 #include "extensions/test/result_catcher.h"
  18 #include "net/dns/mock_host_resolver.h"
  19 #include "net/test/spawned_test_server/spawned_test_server.h"
  20
  21 namespace {
  22
  23 // TODO(jschuh): Hanging plugin tests. crbug.com/244653
  24 #if defined(OS_WIN) && defined(ARCH_CPU_X86_64)
  25 #define MAYBE(x) DISABLED_##x
  26 #else
  27 #define MAYBE(x) x
  28 #endif
  29
  30 const std::string kHostname = "127.0.0.1";
  31
  32 class SocketsTcpApiTest : public ExtensionApiTest {
  33  public:
  34   SocketsTcpApiTest() : resolver_event_(true, false),
  35                         resolver_creator_(
  36                             new extensions::MockHostResolverCreator()) {
  37   }
  38
  39   void SetUpOnMainThread() override {
  40     extensions::HostResolverWrapper::GetInstance()->SetHostResolverForTesting(
  41         resolver_creator_->CreateMockHostResolver());
  42   }
  43
  44   void TearDownOnMainThread() override {
  45     extensions::HostResolverWrapper::GetInstance()->
  46         SetHostResolverForTesting(NULL);
  47     resolver_creator_->DeleteMockHostResolver();
  48   }
  49
  50  private:
  51   base::WaitableEvent resolver_event_;
  52
  53   // The MockHostResolver asserts that it's used on the same thread on which
  54   // it's created, which is actually a stronger rule than its real counterpart.
  55   // But that's fine; it's good practice.
  56   scoped_refptr<extensions::MockHostResolverCreator> resolver_creator_;
  57 };
  58
  59 }  // namespace
  60
  61 IN_PROC_BROWSER_TEST_F(SocketsTcpApiTest, SocketTcpExtension) {
  62   scoped_ptr<net::SpawnedTestServer> test_server(
  63       new net::SpawnedTestServer(
  64           net::SpawnedTestServer::TYPE_TCP_ECHO,
  65           net::SpawnedTestServer::kLocalhost,
  66           base::FilePath(FILE_PATH_LITERAL("net/data"))));
  67   EXPECT_TRUE(test_server->Start());
  68
  69   net::HostPortPair host_port_pair = test_server->host_port_pair();
  70   int port = host_port_pair.port();
  71   ASSERT_TRUE(port > 0);
  72
  73   // Test that connect() is properly resolving hostnames.
  74   host_port_pair.set_host("lOcAlHoSt");
  75
  76   extensions::ResultCatcher catcher;
  77   catcher.RestrictToBrowserContext(browser()->profile());
  78
  79   ExtensionTestMessageListener listener("info_please", true);
  80
  81   ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("sockets_tcp/api")));
  82   EXPECT_TRUE(listener.WaitUntilSatisfied());
  83   listener.Reply(
  84       base::StringPrintf("tcp:%s:%d", host_port_pair.host().c_str(), port));
  85
  86   EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
  87 }
  88
  89 IN_PROC_BROWSER_TEST_F(SocketsTcpApiTest, SocketTcpExtensionTLS) {
  90   scoped_ptr<net::SpawnedTestServer> test_https_server(
  91       new net::SpawnedTestServer(
  92           net::SpawnedTestServer::TYPE_HTTPS,
  93           net::BaseTestServer::SSLOptions(),
  94           base::FilePath(FILE_PATH_LITERAL("net/data"))));
  95   EXPECT_TRUE(test_https_server->Start());
  96
  97   net::HostPortPair https_host_port_pair = test_https_server->host_port_pair();
  98   int https_port = https_host_port_pair.port();
  99   ASSERT_GT(https_port, 0);
 100
 101   extensions::ResultCatcher catcher;
 102   catcher.RestrictToBrowserContext(browser()->profile());
 103
 104   ExtensionTestMessageListener listener("info_please", true);
 105
 106   ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("sockets_tcp/api")));
 107   EXPECT_TRUE(listener.WaitUntilSatisfied());
 108   listener.Reply(base::StringPrintf(
 109       "https:%s:%d", https_host_port_pair.host().c_str(), https_port));
 110
 111   EXPECT_TRUE(catcher.GetNextResult()) << catcher.message();
 112 }