chrome/browser/web_applications/test/fake_install_finalizer.h

// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_WEB_APPLICATIONS_TEST_FAKE_INSTALL_FINALIZER_H_
#define CHROME_BROWSER_WEB_APPLICATIONS_TEST_FAKE_INSTALL_FINALIZER_H_

#include <map>
#include <memory>
#include <set>

#include "base/containers/flat_set.h"
#include "chrome/browser/web_applications/web_app_install_finalizer.h"
#include "third_party/abseil-cpp/absl/types/optional.h"

struct WebAppInstallInfo;

namespace web_app {

class FakeInstallFinalizer final : public WebAppInstallFinalizer {
 public:
  // Returns what would be the AppId if an app is installed with |url|.
  static AppId GetAppIdForUrl(const GURL& url);

  FakeInstallFinalizer();
  FakeInstallFinalizer(const FakeInstallFinalizer&) = delete;
  FakeInstallFinalizer& operator=(const FakeInstallFinalizer&) = delete;
  ~FakeInstallFinalizer() override;

  // WebAppInstallFinalizer:
  void FinalizeInstall(const WebAppInstallInfo& web_app_info,
                       const FinalizeOptions& options,
                       InstallFinalizedCallback callback) override;
  void FinalizeUpdate(const WebAppInstallInfo& web_app_info,
                      InstallFinalizedCallback callback) override;
  void UninstallExternalWebApp(const AppId& app_id,
                               WebAppManagement::Type source,
                               webapps::WebappUninstallSource uninstall_surface,
                               UninstallWebAppCallback callback) override;
  void UninstallExternalWebAppByUrl(
      const GURL& app_url,
      WebAppManagement::Type source,
      webapps::WebappUninstallSource uninstall_surface,
      UninstallWebAppCallback callback) override;
  void UninstallWebApp(const AppId& app_id,
                       webapps::WebappUninstallSource uninstall_source,
                       UninstallWebAppCallback callback) override;
  bool CanReparentTab(const AppId& app_id,
                      bool shortcut_created) const override;
  void ReparentTab(const AppId& app_id,
                   bool shortcut_created,
                   content::WebContents* web_contents) override;

  void SetNextFinalizeInstallResult(const AppId& app_id,
                                    webapps::InstallResultCode code);
  void SetNextUninstallExternalWebAppResult(const GURL& app_url,
                                            webapps::UninstallResultCode code);

  // Uninstall the app and add |app_id| to the map of external extensions
  // uninstalled by the user. May be called on an app that isn't installed to
  // simulate that the app was uninstalled previously.
  void SimulateExternalAppUninstalledByUser(const AppId& app_id);

  bool WasPreinstalledWebAppUninstalled(const AppId& app_id);

  std::unique_ptr<WebAppInstallInfo> web_app_info() {
    return std::move(web_app_info_copy_);
  }

  const std::vector<FinalizeOptions>& finalize_options_list() const {
    return finalize_options_list_;
  }

  const std::vector<GURL>& uninstall_external_web_app_urls() const {
    return uninstall_external_web_app_urls_;
  }

  int num_reparent_tab_calls() const { return num_reparent_tab_calls_; }

 private:
  void Finalize(const WebAppInstallInfo& web_app_info,
                webapps::InstallResultCode code,
                InstallFinalizedCallback callback);

  std::unique_ptr<WebAppInstallInfo> web_app_info_copy_;
  std::vector<FinalizeOptions> finalize_options_list_;
  std::vector<GURL> uninstall_external_web_app_urls_;

  absl::optional<AppId> next_app_id_;
  absl::optional<webapps::InstallResultCode> next_result_code_;
  std::map<GURL, webapps::UninstallResultCode>
      next_uninstall_external_web_app_results_;
  std::set<AppId> user_uninstalled_external_apps_;

  int num_reparent_tab_calls_ = 0;
};

}  // namespace web_app

#endif  // CHROME_BROWSER_WEB_APPLICATIONS_TEST_FAKE_INSTALL_FINALIZER_H_