Refactored code to allow associating workers with multiple renderers.

chrome/worker/worker_uitest.cc

 // Copyright (c) 2009 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/string_util.h"
 #include "chrome/app/chrome_dll_resource.h"
 #include "chrome/browser/worker_host/worker_service.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/automation/browser_proxy.h"
 #include "chrome/test/automation/tab_proxy.h"
 #include "chrome/test/ui/ui_layout_test.h"
+#include "net/url_request/url_request_unittest.h"
 
 static const char kTestCompleteCookie[] = "status";
 static const char kTestCompleteSuccess[] = "OK";
 
 // Layout test files for WorkerFastLayoutTest for the WorkerFastLayoutTest
 // shards.
 static const char* kWorkerFastLayoutTestFiles[] = {
   "stress-js-execution.html",
   "use-machine-stack.html",
   "worker-call.html",
@@ skipping 107 matching lines @@
         RunLayoutTest(kWorkerFastLayoutTestFiles[i], kNoHttpPort);
       }
     }
 
     // Navigate away from to a blank page so that any workers are cleaned up.
     // This helps leaks trackers do a better job of reporting.
     scoped_refptr<TabProxy> tab(GetActiveTab());
     GURL about_url(std::string("file://localhost/"));
     EXPECT_EQ(AUTOMATION_MSG_NAVIGATION_SUCCESS, tab->NavigateToURL(about_url));
   }
+
+  bool NavigateAndWaitForAuth(TabProxy* tab, const GURL& url) {
+    // Pass a large number of navigations to tell the tab to block until an auth
+    // dialog pops up.
+    bool timeout = false;
+    tab->NavigateToURLWithTimeout(url, 100, kTestWaitTimeoutMs, &timeout);
+    EXPECT_FALSE(timeout);
+    return tab->NeedsAuth();
+  }
 };
 
 
 TEST_F(WorkerTest, SingleWorker) {
   RunTest(L"single_worker.html");
 }
 
 #if defined(OS_LINUX)
 // MultipleWorkers times out (process hangs, does not exit) occasionally.
 // http://crbug.com/30353
@@ skipping 17 matching lines @@
 #endif
 
 // Incognito windows should not share workers with non-incognito windows
 TEST_F(WorkerTest, IncognitoSharedWorkers) {
   // Load a non-incognito tab and have it create a shared worker
   RunTest(L"incognito_worker.html");
   // Incognito worker should not share with non-incognito
   RunIncognitoTest(L"incognito_worker.html");
 }
 
+const wchar_t kDocRoot[] = L"chrome/test/data/workers";
+
+// Make sure that auth dialog is displayed from worker context.
+TEST_F(WorkerTest, WorkerHttpAuth) {
+  scoped_refptr<HTTPTestServer> server =
+      HTTPTestServer::CreateServer(kDocRoot, NULL);
+
+  ASSERT_TRUE(NULL != server.get());
+  scoped_refptr<TabProxy> tab(GetActiveTab());
+  GURL url = server->TestServerPage("files/worker_auth.html");
+  EXPECT_TRUE(NavigateAndWaitForAuth(tab, url));
+}
+
+// Make sure that auth dialog is displayed from shared worker context.
+TEST_F(WorkerTest, SharedWorkerHttpAuth) {
+  scoped_refptr<HTTPTestServer> server =
+      HTTPTestServer::CreateServer(kDocRoot, NULL);
+  ASSERT_TRUE(NULL != server.get());
+  scoped_refptr<TabProxy> tab(GetActiveTab());
+  EXPECT_EQ(1, GetTabCount());
+  GURL url = server->TestServerPage("files/shared_worker_auth.html");
+  EXPECT_TRUE(NavigateAndWaitForAuth(tab, url));
+  // TODO(atwilson): Add support to automation framework to test for auth
+  // dialogs displayed by non-navigating tabs.
+}
+
 #if defined(OS_LINUX)
 // Crashes on Linux - http://crbug.com/22898
 #define WorkerFastLayoutTests0 DISABLED_WorkerFastLayoutTests0
 #define WorkerFastLayoutTests1 DISABLED_WorkerFastLayoutTests1
 #define WorkerFastLayoutTests2 DISABLED_WorkerFastLayoutTests2
 #elif defined(OS_MACOSX)
 // Flaky on Mac - http://crbug.com/28445
 #define WorkerFastLayoutTests0 FLAKY_WorkerFastLayoutTests0
 #endif
 
@@ skipping 231 matching lines @@
 
   // Check that we didn't create more than the max number of workers.
   ASSERT_TRUE(WaitForProcessCountToBe(tab_count, total_workers));
 
   // Now close a page and check that the queued workers were started.
   tab->NavigateToURL(GetTestUrl(L"google", L"google.html"));
 
   ASSERT_TRUE(WaitForProcessCountToBe(tab_count, total_workers));
 #endif
 }