Add test case to see if HTTP auth interstitials work nicely with other interstitials

chrome/browser/ui/login/login_prompt_browsertest.cc

 // Copyright (c) 2012 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 "base/metrics/field_trial.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/prerender/prerender_manager.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_commands.h"
 #include "chrome/browser/ui/login/login_prompt.h"
 #include "chrome/browser/ui/login/login_prompt_test_utils.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/interstitial_page.h"
 #include "content/public/browser/notification_details.h"
 #include "content/public/browser/notification_source.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/test/browser_test_utils.h"
 #include "content/public/test/test_navigation_observer.h"
 #include "net/base/auth.h"
 #include "net/dns/mock_host_resolver.h"
+#include "net/test/spawned_test_server/spawned_test_server.h"
 
 using content::NavigationController;
 using content::OpenURLParams;
 using content::Referrer;
 
 namespace {
 
 class LoginPromptBrowserTest : public InProcessBrowserTest {
  public:
   LoginPromptBrowserTest()
@@ skipping 1204 matching lines @@
   host_resolver()->AddRule("www.a.com", "127.0.0.1");
   ASSERT_TRUE(test_server()->Start());
 
   const char* kTestPage = "files/login/cross_origin.html";
   GURL test_page = test_server()->GetURL(kTestPage);
   ASSERT_EQ("127.0.0.1", test_page.host());
   std::string auth_host("www.a.com");
   TestCrossOriginPrompt(test_page, auth_host);
 }
 
+IN_PROC_BROWSER_TEST_F(LoginPromptBrowserTest,
+                       LoginInterstitialShouldReplaceExistingInterstitial) {
+  net::SpawnedTestServer https_server(
+    net::SpawnedTestServer::TYPE_HTTPS,

[msw, 2014/08/19 01:16:39]

nit: indent four spaces for wrapped function arguments.

[meacer, 2014/08/20 20:13:45]

Done.

+    net::SpawnedTestServer::SSLOptions(
+        net::SpawnedTestServer::SSLOptions::CERT_EXPIRED),
+    base::FilePath()
+  );

[msw, 2014/08/19 01:16:39]

nit: move this to the line above.

[meacer, 2014/08/20 20:13:45]

Done.

+  ASSERT_TRUE(https_server.Start());
+
+  content::WebContents* contents =
+      browser()->tab_strip_model()->GetActiveWebContents();
+  NavigationController* controller = &contents->GetController();
+  LoginPromptBrowserTestObserver observer;
+
+  observer.Register(content::Source<NavigationController>(controller));
+
+  // Load a page which triggers an SSL interstitial. Proceeding through it
+  // should show the login page with the blank interstitial.
+  {
+    GURL test_page = https_server.GetURL(kAuthBasicPage);
+    ASSERT_EQ("127.0.0.1", test_page.host());
+
+    WindowedAuthNeededObserver auth_needed_waiter(controller);
+    browser()->OpenURL(OpenURLParams(
+        test_page, Referrer(), CURRENT_TAB, content::PAGE_TRANSITION_TYPED,
+        false));
+    ASSERT_EQ("127.0.0.1", contents->GetURL().host());
+    WaitForInterstitialAttach(contents);
+
+    // An overrideable SSL interstitial is now being displayed. Proceed through
+    // the interstitial to see the login prompt.
+    contents->GetInterstitialPage()->Proceed();
+    auth_needed_waiter.Wait();
+    ASSERT_EQ(1u, observer.handlers().size());
+    WaitForInterstitialAttach(contents);
+
+    // The omnibox should show the correct origin while the login prompt is
+    // being displayed.
+    EXPECT_EQ("127.0.0.1", contents->GetVisibleURL().host());
+    EXPECT_TRUE(contents->ShowingInterstitialPage());
+
+    // Cancelling the login prompt should detach the interstitial while keeping
+    // the correct origin.
+    LoginHandler* handler = *observer.handlers().begin();
+    scoped_refptr<content::MessageLoopRunner> loop_runner(
+        new content::MessageLoopRunner);
+    InterstitialObserver interstitial_observer(contents,
+                                               base::Closure(),
+                                               loop_runner->QuitClosure());
+    handler->CancelAuth();
+    if (content::InterstitialPage::GetInterstitialPage(contents))
+      loop_runner->Run();
+    EXPECT_EQ("127.0.0.1", contents->GetVisibleURL().host());
+    EXPECT_FALSE(contents->ShowingInterstitialPage());
+  }
+}
+
 }  // namespace