Merge Chromium src@r53293

Change-Id: Ia79acf8670f385cee48c45b0a75371d8e950af34

1 files changed, 233 insertions, 0 deletions

diff --git a/chrome/browser/login_prompt_uitest.cc b/chrome/browser/login_prompt_uitest.cc
new file mode 100644
index 0000000..e1dc5af
--- /dev/null
+++ b/chrome/browser/login_prompt_uitest.cc
@@ -0,0 +1,233 @@
+// Copyright (c) 2006-2008 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 <string>
+
+#include "chrome/browser/net/url_fixer_upper.h"
+#include "chrome/common/chrome_switches.h"
+#include "chrome/common/url_constants.h"
+#include "chrome/test/automation/tab_proxy.h"
+#include "chrome/test/automation/browser_proxy.h"
+#include "chrome/test/ui/ui_test.h"
+#include "net/url_request/url_request_unittest.h"
+
+using std::wstring;
+
+namespace {
+
+const wchar_t kDocRoot[] = L"chrome/test/data";
+
+}  // namespace
+
+class LoginPromptTest : public UITest {
+ protected:
+  LoginPromptTest()
+      : UITest(),
+        username_basic_(L"basicuser"),
+        username_digest_(L"digestuser"),
+        password_(L"secret"),
+        password_bad_(L"denyme") {
+  }
+
+  void AppendTab(const GURL& url) {
+    scoped_refptr<BrowserProxy> window_proxy(automation()->GetBrowserWindow(0));
+    ASSERT_TRUE(window_proxy.get());
+    ASSERT_TRUE(window_proxy->AppendTab(url));
+  }
+
+ protected:
+  wstring username_basic_;
+  wstring username_digest_;
+  wstring password_;
+  wstring password_bad_;
+};
+
+wstring ExpectedTitleFromAuth(const wstring& username,
+                              const wstring& password) {
+  // The TestServer sets the title to username/password on successful login.
+  return username + L"/" + password;
+}
+
+// Test that "Basic" HTTP authentication works.
+TEST_F(LoginPromptTest, TestBasicAuth) {
+  scoped_refptr<HTTPTestServer> server =
+      HTTPTestServer::CreateServer(kDocRoot, NULL);
+  ASSERT_TRUE(NULL != server.get());
+  scoped_refptr<TabProxy> tab(GetActiveTab());
+  ASSERT_TRUE(tab.get());
+  ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-basic")));
+
+  EXPECT_TRUE(tab->NeedsAuth());
+  EXPECT_FALSE(tab->SetAuth(username_basic_, password_bad_));
+  EXPECT_TRUE(tab->NeedsAuth());
+  EXPECT_TRUE(tab->CancelAuth());
+  EXPECT_EQ(L"Denied: wrong password", GetActiveTabTitle());
+
+  ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-basic")));
+
+  EXPECT_TRUE(tab->NeedsAuth());
+  EXPECT_TRUE(tab->SetAuth(username_basic_, password_));
+  EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_),
+            GetActiveTabTitle());
+}
+
+// Test that "Digest" HTTP authentication works.
+TEST_F(LoginPromptTest, TestDigestAuth) {
+  scoped_refptr<HTTPTestServer> server =
+      HTTPTestServer::CreateServer(kDocRoot, NULL);
+  ASSERT_TRUE(NULL != server.get());
+  scoped_refptr<TabProxy> tab(GetActiveTab());
+  ASSERT_TRUE(tab.get());
+  ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-digest")));
+
+  EXPECT_TRUE(tab->NeedsAuth());
+  EXPECT_FALSE(tab->SetAuth(username_digest_, password_bad_));
+  EXPECT_TRUE(tab->CancelAuth());
+  EXPECT_EQ(L"Denied: wrong password", GetActiveTabTitle());
+
+  ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-digest")));
+
+  EXPECT_TRUE(tab->NeedsAuth());
+  EXPECT_TRUE(tab->SetAuth(username_digest_, password_));
+  EXPECT_EQ(ExpectedTitleFromAuth(username_digest_, password_),
+            GetActiveTabTitle());
+}
+
+// Test that logging in on 2 tabs at once works.
+TEST_F(LoginPromptTest, TestTwoAuths) {
+  scoped_refptr<HTTPTestServer> server =
+      HTTPTestServer::CreateServer(kDocRoot, NULL);
+  ASSERT_TRUE(NULL != server.get());
+
+  scoped_refptr<TabProxy> basic_tab(GetActiveTab());
+  ASSERT_TRUE(basic_tab.get());
+  ASSERT_TRUE(basic_tab->NavigateToURL(server->TestServerPage("auth-basic")));
+
+  AppendTab(GURL(chrome::kAboutBlankURL));
+  scoped_refptr<TabProxy> digest_tab(GetActiveTab());
+  ASSERT_TRUE(digest_tab.get());
+  ASSERT_TRUE(
+      digest_tab->NavigateToURL(server->TestServerPage("auth-digest")));
+
+  EXPECT_TRUE(basic_tab->NeedsAuth());
+  EXPECT_TRUE(basic_tab->SetAuth(username_basic_, password_));
+  EXPECT_TRUE(digest_tab->NeedsAuth());
+  EXPECT_TRUE(digest_tab->SetAuth(username_digest_, password_));
+
+  wstring title;
+  EXPECT_TRUE(basic_tab->GetTabTitle(&title));
+  EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), title);
+
+  EXPECT_TRUE(digest_tab->GetTabTitle(&title));
+  EXPECT_EQ(ExpectedTitleFromAuth(username_digest_, password_), title);
+}
+
+// Test that cancelling authentication works.
+TEST_F(LoginPromptTest, TestCancelAuth) {
+  scoped_refptr<HTTPTestServer> server =
+      HTTPTestServer::CreateServer(kDocRoot, NULL);
+  ASSERT_TRUE(NULL != server.get());
+  scoped_refptr<TabProxy> tab(GetActiveTab());
+  ASSERT_TRUE(tab.get());
+
+  // First navigate to a test server page so we have something to go back to.
+  ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("a")));
+
+  // Navigating while auth is requested is the same as cancelling.
+  ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-basic")));
+  EXPECT_TRUE(tab->NeedsAuth());
+  ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("b")));
+  EXPECT_FALSE(tab->NeedsAuth());
+
+  ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-basic")));
+  EXPECT_TRUE(tab->NeedsAuth());
+  EXPECT_TRUE(tab->GoBack());  // should bring us back to 'a'
+  EXPECT_FALSE(tab->NeedsAuth());
+
+  // Now add a page and go back, so we have something to go forward to.
+  ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("c")));
+  EXPECT_TRUE(tab->GoBack());  // should bring us back to 'a'
+
+  ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-basic")));
+  EXPECT_TRUE(tab->NeedsAuth());
+  EXPECT_TRUE(tab->GoForward());  // should bring us to 'c'
+  EXPECT_FALSE(tab->NeedsAuth());
+
+  // Now test that cancelling works as expected.
+  ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-basic")));
+  EXPECT_TRUE(tab->NeedsAuth());
+  EXPECT_TRUE(tab->CancelAuth());
+  EXPECT_FALSE(tab->NeedsAuth());
+  EXPECT_EQ(L"Denied: no auth", GetActiveTabTitle());
+}
+
+// If multiple tabs are looking for the same auth, the user should only have to
+// enter it once (http://crbug.com/8914).
+TEST_F(LoginPromptTest, SupplyRedundantAuths) {
+  scoped_refptr<HTTPTestServer> server =
+      HTTPTestServer::CreateServer(kDocRoot, NULL);
+  ASSERT_TRUE(NULL != server.get());
+
+  scoped_refptr<TabProxy> basic_tab1(GetActiveTab());
+  ASSERT_TRUE(basic_tab1.get());
+  ASSERT_TRUE(
+      basic_tab1->NavigateToURL(server->TestServerPage("auth-basic/1")));
+  EXPECT_TRUE(basic_tab1->NeedsAuth());
+
+  AppendTab(GURL(chrome::kAboutBlankURL));
+  scoped_refptr<TabProxy> basic_tab2(GetActiveTab());
+  ASSERT_TRUE(basic_tab2.get());
+  ASSERT_TRUE(
+      basic_tab2->NavigateToURL(server->TestServerPage("auth-basic/2")));
+  EXPECT_TRUE(basic_tab2->NeedsAuth());
+
+  // Set the auth in only one of the tabs (but wait for the other to load).
+  int64 last_navigation_time;
+  ASSERT_TRUE(basic_tab2->GetLastNavigationTime(&last_navigation_time));
+  EXPECT_TRUE(basic_tab1->SetAuth(username_basic_, password_));
+  EXPECT_TRUE(basic_tab2->WaitForNavigation(last_navigation_time));
+
+  // Now both tabs have loaded.
+  wstring title1;
+  EXPECT_TRUE(basic_tab1->GetTabTitle(&title1));
+  EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), title1);
+  wstring title2;
+  EXPECT_TRUE(basic_tab2->GetTabTitle(&title2));
+  EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), title2);
+}
+
+// If multiple tabs are looking for the same auth, and one is cancelled, the
+// other should be cancelled as well.
+TEST_F(LoginPromptTest, CancelRedundantAuths) {
+  scoped_refptr<HTTPTestServer> server =
+      HTTPTestServer::CreateServer(kDocRoot, NULL);
+  ASSERT_TRUE(NULL != server.get());
+
+  scoped_refptr<TabProxy> basic_tab1(GetActiveTab());
+  ASSERT_TRUE(basic_tab1.get());
+  ASSERT_TRUE(
+      basic_tab1->NavigateToURL(server->TestServerPage("auth-basic/1")));
+  EXPECT_TRUE(basic_tab1->NeedsAuth());
+
+  AppendTab(GURL(chrome::kAboutBlankURL));
+  scoped_refptr<TabProxy> basic_tab2(GetActiveTab());
+  ASSERT_TRUE(basic_tab2.get());
+  ASSERT_TRUE(
+      basic_tab2->NavigateToURL(server->TestServerPage("auth-basic/2")));
+  EXPECT_TRUE(basic_tab2->NeedsAuth());
+
+  // Cancel the auth in only one of the tabs (but wait for the other to load).
+  int64 last_navigation_time;
+  ASSERT_TRUE(basic_tab2->GetLastNavigationTime(&last_navigation_time));
+  EXPECT_TRUE(basic_tab1->CancelAuth());
+  EXPECT_TRUE(basic_tab2->WaitForNavigation(last_navigation_time));
+
+  // Now both tabs have been denied.
+  wstring title1;
+  EXPECT_TRUE(basic_tab1->GetTabTitle(&title1));
+  EXPECT_EQ(L"Denied: no auth", title1);
+  wstring title2;
+  EXPECT_TRUE(basic_tab2->GetTabTitle(&title2));
+  EXPECT_EQ(L"Denied: no auth", title2);
+}