Index: chrome/browser/net/proxy_browsertest.cc |
diff --git a/chrome/browser/net/proxy_browsertest.cc b/chrome/browser/net/proxy_browsertest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..dbfa42066a0e8108d240c8d5817d0d3c625ba1e9 |
--- /dev/null |
+++ b/chrome/browser/net/proxy_browsertest.cc |
@@ -0,0 +1,118 @@ |
+// 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/command_line.h" |
+#include "base/path_service.h" |
+#include "base/string_util.h" |
+#include "base/stringprintf.h" |
+#include "base/utf_string_conversions.h" |
+#include "chrome/browser/ui/browser.h" |
+#include "chrome/browser/ui/browser_tabstrip.h" |
+#include "chrome/browser/ui/login/login_prompt.h" |
+#include "chrome/browser/ui/tab_contents/tab_contents.h" |
+#include "chrome/common/chrome_notification_types.h" |
+#include "chrome/common/chrome_paths.h" |
+#include "chrome/common/chrome_switches.h" |
+#include "chrome/common/pref_names.h" |
+#include "chrome/test/base/in_process_browser_test.h" |
+#include "chrome/test/base/ui_test_utils.h" |
+#include "content/public/browser/notification_details.h" |
+#include "content/public/browser/notification_source.h" |
+#include "content/public/browser/web_contents.h" |
+#include "content/public/browser/web_contents_observer.h" |
+#include "content/public/test/browser_test_utils.h" |
+#include "net/test/test_server.h" |
+ |
+namespace { |
+ |
+// This class observes chrome::NOTIFICATION_AUTH_NEEDED and supplies |
+// the credential which is required by the test proxy server. |
+// "foo:bar" is the required username and password for our test proxy server. |
+class LoginPromptObserver : public content::NotificationObserver { |
+ public: |
+ LoginPromptObserver() : auth_handled_(false) {} |
+ |
+ virtual void Observe(int type, |
+ const content::NotificationSource& source, |
+ const content::NotificationDetails& details) OVERRIDE { |
+ if (type == chrome::NOTIFICATION_AUTH_NEEDED) { |
+ LoginNotificationDetails* login_details = |
+ content::Details<LoginNotificationDetails>(details).ptr(); |
+ // |login_details->handler()| is the associated LoginHandler object. |
+ // SetAuth() will close the login dialog. |
+ login_details->handler()->SetAuth(ASCIIToUTF16("foo"), |
+ ASCIIToUTF16("bar")); |
+ auth_handled_ = true; |
+ } |
+ } |
+ |
+ bool auth_handled() const { return auth_handled_; } |
+ |
+ private: |
+ bool auth_handled_; |
+}; |
+ |
+class ProxyBrowserTest : public InProcessBrowserTest { |
+ public: |
+ ProxyBrowserTest() |
+ : proxy_server_(net::TestServer::TYPE_BASIC_AUTH_PROXY, |
+ net::TestServer::kLocalhost, |
+ FilePath()) { |
+ } |
+ |
+ virtual void SetUp() OVERRIDE { |
+ ASSERT_TRUE(proxy_server_.Start()); |
+ InProcessBrowserTest::SetUp(); |
+ } |
+ |
+ virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { |
+ command_line->AppendSwitchASCII(switches::kProxyServer, |
+ proxy_server_.host_port_pair().ToString()); |
+ } |
+ |
+ protected: |
+ net::TestServer proxy_server_; |
+}; |
+ |
+#if defined(OS_CHROMEOS) |
+// We bypass manually installed proxy for localhost on chromeos. |
+#define MAYBE_BasicAuthWSConnect DISABLED_BasicAuthWSConnect |
+#else |
+#define MAYBE_BasicAuthWSConnect BasicAuthWSConnect |
+#endif |
+// Test that the browser can establish a WebSocket connection via a proxy |
+// that requires basic authentication. |
+IN_PROC_BROWSER_TEST_F(ProxyBrowserTest, MAYBE_BasicAuthWSConnect) { |
+ // Launch WebSocket server. |
+ content::TestWebSocketServer ws_server; |
+ int port = ws_server.UseRandomPort(); |
+ FilePath ws_root_dir; |
+ ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &ws_root_dir)); |
+ ASSERT_TRUE(ws_server.Start(ws_root_dir)); |
+ |
+ content::WebContents* tab = chrome::GetActiveWebContents(browser()); |
+ content::NavigationController* controller = &tab->GetController(); |
+ content::NotificationRegistrar registrar; |
+ // The proxy server will request basic authentication. |
+ // |observer| supplies the credential. |
+ LoginPromptObserver observer; |
+ registrar.Add(&observer, chrome::NOTIFICATION_AUTH_NEEDED, |
+ content::Source<content::NavigationController>(controller)); |
+ |
+ content::TitleWatcher watcher(tab, ASCIIToUTF16("PASS")); |
+ watcher.AlsoWaitForTitle(ASCIIToUTF16("FAIL")); |
+ |
+ // Visit a page that tries to establish WebSocket connection. The title |
+ // of the page will be 'PASS' on success. |
+ // TODO(bashi): Add TestWebSocketServer::GetURL() instead creating url here. |
+ std::string url_path = |
+ StringPrintf("%s%d%s", "http://localhost:", port, "/ws.html"); |
+ ui_test_utils::NavigateToURL(browser(), GURL(url_path)); |
+ |
+ const string16 result = watcher.WaitAndGetTitle(); |
+ EXPECT_TRUE(EqualsASCII(result, "PASS")); |
+ EXPECT_TRUE(observer.auth_handled()); |
+} |
+ |
+} // namespace |