chrome/browser/password_manager/save_password_infobar_delegate_unittest.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/save_password_infobar_delegate.h"

#include <string>

#include "base/prefs/pref_service.h"
#include "base/strings/utf_string_conversions.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/test/base/chrome_render_view_host_test_harness.h"
#include "components/autofill/core/common/password_form.h"
#include "components/password_manager/core/browser/password_form_manager.h"
#include "components/password_manager/core/browser/password_manager.h"
#include "components/password_manager/core/browser/stub_password_manager_client.h"
#include "components/password_manager/core/browser/stub_password_manager_driver.h"
#include "components/password_manager/core/common/password_manager_pref_names.h"
#include "content/public/browser/web_contents.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace {

class MockPasswordFormManager : public password_manager::PasswordFormManager {
 public:
  MOCK_METHOD0(PermanentlyBlacklist, void());

  MockPasswordFormManager(
      password_manager::PasswordManager* password_manager,
      password_manager::PasswordManagerClient* client,
      base::WeakPtr<password_manager::PasswordManagerDriver> driver,
      const autofill::PasswordForm& form)
      : PasswordFormManager(password_manager, client, driver, form, false) {}

  ~MockPasswordFormManager() override {}

 private:
  DISALLOW_COPY_AND_ASSIGN(MockPasswordFormManager);
};

class TestSavePasswordInfobarDelegate : public SavePasswordInfoBarDelegate {
 public:
  TestSavePasswordInfobarDelegate(
      content::WebContents* web_contents,
      scoped_ptr<password_manager::PasswordFormManager> form_to_save,
      password_manager::CredentialSourceType source_type,
      bool should_show_first_run_experience)
      : SavePasswordInfoBarDelegate(web_contents,
                                    form_to_save.Pass(),
                                    std::string(),
                                    source_type,
                                    true /* is_smartlock_branding_enabled */,
                                    should_show_first_run_experience) {}

  ~TestSavePasswordInfobarDelegate() override {}
};

}  // namespace

class SavePasswordInfoBarDelegateTest : public ChromeRenderViewHostTestHarness {
 public:
  SavePasswordInfoBarDelegateTest();
  ~SavePasswordInfoBarDelegateTest() override{};

  void SetUp() override;
  void TearDown() override;

  PrefService* prefs();
  const autofill::PasswordForm& test_form() { return test_form_; }
  scoped_ptr<MockPasswordFormManager> CreateMockFormManager();

 protected:
  scoped_ptr<ConfirmInfoBarDelegate> CreateDelegate(
      scoped_ptr<password_manager::PasswordFormManager> password_form_manager,
      password_manager::CredentialSourceType type,
      bool should_show_first_run_experience);

  password_manager::StubPasswordManagerClient client_;
  password_manager::StubPasswordManagerDriver driver_;
  password_manager::PasswordManager password_manager_;

  autofill::PasswordForm test_form_;

 private:
  DISALLOW_COPY_AND_ASSIGN(SavePasswordInfoBarDelegateTest);
};

SavePasswordInfoBarDelegateTest::SavePasswordInfoBarDelegateTest()
    : password_manager_(&client_) {
  test_form_.origin = GURL("http://example.com");
  test_form_.username_value = base::ASCIIToUTF16("username");
  test_form_.password_value = base::ASCIIToUTF16("12345");
}

PrefService* SavePasswordInfoBarDelegateTest::prefs() {
  Profile* profile =
      Profile::FromBrowserContext(web_contents()->GetBrowserContext());
  return profile->GetPrefs();
}

scoped_ptr<MockPasswordFormManager>
SavePasswordInfoBarDelegateTest::CreateMockFormManager() {
  return scoped_ptr<MockPasswordFormManager>(new MockPasswordFormManager(
      &password_manager_, &client_, driver_.AsWeakPtr(), test_form()));
}

scoped_ptr<ConfirmInfoBarDelegate>
SavePasswordInfoBarDelegateTest::CreateDelegate(
    scoped_ptr<password_manager::PasswordFormManager> password_form_manager,
    password_manager::CredentialSourceType type,
    bool should_show_first_run_experience) {
  scoped_ptr<ConfirmInfoBarDelegate> delegate(
      new TestSavePasswordInfobarDelegate(web_contents(),
                                          password_form_manager.Pass(), type,
                                          should_show_first_run_experience));
  return delegate.Pass();
}

void SavePasswordInfoBarDelegateTest::SetUp() {
  ChromeRenderViewHostTestHarness::SetUp();
}

void SavePasswordInfoBarDelegateTest::TearDown() {
  ChromeRenderViewHostTestHarness::TearDown();
}

TEST_F(SavePasswordInfoBarDelegateTest, CancelTestCredentialSourceAPI) {
  scoped_ptr<MockPasswordFormManager> password_form_manager(
      CreateMockFormManager());
  EXPECT_CALL(*password_form_manager.get(), PermanentlyBlacklist());
  scoped_ptr<ConfirmInfoBarDelegate> infobar(CreateDelegate(
      password_form_manager.Pass(),
      password_manager::CredentialSourceType::CREDENTIAL_SOURCE_API, false));
  EXPECT_TRUE(infobar->Cancel());
}

TEST_F(SavePasswordInfoBarDelegateTest,
       CancelTestCredentialSourcePasswordManager) {
  scoped_ptr<MockPasswordFormManager> password_form_manager(
      CreateMockFormManager());
  EXPECT_CALL(*password_form_manager.get(), PermanentlyBlacklist());
  scoped_ptr<ConfirmInfoBarDelegate> infobar(CreateDelegate(
      password_form_manager.Pass(), password_manager::CredentialSourceType::
                                        CREDENTIAL_SOURCE_PASSWORD_MANAGER,
      false));
  EXPECT_TRUE(infobar->Cancel());
}

TEST_F(SavePasswordInfoBarDelegateTest,
       CheckResetOfPrefAfterFirstRunMessageWasShown) {
  using password_manager::CredentialSourceType;
  prefs()->SetBoolean(
      password_manager::prefs::kWasSavePrompFirstRunExperienceShown, false);
  scoped_ptr<MockPasswordFormManager> password_form_manager(
      CreateMockFormManager());
  scoped_ptr<ConfirmInfoBarDelegate> infobar(CreateDelegate(
      password_form_manager.Pass(), password_manager::CredentialSourceType::
                                        CREDENTIAL_SOURCE_PASSWORD_MANAGER,
      true));
  EXPECT_TRUE(infobar->Cancel());
  infobar.reset();
  EXPECT_TRUE(prefs()->GetBoolean(
      password_manager::prefs::kWasSavePrompFirstRunExperienceShown));
}

TEST_F(SavePasswordInfoBarDelegateTest,
       CheckNoStateOfPrefChangeWhenNoFirstRunExperienceShown) {
  using password_manager::CredentialSourceType;
  prefs()->SetBoolean(
      password_manager::prefs::kWasSavePrompFirstRunExperienceShown, false);
  scoped_ptr<MockPasswordFormManager> password_form_manager(
      CreateMockFormManager());
  scoped_ptr<ConfirmInfoBarDelegate> infobar(CreateDelegate(
      password_form_manager.Pass(), password_manager::CredentialSourceType::
                                        CREDENTIAL_SOURCE_PASSWORD_MANAGER,
      false));
  EXPECT_TRUE(infobar->Cancel());
  infobar.reset();
  EXPECT_FALSE(prefs()->GetBoolean(
      password_manager::prefs::kWasSavePrompFirstRunExperienceShown));
}