[Password Manager] Presave the form with generated password till successful login

chrome/renderer/autofill/password_generation_agent_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.h>
 
 #include "base/macros.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/histogram_tester.h"
@@ skipping 557 matching lines @@
 
   ExpectGenerationAvailable("first_password", false);
 }
 
 TEST_F(PasswordGenerationAgentTest, ChangePasswordFormDetectionTest) {
   // Verify that generation is shown on correct field after message receiving.
   LoadHTMLWithUserGesture(kPasswordChangeFormHTML);
   SetNotBlacklistedMessage(password_generation_, kPasswordChangeFormHTML);
   ExpectGenerationAvailable("password", false);
   ExpectGenerationAvailable("newpassword", false);
+  ExpectGenerationAvailable("confirmpassword", false);
 
   SetAccountCreationFormsDetectedMessage(password_generation_,
                                          GetMainFrame()->document(), 0, 2);
   ExpectGenerationAvailable("password", false);
   ExpectGenerationAvailable("newpassword", true);
+  ExpectGenerationAvailable("confirmpassword", false);
 }
 
 TEST_F(PasswordGenerationAgentTest, ManualGenerationInFormTest) {
   LoadHTMLWithUserGesture(kAccountCreationFormHTML);
   ShowGenerationPopUpManually("first_password");
   ExpectGenerationAvailable("first_password", true);
   ExpectGenerationAvailable("second_password", false);
 }
 
 TEST_F(PasswordGenerationAgentTest, ManualGenerationNoFormTest) {
   LoadHTMLWithUserGesture(kAccountCreationNoForm);
   ShowGenerationPopUpManually("first_password");
   ExpectGenerationAvailable("first_password", true);
   ExpectGenerationAvailable("second_password", false);
 }
 
 TEST_F(PasswordGenerationAgentTest, ManualGenerationChangeFocusTest) {
   // This test simulates focus change after user triggered password generation.
   // PasswordGenerationAgent should save last focused password element and
   // generate password, even if focused element has changed.
   LoadHTMLWithUserGesture(kAccountCreationFormHTML);
   FocusField("first_password");
   ShowGenerationPopUpManually("username" /* current focus */);
   ExpectGenerationAvailable("first_password", true);
   ExpectGenerationAvailable("second_password", false);
 }
 
+TEST_F(PasswordGenerationAgentTest, PresavingGeneratedPassword) {
+  const struct {
+    const char* form;
+    const char* generation_element;
+  } kTestCases[] = {{kAccountCreationFormHTML, "first_password"},
+                    {kAccountCreationNoForm, "first_password"},
+                    {kPasswordChangeFormHTML, "newpassword"}};
+  for (auto& test_case : kTestCases) {
+    SCOPED_TRACE(testing::Message("form: ") << test_case.form);
+    LoadHTMLWithUserGesture(test_case.form);
+    // To be able to work with input elements outside <form>'s, use manual
+    // generation.
+    ShowGenerationPopUpManually(test_case.generation_element);
+    ExpectGenerationAvailable(test_case.generation_element, true);
+
+    base::string16 password = base::ASCIIToUTF16("random_password");
+    AutofillMsg_GeneratedPasswordAccepted msg(0, password);
+    password_generation_->OnMessageReceived(msg);
+    EXPECT_TRUE(render_thread_->sink().GetFirstMessageMatching(
+        AutofillHostMsg_PresaveGeneratedPassword::ID));
+    render_thread_->sink().ClearMessages();
+
+    FocusField(test_case.generation_element);
+    SimulateUserTypingASCIICharacter('a', true);
+    EXPECT_TRUE(render_thread_->sink().GetFirstMessageMatching(
+        AutofillHostMsg_PresaveGeneratedPassword::ID));
+    render_thread_->sink().ClearMessages();
+
+    for (size_t i = 0; i < password.length(); ++i)
+      SimulateUserTypingASCIICharacter(ui::VKEY_BACK, false);
+    SimulateUserTypingASCIICharacter(ui::VKEY_BACK, true);
+    EXPECT_TRUE(render_thread_->sink().GetFirstMessageMatching(
+        AutofillHostMsg_PasswordNoLongerGenerated::ID));
+    render_thread_->sink().ClearMessages();
+  }
+}
+
 }  // namespace autofill