chrome/browser/web_applications/test/fake_os_integration_manager.h

// Copyright 2020 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_OS_INTEGRATION_MANAGER_H_
#define CHROME_BROWSER_WEB_APPLICATIONS_TEST_FAKE_OS_INTEGRATION_MANAGER_H_

#include <map>
#include <optional>
#include <string_view>

#include "chrome/browser/web_applications/os_integration/os_integration_manager.h"
#include "components/webapps/common/web_app_id.h"

namespace web_app {

class WebAppFileHandlerManager;
class WebAppProtocolHandlerManager;

class FakeOsIntegrationManager : public OsIntegrationManager {
 public:
  FakeOsIntegrationManager(
      Profile* profile,
      std::unique_ptr<WebAppFileHandlerManager> file_handler_manager,
      std::unique_ptr<WebAppProtocolHandlerManager> protocol_handler_manager);
  ~FakeOsIntegrationManager() override;

  void SetNextCreateShortcutsResult(const webapps::AppId& app_id, bool success);

  FakeOsIntegrationManager* AsTestOsIntegrationManager() override;

  void SetAppExistingShortcuts(const GURL& app_url,
                               ShortcutLocations locations);
  void SetShortcutInfoForApp(const webapps::AppId& app_id,
                             std::unique_ptr<ShortcutInfo> shortcut_info);

  void GetShortcutInfoForAppFromRegistrar(
      const webapps::AppId& app_id,
      GetShortcutInfoCallback callback) override;
  void GetAppExistingShortCutLocation(
      ShortcutLocationCallback callback,
      std::unique_ptr<ShortcutInfo> shortcut_info) override;

 private:
  std::unique_ptr<OsIntegrationManager::ScopedSuppressForTesting>
      scoped_suppress_;
  std::map<webapps::AppId, std::unique_ptr<ShortcutInfo>> shortcut_info_map_;
  std::map<GURL, ShortcutLocations> existing_shortcut_locations_;
};
}  // namespace web_app

#endif  // CHROME_BROWSER_WEB_APPLICATIONS_TEST_FAKE_OS_INTEGRATION_MANAGER_H_