Use HttpAuthController in SocketStream
We need to share http auth cache with http stack because WebSocket is a
subresource and chromium won't show the login dialog for subresource
loading. Use HttpAuthController and pass the common http auth cache to
it to solve the issue.
BUG=47069
TEST=net_unittests
Review URL: https://chromiumcodereview.appspot.com/10854139
git-svn-id: svn://svn.chromium.org/chrome/trunk/src@155582 0039d316-1c4b-4281-b951-d872f2087c98
diff --git a/chrome/browser/net/proxy_browsertest.cc b/chrome/browser/net/proxy_browsertest.cc
new file mode 100644
index 0000000..dbfa420
--- /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
