blob: c748b5e8d8c2619db90c0dec2368d7b04f77f4f5 [file] [log] [blame]
Avi Drissman60039d42022-09-13 21:49:051// Copyright 2014 The Chromium Authors
[email protected]4db043b2014-08-13 09:46:182// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
limasdfd70dc5ae2014-09-13 00:02:225#ifndef EXTENSIONS_BROWSER_MOCK_EXTENSION_SYSTEM_H_
6#define EXTENSIONS_BROWSER_MOCK_EXTENSION_SYSTEM_H_
[email protected]4db043b2014-08-13 09:46:187
Keishi Hattori0e45c022021-11-27 09:25:528#include "base/memory/raw_ptr.h"
Christopher Lamebb90202019-04-04 03:42:369#include "base/one_shot_event.h"
[email protected]4db043b2014-08-13 09:46:1810#include "components/keyed_service/content/browser_context_dependency_manager.h"
11#include "extensions/browser/extension_registry_factory.h"
12#include "extensions/browser/extension_system.h"
13#include "extensions/browser/extension_system_provider.h"
[email protected]4db043b2014-08-13 09:46:1814
15namespace extensions {
16
17// An empty ExtensionSystem for testing. Tests that need only specific
18// parts of ExtensionSystem should derive from this class and override
19// functions as needed. To use this, use
20// TestExtensionsBrowserClient::set_extension_system_factory
21// with the MockExtensionSystemFactory below.
22class MockExtensionSystem : public ExtensionSystem {
23 public:
Minh X. Nguyen30975342017-12-04 22:02:4124 using InstallUpdateCallback = ExtensionSystem::InstallUpdateCallback;
25
[email protected]4db043b2014-08-13 09:46:1826 explicit MockExtensionSystem(content::BrowserContext* context);
Peter Boström951cf77e2021-09-22 00:02:5927
28 MockExtensionSystem(const MockExtensionSystem&) = delete;
29 MockExtensionSystem& operator=(const MockExtensionSystem&) = delete;
30
dcheng9168b2f2014-10-21 12:38:2431 ~MockExtensionSystem() override;
[email protected]4db043b2014-08-13 09:46:1832
33 content::BrowserContext* browser_context() { return browser_context_; }
34
Evan Stade1ba771c2019-12-05 22:55:1935 void SetReady();
36
[email protected]4db043b2014-08-13 09:46:1837 // ExtensionSystem overrides:
dcheng9168b2f2014-10-21 12:38:2438 void InitForRegularProfile(bool extensions_enabled) override;
39 ExtensionService* extension_service() override;
dcheng9168b2f2014-10-21 12:38:2440 ManagementPolicy* management_policy() override;
rdevlin.croninf5863da2015-09-10 19:21:4541 ServiceWorkerManager* service_worker_manager() override;
Kelvin Jianga115a752021-02-11 03:04:0142 UserScriptManager* user_script_manager() override;
dcheng9168b2f2014-10-21 12:38:2443 StateStore* state_store() override;
44 StateStore* rules_store() override;
Kelvin Jiang31277362021-09-24 00:41:3345 StateStore* dynamic_user_scripts_store() override;
Louise Brett88346ae12021-09-09 04:47:3246 scoped_refptr<value_store::ValueStoreFactory> store_factory() override;
dcheng9168b2f2014-10-21 12:38:2447 InfoMap* info_map() override;
dcheng9168b2f2014-10-21 12:38:2448 QuotaService* quota_service() override;
treib926ee2d2015-08-06 10:55:4249 AppSorting* app_sorting() override;
Christopher Lamebb90202019-04-04 03:42:3650 const base::OneShotEvent& ready() const override;
Ghazale Hosseinabadi1d810e92020-06-01 20:43:0251 bool is_ready() const override;
dcheng9168b2f2014-10-21 12:38:2452 ContentVerifier* content_verifier() override;
dchengf5d241082016-04-21 03:43:1153 std::unique_ptr<ExtensionSet> GetDependentExtensions(
mostynb0eac4e1b2014-10-03 16:32:1954 const Extension* extension) override;
asargent631a99a2015-10-15 21:51:4855 void InstallUpdate(const std::string& extension_id,
Minh X. Nguyen30975342017-12-04 22:02:4156 const std::string& public_key,
57 const base::FilePath& temp_dir,
Minh X. Nguyen039eb992018-06-26 20:12:1358 bool install_immediately,
Minh X. Nguyen30975342017-12-04 22:02:4159 InstallUpdateCallback install_update_callback) override;
Bettina45e1a942020-04-10 22:29:4760 void PerformActionBasedOnOmahaAttributes(
61 const std::string& extension_id,
62 const base::Value& attributes) override;
Minh X. Nguyenb4fbf922018-01-29 18:28:1063 bool FinishDelayedInstallationIfReady(const std::string& extension_id,
64 bool install_immediately) override;
[email protected]4db043b2014-08-13 09:46:1865
66 private:
Keishi Hattori0e45c022021-11-27 09:25:5267 raw_ptr<content::BrowserContext> browser_context_;
Christopher Lamebb90202019-04-04 03:42:3668 base::OneShotEvent ready_;
[email protected]4db043b2014-08-13 09:46:1869};
70
71// A factory to create a MockExtensionSystem. Sample use:
72//
73// MockExtensionSystemFactory<MockExtensionSystemSubclass> factory;
74// TestExtensionsBrowserClient::set_extension_system_factory(factory);
75template <typename T>
76class MockExtensionSystemFactory : public ExtensionSystemProvider {
77 public:
78 MockExtensionSystemFactory()
79 : ExtensionSystemProvider(
80 "MockExtensionSystem",
81 BrowserContextDependencyManager::GetInstance()) {
82 DependsOn(ExtensionRegistryFactory::GetInstance());
83 }
84
Peter Boström951cf77e2021-09-22 00:02:5985 MockExtensionSystemFactory(const MockExtensionSystemFactory&) = delete;
86 MockExtensionSystemFactory& operator=(const MockExtensionSystemFactory&) =
87 delete;
88
nick0ec9f9f2015-04-23 16:43:0789 ~MockExtensionSystemFactory() override {}
[email protected]4db043b2014-08-13 09:46:1890
91 // BrowserContextKeyedServiceFactory overrides:
nick0ec9f9f2015-04-23 16:43:0792 KeyedService* BuildServiceInstanceFor(
mostynb0eac4e1b2014-10-03 16:32:1993 content::BrowserContext* context) const override {
[email protected]4db043b2014-08-13 09:46:1894 return new T(context);
95 }
Karandeep Bhatia670f41ee2018-01-24 23:56:1096 content::BrowserContext* GetBrowserContextToUse(
97 content::BrowserContext* context) const override {
98 // Separate instance in incognito.
99 return context;
100 }
[email protected]4db043b2014-08-13 09:46:18101
102 // ExtensionSystemProvider overrides:
nick0ec9f9f2015-04-23 16:43:07103 ExtensionSystem* GetForBrowserContext(
mostynb0eac4e1b2014-10-03 16:32:19104 content::BrowserContext* context) override {
[email protected]4db043b2014-08-13 09:46:18105 return static_cast<ExtensionSystem*>(
106 GetServiceForBrowserContext(context, true));
107 }
[email protected]4db043b2014-08-13 09:46:18108};
109
110} // namespace extensions
111
limasdfd70dc5ae2014-09-13 00:02:22112#endif // EXTENSIONS_BROWSER_MOCK_EXTENSION_SYSTEM_H_