PasswordManager browser test: simplify checking for password prompt

chrome/browser/password_manager/password_manager_browsertest.cc

 // Copyright 2013 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 <string>
 
 #include "base/command_line.h"
 #include "base/metrics/histogram_samples.h"
 #include "base/metrics/statistics_recorder.h"
 #include "base/run_loop.h"
@@ skipping 37 matching lines @@
 #include "ui/events/keycodes/keyboard_codes.h"
 #include "ui/gfx/geometry/point.h"
 
 
 // NavigationObserver ---------------------------------------------------------
 
 namespace {
 
 // Observer that waits for navigation to complete and for the password infobar
 // to be shown.
-class NavigationObserver : public content::WebContentsObserver,
-                           public infobars::InfoBarManager::Observer {
+class NavigationObserver : public content::WebContentsObserver {
  public:
   explicit NavigationObserver(content::WebContents* web_contents)
       : content::WebContentsObserver(web_contents),
-        message_loop_runner_(new content::MessageLoopRunner),
-        infobar_shown_(false),
-        infobar_removed_(false),
-        should_automatically_accept_infobar_(true),
-        infobar_service_(InfoBarService::FromWebContents(web_contents)) {
-    infobar_service_->AddObserver(this);
-  }
+        message_loop_runner_(new content::MessageLoopRunner) {}
 
-  virtual ~NavigationObserver() {
-    if (infobar_service_)
-      infobar_service_->RemoveObserver(this);
-  }
+  virtual ~NavigationObserver() {}
 
   // Normally Wait() will not return until a main frame navigation occurs.
   // If a path is set, Wait() will return after this path has been seen,
   // regardless of the frame that navigated. Useful for multi-frame pages.
   void SetPathToWaitFor(const std::string& path) {
     wait_for_path_ = path;
   }
 
   // content::WebContentsObserver:
   virtual void DidFinishLoad(content::RenderFrameHost* render_frame_host,
                              const GURL& validated_url) OVERRIDE {
     if (!wait_for_path_.empty()) {
       if (validated_url.path() == wait_for_path_)
         message_loop_runner_->Quit();
     } else if (!render_frame_host->GetParent()) {
       message_loop_runner_->Quit();
     }
   }
 
-  bool infobar_shown() const { return infobar_shown_; }
-  bool infobar_removed() const { return infobar_removed_; }
-
-  void disable_should_automatically_accept_infobar() {
-    should_automatically_accept_infobar_ = false;
-  }
-
   void Wait() {
     message_loop_runner_->Run();
   }
 
  private:
+
+  std::string wait_for_path_;
+  scoped_refptr<content::MessageLoopRunner> message_loop_runner_;
+
+  DISALLOW_COPY_AND_ASSIGN(NavigationObserver);
+};
+
+// Observes save password prompt (bubble or infobar) for a specified

[engedy, 2014/07/11 18:07:26]

nits: Observes *the* save password prompt (bubble or infobar) for a specified
WebContents, keeps track *of whether or not* it is currently shown, and allows
accepting *saving* (?) the password through it.

[vabr (Chromium), 2014/07/11 18:48:01]

Done.

+// WebContents, keeps track if it is currently shown, and allows accepting
+// passwords through it.
+class PromptObserver {
+ public:
+  virtual ~PromptObserver() {}
+
+  // Check if the prompt is being currently shown.

[engedy, 2014/07/11 18:07:27]

nits: Please use third person in comments. (3 verbs.)

[vabr (Chromium), 2014/07/11 18:48:02]

Done.

+  virtual bool IsPromptShown() const = 0;

[engedy, 2014/07/11 18:07:26]

nit: IsShowingPrompt

[vabr (Chromium), 2014/07/11 18:48:02]

Done.

+
+  // Expecting that the prompt is shown, save the password. Check that the
+  // prompt is no more visible.

[engedy, 2014/07/11 18:07:27]

nit: no longer visible afterwards.

[vabr (Chromium), 2014/07/11 18:48:02]

Done.

+  void Accept() const {
+    EXPECT_TRUE(IsPromptShown());
+    AcceptImpl();
+  }
+
+  // Chooses the right implementation of PromptObserver and creates an instance
+  // of it.
+  static scoped_ptr<PromptObserver> Create(content::WebContents* web_contents);

[engedy, 2014/07/11 18:07:26]

Nice!

[vabr (Chromium), 2014/07/11 18:48:02]

Acknowledged.
(And thanks! :))

+
+ protected:
+  PromptObserver() {}
+
+  // Accepts the password. Can assume that the prompt is currently shown, makes

[engedy, 2014/07/11 18:07:26]

Nit: the implementation can assume ... but is required to verify that ...

[vabr (Chromium), 2014/07/11 18:48:02]

Done.

+  // sure that the prompt is eventually closed.
+  virtual void AcceptImpl() const = 0;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(PromptObserver);
+};
+
+class InfoBarObserver : public PromptObserver,
+                        public infobars::InfoBarManager::Observer {
+ public:
+  explicit InfoBarObserver(content::WebContents* web_contents)
+      : infobar_is_shown_(false),

[engedy, 2014/07/11 18:07:27]

nit: infobar_is_showing_ or is_infobar_showing_.

[vabr (Chromium), 2014/07/11 18:48:01]

Done, with a change to infobar_is_being_shown_, which was confirmed by Balázs as
acceptable during an offline chat.

+        infobar_service_(InfoBarService::FromWebContents(web_contents)) {
+    infobar_service_->AddObserver(this);
+  }
+
+  virtual ~InfoBarObserver() {
+    if (infobar_service_)
+      infobar_service_->RemoveObserver(this);
+  }
+
+ private:
+  // PromptObserver:
+  virtual bool IsPromptShown() const OVERRIDE { return infobar_is_shown_; }
+
+  virtual void AcceptImpl() const OVERRIDE {
+    EXPECT_EQ(1u, infobar_service_->infobar_count());

[engedy, 2014/07/11 18:07:26]

I would assert here instead. (Or your if-solution, if the test will not hang
when no bubble is accepted.)

[vabr (Chromium), 2014/07/11 18:48:01]

Done.
I don't think it would hang because of this. Usually, it would check the
password store, and complain about the credentials missing.

+    // ConfirmInfoBarDelegate::Accept returning true means the infobar is
+    // immediately closed. Checking the return value is preferred to testing
+    // IsPromptShown() here, for it avoids the delay until the closing
+    // notification is received.
+    EXPECT_TRUE(infobar_service_->infobar_at(0)
+                    ->delegate()
+                    ->AsConfirmInfoBarDelegate()
+                    ->Accept());
+  }
+
   // infobars::InfoBarManager::Observer:
   virtual void OnInfoBarAdded(infobars::InfoBar* infobar) OVERRIDE {
-    if (should_automatically_accept_infobar_) {
-      infobar_service_->infobar_at(0)->delegate()->
-          AsConfirmInfoBarDelegate()->Accept();
-    }
-    infobar_shown_ = true;
+    infobar_is_shown_ = true;
   }
 
   virtual void OnInfoBarRemoved(infobars::InfoBar* infobar,
                                 bool animate) OVERRIDE {
-    infobar_removed_ = true;
+    infobar_is_shown_ = false;
   }
 
   virtual void OnManagerShuttingDown(
       infobars::InfoBarManager* manager) OVERRIDE {
     ASSERT_EQ(infobar_service_, manager);
     infobar_service_->RemoveObserver(this);
     infobar_service_ = NULL;
   }
 
-  std::string wait_for_path_;
-  scoped_refptr<content::MessageLoopRunner> message_loop_runner_;
-  bool infobar_shown_;
-  bool infobar_removed_;
-  // If |should_automatically_accept_infobar_| is true, then whenever the test
-  // sees an infobar added, it will click its accepting button. Default = true.
-  bool should_automatically_accept_infobar_;
+  bool infobar_is_shown_;
   InfoBarService* infobar_service_;
 
-  DISALLOW_COPY_AND_ASSIGN(NavigationObserver);
+  DISALLOW_COPY_AND_ASSIGN(InfoBarObserver);
 };
 
+class BubbleObserver : public PromptObserver {
+ public:
+  explicit BubbleObserver(content::WebContents* web_contents)
+      : ui_controller_(
+            ManagePasswordsUIController::FromWebContents(web_contents)) {}
+
+  virtual ~BubbleObserver() {}
+
+ private:
+  // PromptObserver:
+  virtual bool IsPromptShown() const OVERRIDE {
+    return ui_controller_->PasswordPendingUserDecision();
+  }
+
+  virtual void AcceptImpl() const OVERRIDE {
+    ui_controller_->SavePassword();
+    EXPECT_FALSE(IsPromptShown());
+  }
+
+  ManagePasswordsUIController* const ui_controller_;
+
+  DISALLOW_COPY_AND_ASSIGN(BubbleObserver);
+};
+
+// static
+scoped_ptr<PromptObserver> PromptObserver::Create(
+    content::WebContents* web_contents) {
+  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
+    return scoped_ptr<PromptObserver>(new BubbleObserver(web_contents));
+  } else {
+    return scoped_ptr<PromptObserver>(new InfoBarObserver(web_contents));
+  }
+}
+
 // Handles |request| to "/basic_auth". If "Authorization" header is present,
 // responds with a non-empty HTTP 200 page (regardless of its value). Otherwise
 // serves a Basic Auth challenge.
 scoped_ptr<net::test_server::HttpResponse> HandleTestAuthRequest(
     const net::test_server::HttpRequest& request) {
   if (!StartsWithASCII(request.relative_url, "/basic_auth", true))
     return scoped_ptr<net::test_server::HttpResponse>();
 
   if (ContainsKey(request.headers, "Authorization")) {
     scoped_ptr<net::test_server::BasicHttpResponse> http_response(
@@ skipping 40 matching lines @@
 
  protected:
   content::WebContents* WebContents() {
     return browser()->tab_strip_model()->GetActiveWebContents();
   }
 
   content::RenderViewHost* RenderViewHost() {
     return WebContents()->GetRenderViewHost();
   }
 
-  ManagePasswordsUIController* ui_controller() {
-    return ManagePasswordsUIController::FromWebContents(WebContents());
-  }
-
   // Wrapper around ui_test_utils::NavigateToURL that waits until
   // DidFinishLoad() fires. Normally this function returns after
   // DidStopLoading(), which caused flakiness as the NavigationObserver
   // would sometimes see the DidFinishLoad event from a previous navigation and
   // return immediately.
   void NavigateToFile(const std::string& path) {
     if (!embedded_test_server()->Started())
       ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady());
 
     ASSERT_FALSE(CommandLine::ForCurrentProcess()->HasSwitch(
@@ skipping 84 matching lines @@
 }
 
 // Actual tests ---------------------------------------------------------------
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        PromptForNormalSubmit) {
   NavigateToFile("/password/password_form.html");
 
   // Fill a form and submit through a <input type="submit"> button. Nothing
   // special.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('username_field').value = 'temp';"
       "document.getElementById('password_field').value = 'random';"
       "document.getElementById('input_submit_button').click()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        PromptForSubmitWithInPageNavigation) {
   NavigateToFile("/password/password_navigate_before_submit.html");
 
   // Fill a form and submit through a <input type="submit"> button. Nothing
   // special. The form does an in-page navigation before submitting.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('username_field').value = 'temp';"
       "document.getElementById('password_field').value = 'random';"
       "document.getElementById('input_submit_button').click()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        LoginSuccessWithUnrelatedForm) {
   // Log in, see a form on the landing page. That form is not related to the
   // login form (=has a different action), so we should offer saving the
   // password.
   NavigateToFile("/password/password_form.html");
 
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('username_unrelated').value = 'temp';"
       "document.getElementById('password_unrelated').value = 'random';"
       "document.getElementById('submit_unrelated').click()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest, LoginFailed) {
   // Log in, see a form on the landing page. That form is not related to the
   // login form (=has a different action), so we should offer saving the
   // password.
   NavigateToFile("/password/password_form.html");
 
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('username_failed').value = 'temp';"
       "document.getElementById('password_failed').value = 'random';"
       "document.getElementById('submit_failed').click()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_FALSE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_FALSE(observer.infobar_shown());
-  }
+  EXPECT_FALSE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest, Redirects) {
   NavigateToFile("/password/password_form.html");
 
   // Fill a form and submit through a <input type="submit"> button. The form
   // points to a redirection page.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('username_redirect').value = 'temp';"
       "document.getElementById('password_redirect').value = 'random';"
       "document.getElementById('submit_redirect').click()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
-  observer.disable_should_automatically_accept_infobar();
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 
   // The redirection page now redirects via Javascript. We check that the
   // infobar stays.
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(),
                                      "window.location.href = 'done.html';"));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_FALSE(observer.infobar_removed());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        PromptForSubmitUsingJavaScript) {
   NavigateToFile("/password/password_form.html");
 
   // Fill a form and submit using <button> that calls submit() on the form.
   // This should work regardless of the type of element, as long as submit() is
   // called.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('username_field').value = 'temp';"
       "document.getElementById('password_field').value = 'random';"
       "document.getElementById('submit_button').click()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
 // Flaky: crbug.com/301547, observed on win and mac. Probably happens on all
 // platforms.
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        DISABLED_PromptForDynamicForm) {
   NavigateToFile("/password/dynamic_password_form.html");
 
   // Fill the dynamic password form and submit.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('create_form_button').click();"
       "window.setTimeout(function() {"
       "  document.dynamic_form.username.value = 'tempro';"
       "  document.dynamic_form.password.value = 'random';"
       "  document.dynamic_form.submit();"
       "}, 0)";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest, NoPromptForNavigation) {
   NavigateToFile("/password/password_form.html");
 
   // Don't fill the password form, just navigate away. Shouldn't prompt.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(),
                                      "window.location.href = 'done.html';"));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_FALSE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_FALSE(observer.infobar_shown());
-  }
+  EXPECT_FALSE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        NoPromptForSubFrameNavigation) {
   NavigateToFile("/password/multi_frames.html");
 
   // If you are filling out a password form in one frame and a different frame
   // navigates, this should not trigger the infobar.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   observer.SetPathToWaitFor("/password/done.html");
   std::string fill =
       "var first_frame = document.getElementById('first_frame');"
       "var frame_doc = first_frame.contentDocument;"
       "frame_doc.getElementById('username_field').value = 'temp';"
       "frame_doc.getElementById('password_field').value = 'random';";
   std::string navigate_frame =
       "var second_iframe = document.getElementById('second_frame');"
       "second_iframe.contentWindow.location.href = 'done.html';";
 
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill));
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), navigate_frame));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_FALSE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_FALSE(observer.infobar_shown());
-  }
+  EXPECT_FALSE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        PromptAfterSubmitWithSubFrameNavigation) {
   NavigateToFile("/password/multi_frames.html");
 
   // Make sure that we prompt to save password even if a sub-frame navigation
   // happens first.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   observer.SetPathToWaitFor("/password/done.html");
   std::string navigate_frame =
       "var second_iframe = document.getElementById('second_frame');"
       "second_iframe.contentWindow.location.href = 'other.html';";
   std::string fill_and_submit =
       "var first_frame = document.getElementById('first_frame');"
       "var frame_doc = first_frame.contentDocument;"
       "frame_doc.getElementById('username_field').value = 'temp';"
       "frame_doc.getElementById('password_field').value = 'random';"
       "frame_doc.getElementById('input_submit_button').click();";
 
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), navigate_frame));
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(
     PasswordManagerBrowserTest,
     NoPromptForFailedLoginFromMainFrameWithMultiFramesInPage) {
   NavigateToFile("/password/multi_frames.html");
 
   // Make sure that we don't prompt to save the password for a failed login
   // from the main frame with multiple frames in the same page.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('username_failed').value = 'temp';"
       "document.getElementById('password_failed').value = 'random';"
       "document.getElementById('submit_failed').click();";
 
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  EXPECT_FALSE(observer.infobar_shown());
+  EXPECT_FALSE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(
     PasswordManagerBrowserTest,
     NoPromptForFailedLoginFromSubFrameWithMultiFramesInPage) {
   NavigateToFile("/password/multi_frames.html");
 
   // Make sure that we don't prompt to save the password for a failed login
   // from a sub-frame with multiple frames in the same page.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "var first_frame = document.getElementById('first_frame');"
       "var frame_doc = first_frame.contentDocument;"
       "frame_doc.getElementById('username_failed').value = 'temp';"
       "frame_doc.getElementById('password_failed').value = 'random';"
       "frame_doc.getElementById('submit_failed').click();"
       "window.parent.location.href = 'multi_frames.html';";
 
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  EXPECT_FALSE(observer.infobar_shown());
+  EXPECT_FALSE(prompt_observer->IsPromptShown());
 }
 
-IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
-                       PromptForXHRSubmit) {
+IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest, PromptForXHRSubmit) {
 #if defined(OS_WIN) && defined(USE_ASH)
   // Disable this test in Metro+Ash for now (http://crbug.com/262796).
   if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kAshBrowserTests))
     return;
 #endif
   NavigateToFile("/password/password_xhr_submit.html");
 
   // Verify that we show the save password prompt if a form returns false
   // in its onsubmit handler but instead logs in/navigates via XHR.
   // Note that calling 'submit()' on a form with javascript doesn't call
   // the onsubmit handler, so we click the submit button instead.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('username_field').value = 'temp';"
       "document.getElementById('password_field').value = 'random';"
       "document.getElementById('submit_button').click()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        PromptForXHRWithoutOnSubmit) {
   NavigateToFile("/password/password_xhr_submit.html");
 
   // Verify that if XHR navigation occurs and the form is properly filled out,
   // we try and save the password even though onsubmit hasn't been called.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_navigate =
       "document.getElementById('username_field').value = 'temp';"
       "document.getElementById('password_field').value = 'random';"
       "send_xhr()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_navigate));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
-IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
-                       NoPromptIfLinkClicked) {
+IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest, NoPromptIfLinkClicked) {
   NavigateToFile("/password/password_form.html");
 
   // Verify that if the user takes a direct action to leave the page, we don't
   // prompt to save the password even if the form is already filled out.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_click_link =
       "document.getElementById('username_field').value = 'temp';"
       "document.getElementById('password_field').value = 'random';"
       "document.getElementById('link').click();";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_click_link));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_FALSE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_FALSE(observer.infobar_shown());
-  }
+  EXPECT_FALSE(prompt_observer->IsPromptShown());
 }
 
 // TODO(jam): http://crbug.com/350550
 #if !defined(OS_WIN)
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        VerifyPasswordGenerationUpload) {
   // Prevent Autofill requests from actually going over the wire.
   net::TestURLFetcherFactory factory;
   // Disable Autofill requesting access to AddressBook data. This causes
   // the test to hang on Mac.
   autofill::test::DisableSystemServices(browser()->profile()->GetPrefs());
 
   // Visit a signup form.
   NavigateToFile("/password/signup_form.html");
 
   // Enter a password and save it.
   NavigationObserver first_observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('other_info').value = 'stuff';"
       "document.getElementById('username_field').value = 'my_username';"
       "document.getElementById('password_field').value = 'password';"
       "document.getElementById('input_submit_button').click()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
 
   first_observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    ASSERT_TRUE(ui_controller()->PasswordPendingUserDecision());
-    ui_controller()->SavePassword();
-  } else {
-    ASSERT_TRUE(first_observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
+  prompt_observer->Accept();
 
   // Now navigate to a login form that has similar HTML markup.
   NavigateToFile("/password/password_form.html");
 
   // Simulate a user click to force an autofill of the form's DOM value, not
   // just the suggested value.
   content::SimulateMouseClick(
       WebContents(), 0, blink::WebMouseEvent::ButtonLeft);
 
   // The form should be filled with the previously submitted username.
   std::string get_username =
       "window.domAutomationController.send("
       "document.getElementById('username_field').value);";
   std::string actual_username;
   ASSERT_TRUE(content::ExecuteScriptAndExtractString(RenderViewHost(),
                                                      get_username,
                                                      &actual_username));
   ASSERT_EQ("my_username", actual_username);
 
   // Submit the form and verify that there is no infobar (as the password
   // has already been saved).
   NavigationObserver second_observer(WebContents());
+  scoped_ptr<PromptObserver> second_prompt_observer(
+      PromptObserver::Create(WebContents()));

[engedy, 2014/07/11 18:07:26]

Could you please confirm that this really a different WebContents()? Otherwise
we could use the same observers.

[vabr (Chromium), 2014/07/11 18:48:01]

The thing is that I'm not sure -- it should be the same, but I don't know the
exact rules when a WebContents can be swapped out. It is more important to me to
make sure the prompt observer is observing the right one (since I want it to
confirm that no prompt is shown), than to spare one variable.

(After speaking with Balázs offline, I'm leaving this as it is.)

   std::string submit_form =
       "document.getElementById('input_submit_button').click()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), submit_form));
   second_observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_FALSE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_FALSE(second_observer.infobar_shown());
-  }
+  EXPECT_FALSE(second_prompt_observer->IsPromptShown());
 
   // Verify that we sent a ping to Autofill saying that the original form
   // was likely an account creation form since it has more than 2 text input
   // fields and was used for the first time on a different form.
   base::HistogramBase* upload_histogram =
       base::StatisticsRecorder::FindHistogram(
           "PasswordGeneration.UploadStarted");
   ASSERT_TRUE(upload_histogram);
   scoped_ptr<base::HistogramSamples> snapshot =
       upload_histogram->SnapshotSamples();
   EXPECT_EQ(0, snapshot->GetCount(0 /* failure */));
   EXPECT_EQ(1, snapshot->GetCount(1 /* success */));
 }
 #endif
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest, PromptForSubmitFromIframe) {
   NavigateToFile("/password/password_submit_from_iframe.html");
 
   // Submit a form in an iframe, then cause the whole page to navigate without a
   // user gesture. We expect the save password prompt to be shown here, because
   // some pages use such iframes for login forms.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "var iframe = document.getElementById('test_iframe');"
       "var iframe_doc = iframe.contentDocument;"
       "iframe_doc.getElementById('username_field').value = 'temp';"
       "iframe_doc.getElementById('password_field').value = 'random';"
       "iframe_doc.getElementById('submit_button').click()";
 
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        PromptForInputElementWithoutName) {
   // Check that the prompt is shown for forms where input elements lack the
   // "name" attribute but the "id" is present.
   NavigateToFile("/password/password_form.html");
 
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('username_field_no_name').value = 'temp';"
       "document.getElementById('password_field_no_name').value = 'random';"
       "document.getElementById('input_submit_button_no_name').click()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        PromptForInputElementWithoutId) {
   // Check that the prompt is shown for forms where input elements lack the
   // "id" attribute but the "name" attribute is present.
   NavigateToFile("/password/password_form.html");
 
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementsByName('username_field_no_id')[0].value = 'temp';"
       "document.getElementsByName('password_field_no_id')[0].value = 'random';"
       "document.getElementsByName('input_submit_button_no_id')[0].click()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        NoPromptForInputElementWithoutIdAndName) {
   // Check that no prompt is shown for forms where the input fields lack both
   // the "id" and the "name" attributes.
   NavigateToFile("/password/password_form.html");
 
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "var form = document.getElementById('testform_elements_no_id_no_name');"
       "var username = form.children[0];"
       "username.value = 'temp';"
       "var password = form.children[1];"
       "password.value = 'random';"
       "form.children[2].click()";  // form.children[2] is the submit button.
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_FALSE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_FALSE(observer.infobar_shown());
-  }
+  EXPECT_FALSE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest, DeleteFrameBeforeSubmit) {
   NavigateToFile("/password/multi_frames.html");
 
   NavigationObserver observer(WebContents());
   // Make sure we save some password info from an iframe and then destroy it.
   std::string save_and_remove =
       "var first_frame = document.getElementById('first_frame');"
       "var frame_doc = first_frame.contentDocument;"
@@ skipping 21 matching lines @@
   // Click on a link to open a new tab, then switch back to the first one.
   EXPECT_EQ(1, browser()->tab_strip_model()->count());
   std::string click =
       "document.getElementById('testlink').click();";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), click));
   EXPECT_EQ(2, browser()->tab_strip_model()->count());
   browser()->tab_strip_model()->ActivateTabAt(0, false);
 
   // Fill in the credentials, and make sure they are saved.
   NavigationObserver form_submit_observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('username_field').value = 'temp';"
       "document.getElementById('password_field').value = 'random';"
       "document.getElementById('input_submit_button').click();";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   form_submit_observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-    ui_controller()->SavePassword();
-  } else {
-    EXPECT_TRUE(form_submit_observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
+  prompt_observer->Accept();
 
   // Reload the original page to have the saved credentials autofilled.
   NavigationObserver reload_observer(WebContents());
   NavigateToFile("/password/form_and_link.html");
   reload_observer.Wait();
 
   // Wait until the username is filled, to make sure autofill kicked in.
   WaitForElementValue("username_field", "temp");
   // Now check that the password is not accessible yet.
   CheckElementValue("password_field", "");
@@ skipping 11 matching lines @@
 // RenderWidgetHostViewGuest::ProcessAckedTouchEvent is, and
 // ProcessAckedTouchEvent is what triggers the translation of touch events to
 // gesture events.
 #if defined(USE_AURA)
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        PasswordValueAccessibleOnSubmit) {
   NavigateToFile("/password/form_and_link.html");
 
   // Fill in the credentials, and make sure they are saved.
   NavigationObserver form_submit_observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('username_field').value = 'temp';"
       "document.getElementById('password_field').value = 'random_secret';"
       "document.getElementById('input_submit_button').click();";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   form_submit_observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-    ui_controller()->SavePassword();
-  } else {
-    EXPECT_TRUE(form_submit_observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
+  prompt_observer->Accept();
 
   // Reload the original page to have the saved credentials autofilled.
   NavigationObserver reload_observer(WebContents());
   NavigateToFile("/password/form_and_link.html");
   reload_observer.Wait();
 
   NavigationObserver submit_observer(WebContents());
   // Submit the form via a tap on the submit button. The button is placed at 0,
   // 100, and has height 300 and width 700.
   content::SimulateTapAt(WebContents(), gfx::Point(350, 250));
   submit_observer.Wait();
   std::string query = WebContents()->GetURL().query();
   EXPECT_NE(std::string::npos, query.find("random_secret")) << query;
 }
 #endif
 
 // Test fix for crbug.com/338650.
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        DontPromptForPasswordFormWithDefaultValue) {
   NavigateToFile("/password/password_form_with_default_value.html");
 
   // Don't prompt if we navigate away even if there is a password value since
   // it's not coming from the user.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   NavigateToFile("/password/done.html");
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_FALSE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_FALSE(observer.infobar_shown());
-  }
+  EXPECT_FALSE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        PromptWhenEnableAutomaticPasswordSavingSwitchIsNotSet) {
   NavigateToFile("/password/password_form.html");
 
   // Fill a form and submit through a <input type="submit"> button.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string fill_and_submit =
       "document.getElementById('username_field').value = 'temp';"
       "document.getElementById('password_field').value = 'random';"
       "document.getElementById('input_submit_button').click()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        DontPromptWhenEnableAutomaticPasswordSavingSwitchIsSet) {
   password_manager::TestPasswordStore* password_store =
       static_cast<password_manager::TestPasswordStore*>(
           PasswordStoreFactory::GetForProfile(browser()->profile(),
                                               Profile::IMPLICIT_ACCESS).get());
 
   EXPECT_TRUE(password_store->IsEmpty());
 
   NavigateToFile("/password/password_form.html");
 
   // Add the enable-automatic-password-saving switch.
   CommandLine::ForCurrentProcess()->AppendSwitch(
       password_manager::switches::kEnableAutomaticPasswordSaving);
 
   // Fill a form and submit through a <input type="submit"> button.
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   // Make sure that the only passwords saved are the auto-saved ones.
-  observer.disable_should_automatically_accept_infobar();
   std::string fill_and_submit =
       "document.getElementById('username_field').value = 'temp';"
       "document.getElementById('password_field').value = 'random';"
       "document.getElementById('input_submit_button').click()";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill_and_submit));
   observer.Wait();
   if (chrome::VersionInfo::GetChannel() ==
       chrome::VersionInfo::CHANNEL_UNKNOWN) {
-    if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-      EXPECT_FALSE(ui_controller()->PasswordPendingUserDecision());
-    } else {
-      EXPECT_FALSE(observer.infobar_shown());
-    }
+    // Passwords getting auto-saved, no prompt.
+    EXPECT_FALSE(prompt_observer->IsPromptShown());
     EXPECT_FALSE(password_store->IsEmpty());
   } else {
-    if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-      EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-    } else {
-      EXPECT_TRUE(observer.infobar_shown());
-    }
+    // Prompt shown, and no passwords saved automatically.
+    EXPECT_TRUE(prompt_observer->IsPromptShown());
     EXPECT_TRUE(password_store->IsEmpty());
   }
 }
 
 // Test fix for crbug.com/368690.
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest, NoPromptWhenReloading) {
   NavigateToFile("/password/password_form.html");
 
   std::string fill =
       "document.getElementById('username_redirect').value = 'temp';"
       "document.getElementById('password_redirect').value = 'random';";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), fill));
 
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   GURL url = embedded_test_server()->GetURL("/password/password_form.html");
   chrome::NavigateParams params(browser(), url,
                                 content::PAGE_TRANSITION_RELOAD);
   ui_test_utils::NavigateToURL(&params);
   observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_FALSE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_FALSE(observer.infobar_shown());
-  }
+  EXPECT_FALSE(prompt_observer->IsPromptShown());
 }
 
 // Test that if a form gets dynamically added between the form parsing and
 // rendering, and while the main frame still loads, it still is registered, and
 // thus saving passwords from it works.
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest,
                        FormsAddedBetweenParsingAndRendering) {
   NavigateToFile("/password/between_parsing_and_rendering.html");
 
   NavigationObserver observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   std::string submit =
       "document.getElementById('username').value = 'temp';"
       "document.getElementById('password').value = 'random';"
       "document.getElementById('submit-button').click();";
   ASSERT_TRUE(content::ExecuteScript(RenderViewHost(), submit));
   observer.Wait();
 
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-  } else {
-    EXPECT_TRUE(observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
 }
 
 // Test that if there was no previous page load then the PasswordManagerDriver
 // does not think that there were SSL errors on the current page. The test opens
 // a new tab with a URL for which the embedded test server issues a basic auth
 // challenge.
 IN_PROC_BROWSER_TEST_F(PasswordManagerBrowserTest, NoLastLoadGoodLastLoad) {
   // Teach the embedded server to handle requests by issuing the basic auth
   // challenge.
   embedded_test_server()->RegisterRequestHandler(
@@ skipping 15 matching lines @@
   // authentication.
   ui_test_utils::NavigateToURLWithDisposition(
       browser(),
       embedded_test_server()->GetURL("/basic_auth"),
       NEW_FOREGROUND_TAB,
       ui_test_utils::BROWSER_TEST_WAIT_FOR_TAB);
 
   content::NavigationController* nav_controller =
       &WebContents()->GetController();
   NavigationObserver nav_observer(WebContents());
+  scoped_ptr<PromptObserver> prompt_observer(
+      PromptObserver::Create(WebContents()));
   WindowedAuthNeededObserver auth_needed_observer(nav_controller);
   auth_needed_observer.Wait();
 
   WindowedAuthSuppliedObserver auth_supplied_observer(nav_controller);
   // Offer valid credentials on the auth challenge.
   ASSERT_EQ(1u, login_observer.handlers().size());
   LoginHandler* handler = *login_observer.handlers().begin();
   ASSERT_TRUE(handler);
   // Any username/password will work.
   handler->SetAuth(base::UTF8ToUTF16("user"), base::UTF8ToUTF16("pwd"));
   auth_supplied_observer.Wait();
 
   // The password manager should be working correctly.
   nav_observer.Wait();
-  if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
-    EXPECT_TRUE(ui_controller()->PasswordPendingUserDecision());
-    ui_controller()->SavePassword();
-  } else {
-    EXPECT_TRUE(nav_observer.infobar_shown());
-  }
+  EXPECT_TRUE(prompt_observer->IsPromptShown());
+  prompt_observer->Accept();
+
   // Spin the message loop to make sure the password store had a chance to save
   // the password.
   base::RunLoop run_loop;
   run_loop.RunUntilIdle();
   EXPECT_FALSE(password_store->IsEmpty());
 }