Stop login prompt from showing for prefetches

chrome/browser/ui/login/login_prompt_browsertest.cc

 // Copyright (c) 2011 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 <algorithm>
 #include <list>
 #include <map>
 
 #include "chrome/browser/browser_thread.h"
+#include "chrome/browser/renderer_host/resource_dispatcher_host.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/login/login_prompt.h"
 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
 #include "chrome/common/notification_service.h"
 #include "chrome/test/in_process_browser_test.h"
 #include "chrome/test/ui_test_utils.h"
 #include "net/base/auth.h"
 
 namespace {
 
@@ skipping 135 matching lines @@
 
 typedef WindowedNavigationObserver<NotificationType::AUTH_NEEDED>
     WindowedAuthNeededObserver;
 
 typedef WindowedNavigationObserver<NotificationType::AUTH_CANCELLED>
     WindowedAuthCancelledObserver;
 
 typedef WindowedNavigationObserver<NotificationType::AUTH_SUPPLIED>
     WindowedAuthSuppliedObserver;
 
+const char* kPrefetchAuthPage = "files/login/prefetch.html";
+
 const char* kMultiRealmTestPage = "files/login/multi_realm.html";
 const int   kMultiRealmTestRealmCount = 2;
 const int   kMultiRealmTestResourceCount = 4;
 
 const char* kSingleRealmTestPage = "files/login/single_realm.html";
 const int   kSingleRealmTestResourceCount = 6;
 
+// Confirm that <link rel="prefetch"> targetting an auth required
+// resource does not provide a login dialog.  These types of requests
+// should instead just cancel the auth.
+
+// Unfortunately, this test doesn't assert on anything for its
+// correctness.  Instead, it relies on the auth dialog blocking the
+// browser, and triggering a timeout to cause failure when the
+// prefetch resource requires authorization.
+IN_PROC_BROWSER_TEST_F(LoginPromptBrowserTest, PrefetchAuthCancels) {

[cbentzel, 2011/01/20 17:59:45]

Should this test also add a LoginPromptBrowserTestObserver and ensure that
observer.handlers_.empty() after load_stop_waiter.Wait() completes? 

It is kind of implicit that there are no authentication prompts due to the load
completing, but it's probably worth explicitly adding that test.

[asanka (google), 2011/01/20 18:16:29]

An alternative would be to wait for both AUTH_NEEDED and LOAD_STOP, but is
probably overkill.

[cbentzel, 2011/01/20 18:19:20]

In this case we do not expect an AUTH_NEEDED, so that won't work.

[gavinp, 2011/01/20 18:36:15]

Added the assertion for the obvserver handlers being empty; this is a good belts
& suspenders move.

+  ASSERT_TRUE(test_server()->Start());
+
+  GURL test_page = test_server()->GetURL(kPrefetchAuthPage);
+
+  const bool was_prefetch_enabled =

[cbentzel, 2011/01/20 17:59:45]

Would it make sense to add a simple RAII class to do this and ensure that
prefetch enabled is returned to original state at the end of the test?

Push back if it's overkill. 

[gavinp, 2011/01/20 18:36:15]

Done.

+      ResourceDispatcherHost::is_prefetch_enabled();
+  ResourceDispatcherHost::set_is_prefetch_enabled(true);
+
+  TabContentsWrapper* contents =
+      browser()->GetSelectedTabContentsWrapper();
+  ASSERT_TRUE(contents);
+  NavigationController* controller = &contents->controller();
+
+  WindowedLoadStopObserver load_stop_waiter(controller);
+  browser()->OpenURL(test_page, GURL(), CURRENT_TAB, PageTransition::TYPED);
+
+  load_stop_waiter.Wait();
+
+  ResourceDispatcherHost::set_is_prefetch_enabled(was_prefetch_enabled);
+  EXPECT_TRUE(test_server()->Stop());
+}
+
 // Test handling of resources that require authentication even though
 // the page they are included on doesn't.  In this case we should only
 // present the minimal number of prompts necessary for successfully
 // displaying the page.  First we check whether cancelling works as
 // expected.
 IN_PROC_BROWSER_TEST_F(LoginPromptBrowserTest, MultipleRealmCancellation) {
   ASSERT_TRUE(test_server()->Start());
   GURL test_page = test_server()->GetURL(kMultiRealmTestPage);
 
   TabContentsWrapper* contents =
@@ skipping 167 matching lines @@
   EXPECT_EQ(1, n_handlers);
   EXPECT_LT(0, observer.auth_needed_count_);
   EXPECT_EQ(0, observer.auth_cancelled_count_);
   EXPECT_EQ(observer.auth_needed_count_, observer.auth_supplied_count_);
   LOG(INFO) << "Waiting for LOAD_STOP";
   load_stop_waiter.Wait();
   EXPECT_TRUE(test_server()->Stop());
   LOG(INFO) << "Done with test";
 }
 }  // namespace