From c407dc5cd9bdc5668497f21b26b09d988ab439de Mon Sep 17 00:00:00 2001 From: Ben Murdoch Date: Thu, 29 Jul 2010 17:14:53 +0100 Subject: Merge Chromium src@r53293 Change-Id: Ia79acf8670f385cee48c45b0a75371d8e950af34 --- chrome/browser/login_prompt_uitest.cc | 233 ++++++++++++++++++++++++++++++++++ 1 file changed, 233 insertions(+) create mode 100644 chrome/browser/login_prompt_uitest.cc (limited to 'chrome/browser/login_prompt_uitest.cc') 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 + +#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 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 server = + HTTPTestServer::CreateServer(kDocRoot, NULL); + ASSERT_TRUE(NULL != server.get()); + scoped_refptr 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 server = + HTTPTestServer::CreateServer(kDocRoot, NULL); + ASSERT_TRUE(NULL != server.get()); + scoped_refptr 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 server = + HTTPTestServer::CreateServer(kDocRoot, NULL); + ASSERT_TRUE(NULL != server.get()); + + scoped_refptr basic_tab(GetActiveTab()); + ASSERT_TRUE(basic_tab.get()); + ASSERT_TRUE(basic_tab->NavigateToURL(server->TestServerPage("auth-basic"))); + + AppendTab(GURL(chrome::kAboutBlankURL)); + scoped_refptr 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 server = + HTTPTestServer::CreateServer(kDocRoot, NULL); + ASSERT_TRUE(NULL != server.get()); + scoped_refptr 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 server = + HTTPTestServer::CreateServer(kDocRoot, NULL); + ASSERT_TRUE(NULL != server.get()); + + scoped_refptr 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 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 server = + HTTPTestServer::CreateServer(kDocRoot, NULL); + ASSERT_TRUE(NULL != server.get()); + + scoped_refptr 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 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); +} -- cgit v1.1