| OLD | NEW |
| 1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <string> | 5 #include <string> |
| 6 | 6 |
| 7 #include "chrome/browser/net/url_fixer_upper.h" | 7 #include "chrome/browser/net/url_fixer_upper.h" |
| 8 #include "chrome/common/chrome_switches.h" | 8 #include "chrome/common/chrome_switches.h" |
| 9 #include "chrome/common/url_constants.h" | 9 #include "chrome/common/url_constants.h" |
| 10 #include "chrome/test/automation/tab_proxy.h" | 10 #include "chrome/test/automation/tab_proxy.h" |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 49 return username + L"/" + password; | 49 return username + L"/" + password; |
| 50 } | 50 } |
| 51 | 51 |
| 52 // Test that "Basic" HTTP authentication works. | 52 // Test that "Basic" HTTP authentication works. |
| 53 TEST_F(LoginPromptTest, TestBasicAuth) { | 53 TEST_F(LoginPromptTest, TestBasicAuth) { |
| 54 scoped_refptr<HTTPTestServer> server = | 54 scoped_refptr<HTTPTestServer> server = |
| 55 HTTPTestServer::CreateServer(kDocRoot, NULL); | 55 HTTPTestServer::CreateServer(kDocRoot, NULL); |
| 56 ASSERT_TRUE(NULL != server.get()); | 56 ASSERT_TRUE(NULL != server.get()); |
| 57 scoped_refptr<TabProxy> tab(GetActiveTab()); | 57 scoped_refptr<TabProxy> tab(GetActiveTab()); |
| 58 ASSERT_TRUE(tab.get()); | 58 ASSERT_TRUE(tab.get()); |
| 59 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"auth-basic"))); | 59 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-basic"))); |
| 60 | 60 |
| 61 EXPECT_TRUE(tab->NeedsAuth()); | 61 EXPECT_TRUE(tab->NeedsAuth()); |
| 62 EXPECT_FALSE(tab->SetAuth(username_basic_, password_bad_)); | 62 EXPECT_FALSE(tab->SetAuth(username_basic_, password_bad_)); |
| 63 EXPECT_TRUE(tab->NeedsAuth()); | 63 EXPECT_TRUE(tab->NeedsAuth()); |
| 64 EXPECT_TRUE(tab->CancelAuth()); | 64 EXPECT_TRUE(tab->CancelAuth()); |
| 65 EXPECT_EQ(L"Denied: wrong password", GetActiveTabTitle()); | 65 EXPECT_EQ(L"Denied: wrong password", GetActiveTabTitle()); |
| 66 | 66 |
| 67 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"auth-basic"))); | 67 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-basic"))); |
| 68 | 68 |
| 69 EXPECT_TRUE(tab->NeedsAuth()); | 69 EXPECT_TRUE(tab->NeedsAuth()); |
| 70 EXPECT_TRUE(tab->SetAuth(username_basic_, password_)); | 70 EXPECT_TRUE(tab->SetAuth(username_basic_, password_)); |
| 71 EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), | 71 EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), |
| 72 GetActiveTabTitle()); | 72 GetActiveTabTitle()); |
| 73 } | 73 } |
| 74 | 74 |
| 75 // Test that "Digest" HTTP authentication works. | 75 // Test that "Digest" HTTP authentication works. |
| 76 TEST_F(LoginPromptTest, TestDigestAuth) { | 76 TEST_F(LoginPromptTest, TestDigestAuth) { |
| 77 scoped_refptr<HTTPTestServer> server = | 77 scoped_refptr<HTTPTestServer> server = |
| 78 HTTPTestServer::CreateServer(kDocRoot, NULL); | 78 HTTPTestServer::CreateServer(kDocRoot, NULL); |
| 79 ASSERT_TRUE(NULL != server.get()); | 79 ASSERT_TRUE(NULL != server.get()); |
| 80 scoped_refptr<TabProxy> tab(GetActiveTab()); | 80 scoped_refptr<TabProxy> tab(GetActiveTab()); |
| 81 ASSERT_TRUE(tab.get()); | 81 ASSERT_TRUE(tab.get()); |
| 82 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"auth-digest"))); | 82 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-digest"))); |
| 83 | 83 |
| 84 EXPECT_TRUE(tab->NeedsAuth()); | 84 EXPECT_TRUE(tab->NeedsAuth()); |
| 85 EXPECT_FALSE(tab->SetAuth(username_digest_, password_bad_)); | 85 EXPECT_FALSE(tab->SetAuth(username_digest_, password_bad_)); |
| 86 EXPECT_TRUE(tab->CancelAuth()); | 86 EXPECT_TRUE(tab->CancelAuth()); |
| 87 EXPECT_EQ(L"Denied: wrong password", GetActiveTabTitle()); | 87 EXPECT_EQ(L"Denied: wrong password", GetActiveTabTitle()); |
| 88 | 88 |
| 89 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"auth-digest"))); | 89 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-digest"))); |
| 90 | 90 |
| 91 EXPECT_TRUE(tab->NeedsAuth()); | 91 EXPECT_TRUE(tab->NeedsAuth()); |
| 92 EXPECT_TRUE(tab->SetAuth(username_digest_, password_)); | 92 EXPECT_TRUE(tab->SetAuth(username_digest_, password_)); |
| 93 EXPECT_EQ(ExpectedTitleFromAuth(username_digest_, password_), | 93 EXPECT_EQ(ExpectedTitleFromAuth(username_digest_, password_), |
| 94 GetActiveTabTitle()); | 94 GetActiveTabTitle()); |
| 95 } | 95 } |
| 96 | 96 |
| 97 // Test that logging in on 2 tabs at once works. | 97 // Test that logging in on 2 tabs at once works. |
| 98 TEST_F(LoginPromptTest, TestTwoAuths) { | 98 TEST_F(LoginPromptTest, TestTwoAuths) { |
| 99 scoped_refptr<HTTPTestServer> server = | 99 scoped_refptr<HTTPTestServer> server = |
| 100 HTTPTestServer::CreateServer(kDocRoot, NULL); | 100 HTTPTestServer::CreateServer(kDocRoot, NULL); |
| 101 ASSERT_TRUE(NULL != server.get()); | 101 ASSERT_TRUE(NULL != server.get()); |
| 102 | 102 |
| 103 scoped_refptr<TabProxy> basic_tab(GetActiveTab()); | 103 scoped_refptr<TabProxy> basic_tab(GetActiveTab()); |
| 104 ASSERT_TRUE(basic_tab.get()); | 104 ASSERT_TRUE(basic_tab.get()); |
| 105 ASSERT_TRUE(basic_tab->NavigateToURL(server->TestServerPageW(L"auth-basic"))); | 105 ASSERT_TRUE(basic_tab->NavigateToURL(server->TestServerPage("auth-basic"))); |
| 106 | 106 |
| 107 AppendTab(GURL(chrome::kAboutBlankURL)); | 107 AppendTab(GURL(chrome::kAboutBlankURL)); |
| 108 scoped_refptr<TabProxy> digest_tab(GetActiveTab()); | 108 scoped_refptr<TabProxy> digest_tab(GetActiveTab()); |
| 109 ASSERT_TRUE(digest_tab.get()); | 109 ASSERT_TRUE(digest_tab.get()); |
| 110 ASSERT_TRUE( | 110 ASSERT_TRUE( |
| 111 digest_tab->NavigateToURL(server->TestServerPageW(L"auth-digest"))); | 111 digest_tab->NavigateToURL(server->TestServerPage("auth-digest"))); |
| 112 | 112 |
| 113 // TODO(devint): http://b/1158262 basic_tab is not active, so this logs in to | 113 // TODO(devint): http://b/1158262 basic_tab is not active, so this logs in to |
| 114 // a page whose tab isn't active, which isn't actually possible for the user | 114 // a page whose tab isn't active, which isn't actually possible for the user |
| 115 // to do. I had a fix for this, but I'm reverting it to see if it makes the | 115 // to do. I had a fix for this, but I'm reverting it to see if it makes the |
| 116 // test less flaky. | 116 // test less flaky. |
| 117 EXPECT_TRUE(basic_tab->NeedsAuth()); | 117 EXPECT_TRUE(basic_tab->NeedsAuth()); |
| 118 EXPECT_TRUE(basic_tab->SetAuth(username_basic_, password_)); | 118 EXPECT_TRUE(basic_tab->SetAuth(username_basic_, password_)); |
| 119 EXPECT_TRUE(digest_tab->NeedsAuth()); | 119 EXPECT_TRUE(digest_tab->NeedsAuth()); |
| 120 EXPECT_TRUE(digest_tab->SetAuth(username_digest_, password_)); | 120 EXPECT_TRUE(digest_tab->SetAuth(username_digest_, password_)); |
| 121 | 121 |
| 122 wstring title; | 122 wstring title; |
| 123 EXPECT_TRUE(basic_tab->GetTabTitle(&title)); | 123 EXPECT_TRUE(basic_tab->GetTabTitle(&title)); |
| 124 EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), title); | 124 EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), title); |
| 125 | 125 |
| 126 EXPECT_TRUE(digest_tab->GetTabTitle(&title)); | 126 EXPECT_TRUE(digest_tab->GetTabTitle(&title)); |
| 127 EXPECT_EQ(ExpectedTitleFromAuth(username_digest_, password_), title); | 127 EXPECT_EQ(ExpectedTitleFromAuth(username_digest_, password_), title); |
| 128 } | 128 } |
| 129 | 129 |
| 130 // Test that cancelling authentication works. | 130 // Test that cancelling authentication works. |
| 131 TEST_F(LoginPromptTest, TestCancelAuth) { | 131 TEST_F(LoginPromptTest, TestCancelAuth) { |
| 132 scoped_refptr<HTTPTestServer> server = | 132 scoped_refptr<HTTPTestServer> server = |
| 133 HTTPTestServer::CreateServer(kDocRoot, NULL); | 133 HTTPTestServer::CreateServer(kDocRoot, NULL); |
| 134 ASSERT_TRUE(NULL != server.get()); | 134 ASSERT_TRUE(NULL != server.get()); |
| 135 scoped_refptr<TabProxy> tab(GetActiveTab()); | 135 scoped_refptr<TabProxy> tab(GetActiveTab()); |
| 136 ASSERT_TRUE(tab.get()); | 136 ASSERT_TRUE(tab.get()); |
| 137 | 137 |
| 138 // First navigate to a test server page so we have something to go back to. | 138 // First navigate to a test server page so we have something to go back to. |
| 139 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"a"))); | 139 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("a"))); |
| 140 | 140 |
| 141 // Navigating while auth is requested is the same as cancelling. | 141 // Navigating while auth is requested is the same as cancelling. |
| 142 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"auth-basic"))); | 142 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-basic"))); |
| 143 EXPECT_TRUE(tab->NeedsAuth()); | 143 EXPECT_TRUE(tab->NeedsAuth()); |
| 144 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"b"))); | 144 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("b"))); |
| 145 EXPECT_FALSE(tab->NeedsAuth()); | 145 EXPECT_FALSE(tab->NeedsAuth()); |
| 146 | 146 |
| 147 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"auth-basic"))); | 147 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-basic"))); |
| 148 EXPECT_TRUE(tab->NeedsAuth()); | 148 EXPECT_TRUE(tab->NeedsAuth()); |
| 149 EXPECT_TRUE(tab->GoBack()); // should bring us back to 'a' | 149 EXPECT_TRUE(tab->GoBack()); // should bring us back to 'a' |
| 150 EXPECT_FALSE(tab->NeedsAuth()); | 150 EXPECT_FALSE(tab->NeedsAuth()); |
| 151 | 151 |
| 152 // Now add a page and go back, so we have something to go forward to. | 152 // Now add a page and go back, so we have something to go forward to. |
| 153 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"c"))); | 153 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("c"))); |
| 154 EXPECT_TRUE(tab->GoBack()); // should bring us back to 'a' | 154 EXPECT_TRUE(tab->GoBack()); // should bring us back to 'a' |
| 155 | 155 |
| 156 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"auth-basic"))); | 156 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-basic"))); |
| 157 EXPECT_TRUE(tab->NeedsAuth()); | 157 EXPECT_TRUE(tab->NeedsAuth()); |
| 158 EXPECT_TRUE(tab->GoForward()); // should bring us to 'c' | 158 EXPECT_TRUE(tab->GoForward()); // should bring us to 'c' |
| 159 EXPECT_FALSE(tab->NeedsAuth()); | 159 EXPECT_FALSE(tab->NeedsAuth()); |
| 160 | 160 |
| 161 // Now test that cancelling works as expected. | 161 // Now test that cancelling works as expected. |
| 162 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"auth-basic"))); | 162 ASSERT_TRUE(tab->NavigateToURL(server->TestServerPage("auth-basic"))); |
| 163 EXPECT_TRUE(tab->NeedsAuth()); | 163 EXPECT_TRUE(tab->NeedsAuth()); |
| 164 EXPECT_TRUE(tab->CancelAuth()); | 164 EXPECT_TRUE(tab->CancelAuth()); |
| 165 EXPECT_FALSE(tab->NeedsAuth()); | 165 EXPECT_FALSE(tab->NeedsAuth()); |
| 166 EXPECT_EQ(L"Denied: no auth", GetActiveTabTitle()); | 166 EXPECT_EQ(L"Denied: no auth", GetActiveTabTitle()); |
| 167 } | 167 } |
| 168 | 168 |
| 169 // If multiple tabs are looking for the same auth, the user should only have to | 169 // If multiple tabs are looking for the same auth, the user should only have to |
| 170 // enter it once (http://crbug.com/8914). | 170 // enter it once (http://crbug.com/8914). |
| 171 TEST_F(LoginPromptTest, SupplyRedundantAuths) { | 171 TEST_F(LoginPromptTest, SupplyRedundantAuths) { |
| 172 scoped_refptr<HTTPTestServer> server = | 172 scoped_refptr<HTTPTestServer> server = |
| 173 HTTPTestServer::CreateServer(kDocRoot, NULL); | 173 HTTPTestServer::CreateServer(kDocRoot, NULL); |
| 174 ASSERT_TRUE(NULL != server.get()); | 174 ASSERT_TRUE(NULL != server.get()); |
| 175 | 175 |
| 176 scoped_refptr<TabProxy> basic_tab1(GetActiveTab()); | 176 scoped_refptr<TabProxy> basic_tab1(GetActiveTab()); |
| 177 ASSERT_TRUE(basic_tab1.get()); | 177 ASSERT_TRUE(basic_tab1.get()); |
| 178 ASSERT_TRUE( | 178 ASSERT_TRUE( |
| 179 basic_tab1->NavigateToURL(server->TestServerPageW(L"auth-basic/1"))); | 179 basic_tab1->NavigateToURL(server->TestServerPage("auth-basic/1"))); |
| 180 EXPECT_TRUE(basic_tab1->NeedsAuth()); | 180 EXPECT_TRUE(basic_tab1->NeedsAuth()); |
| 181 | 181 |
| 182 AppendTab(GURL(chrome::kAboutBlankURL)); | 182 AppendTab(GURL(chrome::kAboutBlankURL)); |
| 183 scoped_refptr<TabProxy> basic_tab2(GetActiveTab()); | 183 scoped_refptr<TabProxy> basic_tab2(GetActiveTab()); |
| 184 ASSERT_TRUE(basic_tab2.get()); | 184 ASSERT_TRUE(basic_tab2.get()); |
| 185 ASSERT_TRUE( | 185 ASSERT_TRUE( |
| 186 basic_tab2->NavigateToURL(server->TestServerPageW(L"auth-basic/2"))); | 186 basic_tab2->NavigateToURL(server->TestServerPage("auth-basic/2"))); |
| 187 EXPECT_TRUE(basic_tab2->NeedsAuth()); | 187 EXPECT_TRUE(basic_tab2->NeedsAuth()); |
| 188 | 188 |
| 189 // Set the auth in only one of the tabs (but wait for the other to load). | 189 // Set the auth in only one of the tabs (but wait for the other to load). |
| 190 int64 last_navigation_time; | 190 int64 last_navigation_time; |
| 191 ASSERT_TRUE(basic_tab2->GetLastNavigationTime(&last_navigation_time)); | 191 ASSERT_TRUE(basic_tab2->GetLastNavigationTime(&last_navigation_time)); |
| 192 EXPECT_TRUE(basic_tab1->SetAuth(username_basic_, password_)); | 192 EXPECT_TRUE(basic_tab1->SetAuth(username_basic_, password_)); |
| 193 EXPECT_TRUE(basic_tab2->WaitForNavigation(last_navigation_time)); | 193 EXPECT_TRUE(basic_tab2->WaitForNavigation(last_navigation_time)); |
| 194 | 194 |
| 195 // Now both tabs have loaded. | 195 // Now both tabs have loaded. |
| 196 wstring title1; | 196 wstring title1; |
| 197 EXPECT_TRUE(basic_tab1->GetTabTitle(&title1)); | 197 EXPECT_TRUE(basic_tab1->GetTabTitle(&title1)); |
| 198 EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), title1); | 198 EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), title1); |
| 199 wstring title2; | 199 wstring title2; |
| 200 EXPECT_TRUE(basic_tab2->GetTabTitle(&title2)); | 200 EXPECT_TRUE(basic_tab2->GetTabTitle(&title2)); |
| 201 EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), title2); | 201 EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), title2); |
| 202 } | 202 } |
| 203 | 203 |
| 204 // If multiple tabs are looking for the same auth, and one is cancelled, the | 204 // If multiple tabs are looking for the same auth, and one is cancelled, the |
| 205 // other should be cancelled as well. | 205 // other should be cancelled as well. |
| 206 TEST_F(LoginPromptTest, CancelRedundantAuths) { | 206 TEST_F(LoginPromptTest, CancelRedundantAuths) { |
| 207 scoped_refptr<HTTPTestServer> server = | 207 scoped_refptr<HTTPTestServer> server = |
| 208 HTTPTestServer::CreateServer(kDocRoot, NULL); | 208 HTTPTestServer::CreateServer(kDocRoot, NULL); |
| 209 ASSERT_TRUE(NULL != server.get()); | 209 ASSERT_TRUE(NULL != server.get()); |
| 210 | 210 |
| 211 scoped_refptr<TabProxy> basic_tab1(GetActiveTab()); | 211 scoped_refptr<TabProxy> basic_tab1(GetActiveTab()); |
| 212 ASSERT_TRUE(basic_tab1.get()); | 212 ASSERT_TRUE(basic_tab1.get()); |
| 213 ASSERT_TRUE( | 213 ASSERT_TRUE( |
| 214 basic_tab1->NavigateToURL(server->TestServerPageW(L"auth-basic/1"))); | 214 basic_tab1->NavigateToURL(server->TestServerPage("auth-basic/1"))); |
| 215 EXPECT_TRUE(basic_tab1->NeedsAuth()); | 215 EXPECT_TRUE(basic_tab1->NeedsAuth()); |
| 216 | 216 |
| 217 AppendTab(GURL(chrome::kAboutBlankURL)); | 217 AppendTab(GURL(chrome::kAboutBlankURL)); |
| 218 scoped_refptr<TabProxy> basic_tab2(GetActiveTab()); | 218 scoped_refptr<TabProxy> basic_tab2(GetActiveTab()); |
| 219 ASSERT_TRUE(basic_tab2.get()); | 219 ASSERT_TRUE(basic_tab2.get()); |
| 220 ASSERT_TRUE( | 220 ASSERT_TRUE( |
| 221 basic_tab2->NavigateToURL(server->TestServerPageW(L"auth-basic/2"))); | 221 basic_tab2->NavigateToURL(server->TestServerPage("auth-basic/2"))); |
| 222 EXPECT_TRUE(basic_tab2->NeedsAuth()); | 222 EXPECT_TRUE(basic_tab2->NeedsAuth()); |
| 223 | 223 |
| 224 // Cancel the auth in only one of the tabs (but wait for the other to load). | 224 // Cancel the auth in only one of the tabs (but wait for the other to load). |
| 225 int64 last_navigation_time; | 225 int64 last_navigation_time; |
| 226 ASSERT_TRUE(basic_tab2->GetLastNavigationTime(&last_navigation_time)); | 226 ASSERT_TRUE(basic_tab2->GetLastNavigationTime(&last_navigation_time)); |
| 227 EXPECT_TRUE(basic_tab1->CancelAuth()); | 227 EXPECT_TRUE(basic_tab1->CancelAuth()); |
| 228 EXPECT_TRUE(basic_tab2->WaitForNavigation(last_navigation_time)); | 228 EXPECT_TRUE(basic_tab2->WaitForNavigation(last_navigation_time)); |
| 229 | 229 |
| 230 // Now both tabs have been denied. | 230 // Now both tabs have been denied. |
| 231 wstring title1; | 231 wstring title1; |
| 232 EXPECT_TRUE(basic_tab1->GetTabTitle(&title1)); | 232 EXPECT_TRUE(basic_tab1->GetTabTitle(&title1)); |
| 233 EXPECT_EQ(L"Denied: no auth", title1); | 233 EXPECT_EQ(L"Denied: no auth", title1); |
| 234 wstring title2; | 234 wstring title2; |
| 235 EXPECT_TRUE(basic_tab2->GetTabTitle(&title2)); | 235 EXPECT_TRUE(basic_tab2->GetTabTitle(&title2)); |
| 236 EXPECT_EQ(L"Denied: no auth", title2); | 236 EXPECT_EQ(L"Denied: no auth", title2); |
| 237 } | 237 } |
| OLD | NEW |