components/password_manager/core/browser/fake_form_fetcher.h

// Copyright 2016 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_FAKE_FORM_FETCHER_H_
#define COMPONENTS_PASSWORD_MANAGER_CORE_BROWSER_FAKE_FORM_FETCHER_H_

#include <set>
#include <vector>

#include "base/macros.h"
#include "components/password_manager/core/browser/form_fetcher.h"
#include "components/password_manager/core/browser/statistics_table.h"

namespace autofill {
struct PasswordForm;
}

namespace password_manager {

struct InteractionsStats;

// Test implementation of FormFetcher useful for simple fakes and as a base for
// mocks.
class FakeFormFetcher : public FormFetcher {
 public:
  FakeFormFetcher();

  ~FakeFormFetcher() override;

  // Registers consumers to be notified when results are set. Unlike the
  // production version, assumes that results have not arrived yet, i.e., one
  // has to first call AddConsumer and then SetNonFederated.
  void AddConsumer(Consumer* consumer) override;

  void RemoveConsumer(Consumer* consumer) override;

  // Returns State::WAITING if Fetch() was called after any Set* calls, and
  // State::NOT_WAITING otherwise.
  State GetState() const override;

  // Statistics for recent password bubble usage.
  const std::vector<InteractionsStats>& GetInteractionsStats() const override;

  void set_stats(const std::vector<InteractionsStats>& stats) {
    state_ = State::NOT_WAITING;
    stats_ = stats;
  }

  const std::vector<const autofill::PasswordForm*>& GetNonFederatedMatches()
      const override;

  const std::vector<const autofill::PasswordForm*>& GetFederatedMatches()
      const override;

  void set_federated(
      const std::vector<const autofill::PasswordForm*>& federated) {
    state_ = State::NOT_WAITING;
    federated_ = federated;
  }

  const std::vector<const autofill::PasswordForm*>& GetSuppressedHTTPSForms()
      const override;

  // The pointees in |suppressed_forms| must outlive the fetcher.
  void set_suppressed_https_forms(
      const std::vector<const autofill::PasswordForm*>& suppressed_forms) {
    suppressed_https_forms_ = suppressed_forms;
  }

  const std::vector<const autofill::PasswordForm*>&
  GetSuppressedPSLMatchingForms() const override;

  // The pointees in |suppressed_forms| must outlive the fetcher.
  void set_suppressed_psl_matching_forms(
      const std::vector<const autofill::PasswordForm*>& suppressed_forms) {
    suppressed_psl_matching_forms_ = suppressed_forms;
  }

  const std::vector<const autofill::PasswordForm*>&
  GetSuppressedSameOrganizationNameForms() const override;

  // The pointees in |suppressed_forms| must outlive the fetcher.
  void set_suppressed_same_organization_name_forms(
      const std::vector<const autofill::PasswordForm*>& suppressed_forms) {
    suppressed_same_organization_name_forms_ = suppressed_forms;
  }

  bool DidCompleteQueryingSuppressedForms() const override;

  void set_did_complete_querying_suppressed_forms(bool value) {
    did_complete_querying_suppressed_forms_ = value;
  }

  void SetNonFederated(
      const std::vector<const autofill::PasswordForm*>& non_federated,
      size_t filtered_count);

  // Only sets the internal state to WAITING, no call to PasswordStore.
  void Fetch() override;

  // Returns a new FakeFormFetcher.
  std::unique_ptr<FormFetcher> Clone() override;

 private:
  std::set<Consumer*> consumers_;
  State state_ = State::NOT_WAITING;
  std::vector<InteractionsStats> stats_;
  std::vector<const autofill::PasswordForm*> non_federated_;
  std::vector<const autofill::PasswordForm*> federated_;
  std::vector<const autofill::PasswordForm*> suppressed_https_forms_;
  std::vector<const autofill::PasswordForm*> suppressed_psl_matching_forms_;
  std::vector<const autofill::PasswordForm*>
      suppressed_same_organization_name_forms_;
  bool did_complete_querying_suppressed_forms_ = false;

  DISALLOW_COPY_AND_ASSIGN(FakeFormFetcher);
};

}  // namespace password_manager

#endif  // COMPONENTS_PASSWORD_MANAGER_CORE_BROWSER_FAKE_FORM_FETCHER_H_