1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/web_applications/test/fake_os_integration_manager.h"
#include <string_view>
#include "base/containers/contains.h"
#include "base/task/sequenced_task_runner.h"
#include "base/test/bind.h"
#include "chrome/browser/web_applications/os_integration/os_integration_manager.h"
#include "chrome/browser/web_applications/os_integration/web_app_file_handler_manager.h"
#include "chrome/browser/web_applications/os_integration/web_app_protocol_handler_manager.h"
#include "chrome/browser/web_applications/web_app_constants.h"
namespace web_app {
FakeOsIntegrationManager::FakeOsIntegrationManager(
Profile* profile,
std::unique_ptr<WebAppFileHandlerManager> file_handler_manager,
std::unique_ptr<WebAppProtocolHandlerManager> protocol_handler_manager)
: OsIntegrationManager(profile,
std::move(file_handler_manager),
std::move(protocol_handler_manager)),
scoped_suppress_(
std::make_unique<OsIntegrationManager::ScopedSuppressForTesting>()) {
if (!has_file_handler_manager()) {
set_file_handler_manager(
std::make_unique<WebAppFileHandlerManager>(profile));
}
}
FakeOsIntegrationManager::~FakeOsIntegrationManager() = default;
FakeOsIntegrationManager*
FakeOsIntegrationManager::AsTestOsIntegrationManager() {
return this;
}
void FakeOsIntegrationManager::SetAppExistingShortcuts(
const GURL& app_url,
ShortcutLocations locations) {
existing_shortcut_locations_[app_url] = locations;
}
void FakeOsIntegrationManager::SetShortcutInfoForApp(
const webapps::AppId& app_id,
std::unique_ptr<ShortcutInfo> shortcut_info) {
shortcut_info_map_[app_id] = std::move(shortcut_info);
}
void FakeOsIntegrationManager::GetShortcutInfoForAppFromRegistrar(
const webapps::AppId& app_id,
GetShortcutInfoCallback callback) {
auto it = shortcut_info_map_.find(app_id);
if (it != shortcut_info_map_.end()) {
std::move(callback).Run(std::move(it->second));
shortcut_info_map_.erase(it);
} else {
std::move(callback).Run(nullptr);
}
}
void FakeOsIntegrationManager::GetAppExistingShortCutLocation(
ShortcutLocationCallback callback,
std::unique_ptr<ShortcutInfo> shortcut_info) {
ShortcutLocations locations;
if (existing_shortcut_locations_.find(shortcut_info->url) !=
existing_shortcut_locations_.end()) {
locations = existing_shortcut_locations_[shortcut_info->url];
}
std::move(callback).Run(locations);
}
} // namespace web_app
|