components/password_manager/core/browser/mock_password_store.h

// Copyright (c) 2012 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.

#ifndef COMPONENTS_PASSWORD_MANAGER_CORE_BROWSER_MOCK_PASSWORD_STORE_H_
#define COMPONENTS_PASSWORD_MANAGER_CORE_BROWSER_MOCK_PASSWORD_STORE_H_

#include <memory>
#include <string>
#include <vector>

#include "components/password_manager/core/browser/field_info_table.h"
#include "components/password_manager/core/browser/insecure_credentials_table.h"
#include "components/password_manager/core/browser/password_form.h"
#include "components/password_manager/core/browser/password_store.h"
#include "components/password_manager/core/browser/statistics_table.h"
#include "testing/gmock/include/gmock/gmock.h"

namespace password_manager {

class MockPasswordStore : public PasswordStore {
 public:
  MockPasswordStore();

  // PasswordStoreInterface

  MOCK_METHOD(void, GetAutofillableLogins, (PasswordStoreConsumer*), (override));

  MOCK_METHOD1(RemoveLogin, void(const PasswordForm&));
  MOCK_METHOD2(Unblocklist, void(const PasswordFormDigest&, base::OnceClosure));
  MOCK_METHOD2(GetLogins,
               void(const PasswordFormDigest&, PasswordStoreConsumer*));
  MOCK_METHOD1(AddLogin, void(const PasswordForm&));
  MOCK_METHOD1(UpdateLogin, void(const PasswordForm&));
  MOCK_METHOD2(UpdateLoginWithPrimaryKey,
               void(const PasswordForm&, const PasswordForm&));
  MOCK_METHOD3(ReportMetrics, void(const std::string&, bool, bool));
  MOCK_METHOD3(ReportMetricsImpl,
               void(const std::string&, bool, BulkCheckDone));
  MOCK_METHOD2(AddLoginImpl,
               PasswordStoreChangeList(const PasswordForm&,
                                       AddLoginError* error));
  MOCK_METHOD2(UpdateLoginImpl,
               PasswordStoreChangeList(const PasswordForm&,
                                       UpdateLoginError* error));
  MOCK_METHOD1(RemoveLoginImpl, PasswordStoreChangeList(const PasswordForm&));
  MOCK_METHOD3(
      RemoveLoginsByURLAndTimeImpl,
      PasswordStoreChangeList(const base::RepeatingCallback<bool(const GURL&)>&,
                              base::Time,
                              base::Time));
  MOCK_METHOD2(RemoveLoginsCreatedBetweenImpl,
               PasswordStoreChangeList(base::Time, base::Time));
  MOCK_METHOD3(RemoveStatisticsByOriginAndTimeImpl,
               bool(const base::RepeatingCallback<bool(const GURL&)>&,
                    base::Time,
                    base::Time));
  MOCK_METHOD1(DisableAutoSignInForOriginsImpl,
               PasswordStoreChangeList(
                   const base::RepeatingCallback<bool(const GURL&)>&));
  std::vector<std::unique_ptr<PasswordForm>> FillMatchingLogins(
      const PasswordFormDigest& form) override {
    return std::vector<std::unique_ptr<PasswordForm>>();
  }
  MOCK_METHOD1(
      FillMatchingLoginsByPassword,
      std::vector<std::unique_ptr<PasswordForm>>(const std::u16string&));
  MOCK_METHOD1(FillAutofillableLogins,
               bool(std::vector<std::unique_ptr<PasswordForm>>*));
  MOCK_METHOD1(FillBlocklistLogins,
               bool(std::vector<std::unique_ptr<PasswordForm>>*));
  MOCK_METHOD0(DeleteUndecryptableLogins, DatabaseCleanupResult());
  MOCK_METHOD1(NotifyLoginsChanged, void(const PasswordStoreChangeList&));
  MOCK_METHOD0(NotifyInsecureCredentialsChanged, void());
  MOCK_METHOD1(GetSiteStatsImpl,
               std::vector<InteractionsStats>(const GURL& origin_domain));
  MOCK_METHOD1(AddSiteStatsImpl, void(const InteractionsStats&));
  MOCK_METHOD1(RemoveSiteStatsImpl, void(const GURL&));
  MOCK_METHOD1(AddInsecureCredentialImpl,
               PasswordStoreChangeList(const InsecureCredential&));
  MOCK_METHOD3(RemoveInsecureCredentialsImpl,
               PasswordStoreChangeList(const std::string&,
                                       const std::u16string&,
                                       RemoveInsecureCredentialsReason));
  MOCK_METHOD0(GetAllInsecureCredentialsImpl,
               std::vector<InsecureCredential>());
  MOCK_METHOD1(
      GetMatchingInsecureCredentialsImpl,
      std::vector<InsecureCredential>(const std::string& signon_realm));
  MOCK_METHOD3(RemoveCompromisedCredentialsByUrlAndTimeImpl,
               bool(const base::RepeatingCallback<bool(const GURL&)>&,
                    base::Time,
                    base::Time));
  MOCK_METHOD1(AddFieldInfoImpl, void(const FieldInfo&));
  MOCK_METHOD0(GetAllFieldInfoImpl, std::vector<FieldInfo>());
  MOCK_METHOD2(RemoveFieldInfoByTimeImpl, void(base::Time, base::Time));
  MOCK_METHOD0(IsEmpty, bool());
  MOCK_METHOD1(GetAllLoginsWithAffiliationAndBrandingInformation,
               void(PasswordStoreConsumer*));

  MOCK_CONST_METHOD0(IsAbleToSavePasswords, bool());

  // TODO(crbug.bom/1226042): Remove this after PasswordStore no longer
  // inherits PasswordStoreSync.
  MOCK_METHOD(PasswordStoreChangeList,
              AddLoginSync,
              (const PasswordForm&, AddLoginError*),
              (override));
  MOCK_METHOD(bool,
              AddInsecureCredentialsSync,
              (base::span<const InsecureCredential>),
              (override));
  MOCK_METHOD(PasswordStoreChangeList,
              UpdateLoginSync,
              (const PasswordForm&, UpdateLoginError*),
              (override));
  MOCK_METHOD(bool,
              UpdateInsecureCredentialsSync,
              (const PasswordForm&, base::span<const InsecureCredential>),
              (override));
  MOCK_METHOD(PasswordStoreChangeList,
              RemoveLoginSync,
              (const PasswordForm& form),
              (override));
  MOCK_METHOD0(BeginTransaction, bool());
  MOCK_METHOD0(RollbackTransaction, void());
  MOCK_METHOD0(CommitTransaction, bool());
  MOCK_METHOD1(ReadAllLogins, FormRetrievalResult(PrimaryKeyToFormMap*));
  MOCK_METHOD1(ReadSecurityIssues,
               std::vector<InsecureCredential>(FormPrimaryKey));
  MOCK_METHOD1(RemoveLoginByPrimaryKeySync,
               PasswordStoreChangeList(FormPrimaryKey));
  MOCK_METHOD0(GetMetadataStore, PasswordStoreSync::MetadataStore*());
  MOCK_CONST_METHOD0(IsAccountStore, bool());
  MOCK_METHOD0(DeleteAndRecreateDatabaseFile, bool());

  PasswordStoreSync* GetSyncInterface() { return this; }

 protected:
  ~MockPasswordStore() override;

 private:
  // PasswordStore:
  scoped_refptr<base::SequencedTaskRunner> CreateBackgroundTaskRunner()
      const override;
  bool InitOnBackgroundSequence() override;
};

}  // namespace password_manager

#endif  // COMPONENTS_PASSWORD_MANAGER_CORE_BROWSER_MOCK_PASSWORD_STORE_H_