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/url_fixer_upper.h" | 7 #include "chrome/browser/url_fixer_upper.h" |
8 #include "chrome/common/chrome_switches.h" | 8 #include "chrome/common/chrome_switches.h" |
9 #include "chrome/test/automation/tab_proxy.h" | 9 #include "chrome/test/automation/tab_proxy.h" |
10 #include "chrome/test/automation/browser_proxy.h" | 10 #include "chrome/test/automation/browser_proxy.h" |
(...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
55 | 55 |
56 wstring ExpectedTitleFromAuth(wstring username, wstring password) { | 56 wstring ExpectedTitleFromAuth(wstring username, wstring password) { |
57 // The TestServer sets the title to username/password on successful login. | 57 // The TestServer sets the title to username/password on successful login. |
58 return username + L"/" + password; | 58 return username + L"/" + password; |
59 } | 59 } |
60 | 60 |
61 } // namespace | 61 } // namespace |
62 | 62 |
63 // Test that "Basic" HTTP authentication works. | 63 // Test that "Basic" HTTP authentication works. |
64 TEST_F(LoginPromptTest, TestBasicAuth) { | 64 TEST_F(LoginPromptTest, TestBasicAuth) { |
65 scoped_refptr<HTTPTestServer> server = | 65 TestServer server(kDocRoot); |
66 HTTPTestServer::CreateServer(kDocRoot); | |
67 ASSERT_TRUE(NULL != server.get()); | |
68 ::scoped_ptr<TabProxy> tab(GetActiveTabProxy()); | 66 ::scoped_ptr<TabProxy> tab(GetActiveTabProxy()); |
69 NavigateTab(tab.get(), server->TestServerPageW(L"auth-basic")); | 67 NavigateTab(tab.get(), server.TestServerPageW(L"auth-basic")); |
70 | 68 |
71 EXPECT_TRUE(tab->NeedsAuth()); | 69 EXPECT_TRUE(tab->NeedsAuth()); |
72 EXPECT_FALSE(tab->SetAuth(username_basic_, password_bad_)); | 70 EXPECT_FALSE(tab->SetAuth(username_basic_, password_bad_)); |
73 EXPECT_TRUE(tab->NeedsAuth()); | 71 EXPECT_TRUE(tab->NeedsAuth()); |
74 EXPECT_TRUE(tab->CancelAuth()); | 72 EXPECT_TRUE(tab->CancelAuth()); |
75 EXPECT_EQ(L"Denied: wrong password", GetActiveTabTitle()); | 73 EXPECT_EQ(L"Denied: wrong password", GetActiveTabTitle()); |
76 | 74 |
77 NavigateTab(tab.get(), server->TestServerPageW(L"auth-basic")); | 75 NavigateTab(tab.get(), server.TestServerPageW(L"auth-basic")); |
78 | 76 |
79 EXPECT_TRUE(tab->NeedsAuth()); | 77 EXPECT_TRUE(tab->NeedsAuth()); |
80 EXPECT_TRUE(tab->SetAuth(username_basic_, password_)); | 78 EXPECT_TRUE(tab->SetAuth(username_basic_, password_)); |
81 EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), | 79 EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), |
82 GetActiveTabTitle()); | 80 GetActiveTabTitle()); |
83 } | 81 } |
84 | 82 |
85 // Test that "Digest" HTTP authentication works. | 83 // Test that "Digest" HTTP authentication works. |
86 TEST_F(LoginPromptTest, TestDigestAuth) { | 84 TEST_F(LoginPromptTest, TestDigestAuth) { |
87 scoped_refptr<HTTPTestServer> server = | 85 TestServer server(kDocRoot); |
88 HTTPTestServer::CreateServer(kDocRoot); | |
89 ASSERT_TRUE(NULL != server.get()); | |
90 ::scoped_ptr<TabProxy> tab(GetActiveTabProxy()); | 86 ::scoped_ptr<TabProxy> tab(GetActiveTabProxy()); |
91 NavigateTab(tab.get(), server->TestServerPageW(L"auth-digest")); | 87 NavigateTab(tab.get(), server.TestServerPageW(L"auth-digest")); |
92 | 88 |
93 EXPECT_TRUE(tab->NeedsAuth()); | 89 EXPECT_TRUE(tab->NeedsAuth()); |
94 EXPECT_FALSE(tab->SetAuth(username_digest_, password_bad_)); | 90 EXPECT_FALSE(tab->SetAuth(username_digest_, password_bad_)); |
95 EXPECT_TRUE(tab->CancelAuth()); | 91 EXPECT_TRUE(tab->CancelAuth()); |
96 EXPECT_EQ(L"Denied: wrong password", GetActiveTabTitle()); | 92 EXPECT_EQ(L"Denied: wrong password", GetActiveTabTitle()); |
97 | 93 |
98 NavigateTab(tab.get(), server->TestServerPageW(L"auth-digest")); | 94 NavigateTab(tab.get(), server.TestServerPageW(L"auth-digest")); |
99 | 95 |
100 EXPECT_TRUE(tab->NeedsAuth()); | 96 EXPECT_TRUE(tab->NeedsAuth()); |
101 EXPECT_TRUE(tab->SetAuth(username_digest_, password_)); | 97 EXPECT_TRUE(tab->SetAuth(username_digest_, password_)); |
102 EXPECT_EQ(ExpectedTitleFromAuth(username_digest_, password_), | 98 EXPECT_EQ(ExpectedTitleFromAuth(username_digest_, password_), |
103 GetActiveTabTitle()); | 99 GetActiveTabTitle()); |
104 } | 100 } |
105 | 101 |
106 // Test that logging in on 2 tabs at once works. | 102 // Test that logging in on 2 tabs at once works. |
107 TEST_F(LoginPromptTest, TestTwoAuths) { | 103 TEST_F(LoginPromptTest, TestTwoAuths) { |
108 scoped_refptr<HTTPTestServer> server = | 104 TestServer server(kDocRoot); |
109 HTTPTestServer::CreateServer(kDocRoot); | |
110 ASSERT_TRUE(NULL != server.get()); | |
111 | 105 |
112 ::scoped_ptr<TabProxy> basic_tab(GetActiveTabProxy()); | 106 ::scoped_ptr<TabProxy> basic_tab(GetActiveTabProxy()); |
113 NavigateTab(basic_tab.get(), server->TestServerPageW(L"auth-basic")); | 107 NavigateTab(basic_tab.get(), server.TestServerPageW(L"auth-basic")); |
114 | 108 |
115 AppendTab(GURL("about:blank")); | 109 AppendTab(GURL("about:blank")); |
116 ::scoped_ptr<TabProxy> digest_tab(GetActiveTabProxy()); | 110 ::scoped_ptr<TabProxy> digest_tab(GetActiveTabProxy()); |
117 NavigateTab(digest_tab.get(), server->TestServerPageW(L"auth-digest")); | 111 NavigateTab(digest_tab.get(), server.TestServerPageW(L"auth-digest")); |
118 | 112 |
119 // 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 |
120 // 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 |
121 // 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 |
122 // test less flaky. | 116 // test less flaky. |
123 EXPECT_TRUE(basic_tab->NeedsAuth()); | 117 EXPECT_TRUE(basic_tab->NeedsAuth()); |
124 EXPECT_TRUE(basic_tab->SetAuth(username_basic_, password_)); | 118 EXPECT_TRUE(basic_tab->SetAuth(username_basic_, password_)); |
125 EXPECT_TRUE(digest_tab->NeedsAuth()); | 119 EXPECT_TRUE(digest_tab->NeedsAuth()); |
126 EXPECT_TRUE(digest_tab->SetAuth(username_digest_, password_)); | 120 EXPECT_TRUE(digest_tab->SetAuth(username_digest_, password_)); |
127 | 121 |
128 wstring title; | 122 wstring title; |
129 EXPECT_TRUE(basic_tab->GetTabTitle(&title)); | 123 EXPECT_TRUE(basic_tab->GetTabTitle(&title)); |
130 EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), title); | 124 EXPECT_EQ(ExpectedTitleFromAuth(username_basic_, password_), title); |
131 | 125 |
132 EXPECT_TRUE(digest_tab->GetTabTitle(&title)); | 126 EXPECT_TRUE(digest_tab->GetTabTitle(&title)); |
133 EXPECT_EQ(ExpectedTitleFromAuth(username_digest_, password_), title); | 127 EXPECT_EQ(ExpectedTitleFromAuth(username_digest_, password_), title); |
134 } | 128 } |
135 | 129 |
136 // Test that cancelling authentication works. | 130 // Test that cancelling authentication works. |
137 TEST_F(LoginPromptTest, TestCancelAuth) { | 131 TEST_F(LoginPromptTest, TestCancelAuth) { |
138 scoped_refptr<HTTPTestServer> server = | 132 TestServer server(kDocRoot); |
139 HTTPTestServer::CreateServer(kDocRoot); | |
140 ASSERT_TRUE(NULL != server.get()); | |
141 ::scoped_ptr<TabProxy> tab(GetActiveTabProxy()); | 133 ::scoped_ptr<TabProxy> tab(GetActiveTabProxy()); |
142 | 134 |
143 // First navigate to a test server page so we have something to go back to. | 135 // First navigate to a test server page so we have something to go back to. |
144 EXPECT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"a"))); | 136 EXPECT_TRUE(tab->NavigateToURL(server.TestServerPageW(L"a"))); |
145 | 137 |
146 // Navigating while auth is requested is the same as cancelling. | 138 // Navigating while auth is requested is the same as cancelling. |
147 NavigateTab(tab.get(), server->TestServerPageW(L"auth-basic")); | 139 NavigateTab(tab.get(), server.TestServerPageW(L"auth-basic")); |
148 EXPECT_TRUE(tab->NeedsAuth()); | 140 EXPECT_TRUE(tab->NeedsAuth()); |
149 EXPECT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"b"))); | 141 EXPECT_TRUE(tab->NavigateToURL(server.TestServerPageW(L"b"))); |
150 EXPECT_FALSE(tab->NeedsAuth()); | 142 EXPECT_FALSE(tab->NeedsAuth()); |
151 | 143 |
152 NavigateTab(tab.get(), server->TestServerPageW(L"auth-basic")); | 144 NavigateTab(tab.get(), server.TestServerPageW(L"auth-basic")); |
153 EXPECT_TRUE(tab->NeedsAuth()); | 145 EXPECT_TRUE(tab->NeedsAuth()); |
154 EXPECT_TRUE(tab->GoBack()); // should bring us back to 'a' | 146 EXPECT_TRUE(tab->GoBack()); // should bring us back to 'a' |
155 EXPECT_FALSE(tab->NeedsAuth()); | 147 EXPECT_FALSE(tab->NeedsAuth()); |
156 | 148 |
157 // Now add a page and go back, so we have something to go forward to. | 149 // Now add a page and go back, so we have something to go forward to. |
158 EXPECT_TRUE(tab->NavigateToURL(server->TestServerPageW(L"c"))); | 150 EXPECT_TRUE(tab->NavigateToURL(server.TestServerPageW(L"c"))); |
159 EXPECT_TRUE(tab->GoBack()); // should bring us back to 'a' | 151 EXPECT_TRUE(tab->GoBack()); // should bring us back to 'a' |
160 | 152 |
161 NavigateTab(tab.get(), server->TestServerPageW(L"auth-basic")); | 153 NavigateTab(tab.get(), server.TestServerPageW(L"auth-basic")); |
162 EXPECT_TRUE(tab->NeedsAuth()); | 154 EXPECT_TRUE(tab->NeedsAuth()); |
163 EXPECT_TRUE(tab->GoForward()); // should bring us to 'c' | 155 EXPECT_TRUE(tab->GoForward()); // should bring us to 'c' |
164 EXPECT_FALSE(tab->NeedsAuth()); | 156 EXPECT_FALSE(tab->NeedsAuth()); |
165 | 157 |
166 // Now test that cancelling works as expected. | 158 // Now test that cancelling works as expected. |
167 NavigateTab(tab.get(), server->TestServerPageW(L"auth-basic")); | 159 NavigateTab(tab.get(), server.TestServerPageW(L"auth-basic")); |
168 EXPECT_TRUE(tab->NeedsAuth()); | 160 EXPECT_TRUE(tab->NeedsAuth()); |
169 EXPECT_TRUE(tab->CancelAuth()); | 161 EXPECT_TRUE(tab->CancelAuth()); |
170 EXPECT_FALSE(tab->NeedsAuth()); | 162 EXPECT_FALSE(tab->NeedsAuth()); |
171 EXPECT_EQ(L"Denied: no auth", GetActiveTabTitle()); | 163 EXPECT_EQ(L"Denied: no auth", GetActiveTabTitle()); |
172 } | 164 } |
173 | 165 |
OLD | NEW |