Update Confirmation UI for saved password change

chrome/browser/password_manager/password_manager_test_base.cc

 // Copyright 2015 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 "chrome/browser/password_manager/password_manager_test_base.h"
 
 #include "base/command_line.h"
 #include "base/run_loop.h"
 #include "base/strings/stringprintf.h"
 #include "chrome/browser/infobars/infobar_service.h"
@@ skipping 44 matching lines @@
 
 void NavigationObserver::Wait() {
   message_loop_runner_->Run();
 }
 
 PromptObserver::PromptObserver() {
 }
 PromptObserver::~PromptObserver() {
 }
 
+bool PromptObserver::IsShowingUpdatePrompt() const {
+  // TODO(dvadym): Make this method pure virtual as soon as update UI is
+  // implemented for infobar. http://crbug.com/359315
+  return false;
+}
+
 void PromptObserver::Accept() const {
   EXPECT_TRUE(IsShowingPrompt());
   AcceptImpl();
 }
 
+void PromptObserver::AcceptUpdatePrompt(
+    const autofill::PasswordForm& form) const {
+  EXPECT_TRUE(IsShowingUpdatePrompt());
+  AcceptUpdatePromptImpl(form);
+}
+
 class InfoBarObserver : public PromptObserver,
                         public infobars::InfoBarManager::Observer {
  public:
   explicit InfoBarObserver(content::WebContents* web_contents)
       : infobar_is_being_shown_(false),
         infobar_service_(InfoBarService::FromWebContents(web_contents)) {
     infobar_service_->AddObserver(this);
   }
 
   ~InfoBarObserver() override {
@@ skipping 48 matching lines @@
             ManagePasswordsUIController::FromWebContents(web_contents)) {}
 
   ~BubbleObserver() override {}
 
  private:
   // PromptObserver:
   bool IsShowingPrompt() const override {
     return ui_controller_->PasswordPendingUserDecision();
   }
 
+  bool IsShowingUpdatePrompt() const override {
+    return ui_controller_->state() ==
+           password_manager::ui::PENDING_PASSWORD_UPDATE_STATE;
+  }
+
   void AcceptImpl() const override {
     ui_controller_->SavePassword();
     EXPECT_FALSE(IsShowingPrompt());
   }
 
+  void AcceptUpdatePromptImpl(
+      const autofill::PasswordForm& form) const override {
+    ui_controller_->UpdatePassword(form);
+    EXPECT_FALSE(IsShowingUpdatePrompt());
+  }
   ManagePasswordsUIController* const ui_controller_;
 
   DISALLOW_COPY_AND_ASSIGN(BubbleObserver);
 };
 
 scoped_ptr<PromptObserver> PromptObserver::Create(
     content::WebContents* web_contents) {
   if (ChromePasswordManagerClient::IsTheHotNewBubbleUIEnabled()) {
     return scoped_ptr<PromptObserver>(new BubbleObserver(web_contents));
   } else {
@@ skipping 117 matching lines @@
       "  var element = document.getElementById('%s');"
       "window.domAutomationController.send(element && element.value == '%s');",
       iframe_id.c_str(), iframe_id.c_str(), element_id.c_str(),
       element_id.c_str(), expected_value.c_str());
   bool return_value = false;
   ASSERT_TRUE(content::ExecuteScriptAndExtractBool(
       RenderViewHost(), value_check_script, &return_value));
   EXPECT_TRUE(return_value) << "element_id = " << element_id
                             << ", expected_value = " << expected_value;
 }