Password bubble: Give users the option of unblacklisting a site.

chrome/browser/ui/passwords/manage_passwords_bubble_model_unittest.cc

 // Copyright 2014 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 "base/metrics/histogram_samples.h"
 #include "base/prefs/pref_service.h"
+#include "base/strings/utf_string_conversions.h"
 #include "base/test/statistics_delta_reader.h"
 #include "chrome/browser/ui/passwords/manage_passwords_bubble.h"
 #include "chrome/browser/ui/passwords/manage_passwords_bubble_model.h"
 #include "chrome/browser/ui/passwords/manage_passwords_bubble_ui_controller_mock.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/password_manager/core/browser/password_manager_metrics_util.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "content/public/test/web_contents_tester.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
@@ skipping 21 matching lines @@
         ManagePasswordsBubbleModel::PASSWORD_TO_BE_SAVED);
     model_->OnBubbleShown(ManagePasswordsBubble::AUTOMATIC);
   }
 
   void PretendManagingPasswords() {
     model_->set_manage_passwords_bubble_state(
         ManagePasswordsBubbleModel::MANAGE_PASSWORDS);
     model_->OnBubbleShown(ManagePasswordsBubble::USER_ACTION);
   }
 
+  void PretendBlacklisted() {
+    model_->set_manage_passwords_bubble_state(
+        ManagePasswordsBubbleModel::NEVER_SAVE_PASSWORDS);
+    model_->OnBubbleShown(ManagePasswordsBubble::USER_ACTION);
+
+    base::string16 kTestUsername = base::ASCIIToUTF16("test_username");
+    autofill::PasswordFormMap map;
+    map[kTestUsername] = &test_form_;
+    controller()->SetPasswordFormMap(map);
+  }
+
   ManagePasswordsBubbleUIControllerMock* controller() {
     return reinterpret_cast<ManagePasswordsBubbleUIControllerMock*>(
         ManagePasswordsBubbleUIController::FromWebContents(
             test_web_contents_.get()));
   }
 
  protected:
   scoped_ptr<ManagePasswordsBubbleModel> model_;
+  autofill::PasswordForm test_form_;
 
  private:
   content::TestBrowserThreadBundle thread_bundle_;
   TestingProfile profile_;
   scoped_ptr<content::WebContents> test_web_contents_;
 };
 
 TEST_F(ManagePasswordsBubbleModelTest, DefaultValues) {
   EXPECT_EQ(model_->display_disposition(),
             password_manager::metrics_util::AUTOMATIC_WITH_PASSWORD_PENDING);
@@ skipping 17 matching lines @@
           kUIDismissalReasonMetric));
   EXPECT_EQ(NULL, samples.get());
 }
 
 TEST_F(ManagePasswordsBubbleModelTest, CloseWithoutInteraction) {
   base::StatisticsDeltaReader statistics_delta_reader;
   PretendPasswordWaiting();
   model_->OnBubbleHidden();
   EXPECT_EQ(model_->dismissal_reason(),
             password_manager::metrics_util::NO_DIRECT_INTERACTION);
+  EXPECT_TRUE(model_->WaitingToSavePassword());
   EXPECT_FALSE(controller()->saved_password());
   EXPECT_FALSE(controller()->never_saved_password());
 
   scoped_ptr<base::HistogramSamples> samples(
       statistics_delta_reader.GetHistogramSamplesSinceCreation(
           kUIDismissalReasonMetric));
   EXPECT_EQ(
       1,
       samples->GetCount(password_manager::metrics_util::NO_DIRECT_INTERACTION));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_SAVE));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_NOPE));
   EXPECT_EQ(0,
             samples->GetCount(password_manager::metrics_util::CLICKED_NEVER));
   EXPECT_EQ(0,
             samples->GetCount(password_manager::metrics_util::CLICKED_MANAGE));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_DONE));
+  EXPECT_EQ(
+      0,
+      samples->GetCount(password_manager::metrics_util::CLICKED_UNBLACKLIST));
 }
 
 TEST_F(ManagePasswordsBubbleModelTest, ClickSave) {
   base::StatisticsDeltaReader statistics_delta_reader;
   PretendPasswordWaiting();
   model_->OnSaveClicked();
   model_->OnBubbleHidden();
   EXPECT_EQ(model_->dismissal_reason(),
             password_manager::metrics_util::CLICKED_SAVE);
+  EXPECT_FALSE(model_->WaitingToSavePassword());
+  EXPECT_FALSE(model_->NeverSavingPasswords());
   EXPECT_TRUE(controller()->saved_password());
   EXPECT_FALSE(controller()->never_saved_password());
 
   scoped_ptr<base::HistogramSamples> samples(
       statistics_delta_reader.GetHistogramSamplesSinceCreation(
           kUIDismissalReasonMetric));
   EXPECT_EQ(
       0,
       samples->GetCount(password_manager::metrics_util::NO_DIRECT_INTERACTION));
   EXPECT_EQ(1, samples->GetCount(password_manager::metrics_util::CLICKED_SAVE));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_NOPE));
   EXPECT_EQ(0,
             samples->GetCount(password_manager::metrics_util::CLICKED_NEVER));
   EXPECT_EQ(0,
             samples->GetCount(password_manager::metrics_util::CLICKED_MANAGE));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_DONE));
+  EXPECT_EQ(
+      0,
+      samples->GetCount(password_manager::metrics_util::CLICKED_UNBLACKLIST));
 }
 
 TEST_F(ManagePasswordsBubbleModelTest, ClickNope) {
   base::StatisticsDeltaReader statistics_delta_reader;
   PretendPasswordWaiting();
   model_->OnNopeClicked();
   model_->OnBubbleHidden();
   EXPECT_EQ(model_->dismissal_reason(),
             password_manager::metrics_util::CLICKED_NOPE);
+  EXPECT_TRUE(model_->WaitingToSavePassword());
+  EXPECT_FALSE(model_->NeverSavingPasswords());
   EXPECT_FALSE(controller()->saved_password());
   EXPECT_FALSE(controller()->never_saved_password());
 
   scoped_ptr<base::HistogramSamples> samples(
       statistics_delta_reader.GetHistogramSamplesSinceCreation(
           kUIDismissalReasonMetric));
   EXPECT_EQ(
       0,
       samples->GetCount(password_manager::metrics_util::NO_DIRECT_INTERACTION));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_SAVE));
   EXPECT_EQ(1, samples->GetCount(password_manager::metrics_util::CLICKED_NOPE));
   EXPECT_EQ(0,
             samples->GetCount(password_manager::metrics_util::CLICKED_NEVER));
   EXPECT_EQ(0,
             samples->GetCount(password_manager::metrics_util::CLICKED_MANAGE));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_DONE));
+  EXPECT_EQ(
+      0,
+      samples->GetCount(password_manager::metrics_util::CLICKED_UNBLACKLIST));
 }
 
 TEST_F(ManagePasswordsBubbleModelTest, ClickNever) {
   base::StatisticsDeltaReader statistics_delta_reader;
   PretendPasswordWaiting();
   model_->OnNeverForThisSiteClicked();
   model_->OnBubbleHidden();
   EXPECT_EQ(model_->dismissal_reason(),
             password_manager::metrics_util::CLICKED_NEVER);
+  EXPECT_FALSE(model_->WaitingToSavePassword());
+  EXPECT_TRUE(model_->NeverSavingPasswords());
   EXPECT_FALSE(controller()->saved_password());
   EXPECT_TRUE(controller()->never_saved_password());
 
   scoped_ptr<base::HistogramSamples> samples(
       statistics_delta_reader.GetHistogramSamplesSinceCreation(
           kUIDismissalReasonMetric));
   EXPECT_EQ(
       0,
       samples->GetCount(password_manager::metrics_util::NO_DIRECT_INTERACTION));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_SAVE));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_NOPE));
   EXPECT_EQ(1,
             samples->GetCount(password_manager::metrics_util::CLICKED_NEVER));
   EXPECT_EQ(0,
             samples->GetCount(password_manager::metrics_util::CLICKED_MANAGE));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_DONE));
+  EXPECT_EQ(
+      0,
+      samples->GetCount(password_manager::metrics_util::CLICKED_UNBLACKLIST));
 }
 
 TEST_F(ManagePasswordsBubbleModelTest, ClickManage) {
   base::StatisticsDeltaReader statistics_delta_reader;
   PretendManagingPasswords();
   model_->OnManageLinkClicked();
   model_->OnBubbleHidden();
   EXPECT_EQ(model_->dismissal_reason(),
             password_manager::metrics_util::CLICKED_MANAGE);
+  EXPECT_FALSE(model_->WaitingToSavePassword());
+  EXPECT_FALSE(model_->NeverSavingPasswords());
   EXPECT_FALSE(controller()->saved_password());
   EXPECT_FALSE(controller()->never_saved_password());
 
   scoped_ptr<base::HistogramSamples> samples(
       statistics_delta_reader.GetHistogramSamplesSinceCreation(
           kUIDismissalReasonMetric));
   EXPECT_EQ(
       0,
       samples->GetCount(password_manager::metrics_util::NO_DIRECT_INTERACTION));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_SAVE));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_NOPE));
   EXPECT_EQ(0,
             samples->GetCount(password_manager::metrics_util::CLICKED_NEVER));
   EXPECT_EQ(1,
             samples->GetCount(password_manager::metrics_util::CLICKED_MANAGE));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_DONE));
+  EXPECT_EQ(
+      0,
+      samples->GetCount(password_manager::metrics_util::CLICKED_UNBLACKLIST));
 }
 
 TEST_F(ManagePasswordsBubbleModelTest, ClickDone) {
   base::StatisticsDeltaReader statistics_delta_reader;
   PretendManagingPasswords();
   model_->OnDoneClicked();
   model_->OnBubbleHidden();
   EXPECT_EQ(model_->dismissal_reason(),
             password_manager::metrics_util::CLICKED_DONE);
+  EXPECT_FALSE(model_->WaitingToSavePassword());
+  EXPECT_FALSE(model_->NeverSavingPasswords());
   EXPECT_FALSE(controller()->saved_password());
   EXPECT_FALSE(controller()->never_saved_password());
 
   scoped_ptr<base::HistogramSamples> samples(
       statistics_delta_reader.GetHistogramSamplesSinceCreation(
           kUIDismissalReasonMetric));
   EXPECT_EQ(
       0,
       samples->GetCount(password_manager::metrics_util::NO_DIRECT_INTERACTION));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_SAVE));
   EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_NOPE));
   EXPECT_EQ(0,
             samples->GetCount(password_manager::metrics_util::CLICKED_NEVER));
   EXPECT_EQ(0,
             samples->GetCount(password_manager::metrics_util::CLICKED_MANAGE));
   EXPECT_EQ(1, samples->GetCount(password_manager::metrics_util::CLICKED_DONE));
+  EXPECT_EQ(
+      0,
+      samples->GetCount(password_manager::metrics_util::CLICKED_UNBLACKLIST));
+}
+
+TEST_F(ManagePasswordsBubbleModelTest, ClickUnblacklist) {
+  base::StatisticsDeltaReader statistics_delta_reader;
+  PretendBlacklisted();
+  model_->OnUnblacklistClicked();
+  model_->OnBubbleHidden();
+  EXPECT_EQ(model_->dismissal_reason(),
+            password_manager::metrics_util::CLICKED_UNBLACKLIST);
+  EXPECT_FALSE(model_->WaitingToSavePassword());
+  EXPECT_FALSE(model_->NeverSavingPasswords());
+  EXPECT_FALSE(controller()->saved_password());
+  EXPECT_FALSE(controller()->never_saved_password());
+
+  scoped_ptr<base::HistogramSamples> samples(
+      statistics_delta_reader.GetHistogramSamplesSinceCreation(
+          kUIDismissalReasonMetric));
+  EXPECT_EQ(
+      0,
+      samples->GetCount(password_manager::metrics_util::NO_DIRECT_INTERACTION));
+  EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_SAVE));
+  EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_NOPE));
+  EXPECT_EQ(0,
+            samples->GetCount(password_manager::metrics_util::CLICKED_NEVER));
+  EXPECT_EQ(0,
+            samples->GetCount(password_manager::metrics_util::CLICKED_MANAGE));
+  EXPECT_EQ(0, samples->GetCount(password_manager::metrics_util::CLICKED_DONE));
+  EXPECT_EQ(
+      1,
+      samples->GetCount(password_manager::metrics_util::CLICKED_UNBLACKLIST));
 }
 
 TEST_F(ManagePasswordsBubbleModelTest, WaitingToSavePassword) {
   EXPECT_FALSE(model_->WaitingToSavePassword());
-
   model_->set_manage_passwords_bubble_state(
       ManagePasswordsBubbleModel::PASSWORD_TO_BE_SAVED);
   EXPECT_TRUE(model_->WaitingToSavePassword());
 }
+
+TEST_F(ManagePasswordsBubbleModelTest, NeverSavingPasswords) {
+  EXPECT_FALSE(model_->NeverSavingPasswords());
+  model_->set_manage_passwords_bubble_state(
+      ManagePasswordsBubbleModel::NEVER_SAVE_PASSWORDS);
+  EXPECT_TRUE(model_->NeverSavingPasswords());
+}