Replace WorkerWebSocketHttpLayoutTest with WorkerTest.WebSocketSharedWorker

content/browser/worker_host/test/worker_browsertest.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 "base/bind.h"
 #include "base/file_path.h"
 #include "base/logging.h"
 #include "base/path_service.h"
 #include "base/stringprintf.h"
 #include "base/string_util.h"
 #include "base/sys_info.h"
 #include "base/test/test_timeouts.h"
 #include "base/utf_string_conversions.h"
 #include "content/browser/worker_host/worker_process_host.h"
 #include "content/browser/worker_host/worker_service_impl.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/common/content_paths.h"
 #include "content/public/test/browser_test_utils.h"
 #include "content/public/test/test_utils.h"
 #include "content/shell/shell.h"
 #include "content/shell/shell_content_browser_client.h"
 #include "content/shell/shell_resource_dispatcher_host_delegate.h"
 #include "content/test/content_browser_test_utils.h"
 #include "content/test/layout_browsertest.h"
 #include "googleurl/src/gurl.h"
+#include "net/test/test_server.h"
 
 using content::BrowserThread;
 using content::WorkerServiceImpl;
 
 class WorkerLayoutTest : public InProcessBrowserLayoutTest {
  public:
   WorkerLayoutTest() : InProcessBrowserLayoutTest(
       FilePath(), FilePath().AppendASCII("fast").AppendASCII("workers")) {
   }
 };
@@ skipping 178 matching lines @@
     // "shared-worker-methods.html",
     "shared-worker-xhr-file-not-found.html",
 
     "xmlhttprequest-file-not-found.html"
   };
 
   for (size_t i = 0; i < arraysize(kLayoutTestFiles); ++i)
     RunHttpLayoutTest(kLayoutTestFiles[i]);
 }
 
-class WorkerWebSocketHttpLayoutTest : public InProcessBrowserLayoutTest {
- public:
-  WorkerWebSocketHttpLayoutTest() : InProcessBrowserLayoutTest(
-      FilePath(),
-      FilePath().AppendASCII("http").AppendASCII("tests").
-          AppendASCII("websocket").AppendASCII("tests").AppendASCII("hybi").
-          AppendASCII("workers"),
-      -1) {
-  }
-};
-
-IN_PROC_BROWSER_TEST_F(WorkerWebSocketHttpLayoutTest, DISABLED_Tests) {
-  static const char* kLayoutTestFiles[] = {
-    "close-in-onmessage-crash.html",
-    "close-in-shared-worker.html",
-    "close-in-worker.html",
-    "shared-worker-simple.html",
-    "worker-handshake-challenge-randomness.html",
-    "worker-simple.html"
-  };
-
-  FilePath websocket_test_dir;
-  ASSERT_TRUE(PathService::Get(content::DIR_LAYOUT_TESTS, &websocket_test_dir));
-
-  content::TestWebSocketServer websocket_server;
-  ASSERT_TRUE(websocket_server.Start(websocket_test_dir));
-
-  for (size_t i = 0; i < arraysize(kLayoutTestFiles); ++i)
-    RunHttpLayoutTest(kLayoutTestFiles[i]);
-}
-
 class WorkerTest : public content::ContentBrowserTest {
  public:
   WorkerTest() {}
 
   GURL GetTestURL(const std::string& test_case, const std::string& query) {
     FilePath test_file_path = content::GetTestFilePath(
         "workers", test_case.c_str());
     return content::GetFileUrlWithQuery(test_file_path, query);
   }
 
@@ skipping 208 matching lines @@
 
   // First window has hit its limit. Now launch second window which creates
   // the same worker that was queued in the first window, to ensure it gets
   // connected to the first window too.
   query = StringPrintf("?id=%d", max_workers_per_tab);
   url = GetTestURL("single_shared_worker.html", query);
   content::NavigateToURL(CreateBrowser(), url);
 
   ASSERT_TRUE(WaitForWorkerProcessCount(max_workers_per_tab + 1));
 }
+
+IN_PROC_BROWSER_TEST_F(WorkerTest, WebSocketSharedWorker) {
+  // Launch WebSocket server.
+  net::TestServer ws_server(net::TestServer::TYPE_WS,
+                            net::TestServer::kLocalhost,
+                            FilePath(FILE_PATH_LITERAL("net/data/websocket")));
+  ASSERT_TRUE(ws_server.Start());
+
+  // Generate test URL.
+  std::string scheme("http");
+  GURL::Replacements replacements;
+  replacements.SetSchemeStr(scheme);
+  GURL url = ws_server.GetURL(
+      "websocket_shared_worker.html").ReplaceComponents(replacements);
+
+  // Run test.
+  content::Shell* window = shell();
+  const string16 expected_title = ASCIIToUTF16("OK");
+  content::TitleWatcher title_watcher(window->web_contents(), expected_title);
+  content::NavigateToURL(window, url);
+  string16 final_title = title_watcher.WaitAndGetTitle();
+  EXPECT_EQ(expected_title, final_title);
+}