extensions/browser/mock_extension_system.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef EXTENSIONS_BROWSER_MOCK_EXTENSION_SYSTEM_H_
   6 #define EXTENSIONS_BROWSER_MOCK_EXTENSION_SYSTEM_H_
   7
   8 #include "components/keyed_service/content/browser_context_dependency_manager.h"
   9 #include "extensions/browser/extension_registry_factory.h"
  10 #include "extensions/browser/extension_system.h"
  11 #include "extensions/browser/extension_system_provider.h"
  12 #include "extensions/common/one_shot_event.h"
  13
  14 namespace extensions {
  15
  16 // An empty ExtensionSystem for testing. Tests that need only specific
  17 // parts of ExtensionSystem should derive from this class and override
  18 // functions as needed. To use this, use
  19 // TestExtensionsBrowserClient::set_extension_system_factory
  20 // with the MockExtensionSystemFactory below.
  21 class MockExtensionSystem : public ExtensionSystem {
  22  public:
  23   explicit MockExtensionSystem(content::BrowserContext* context);
  24   ~MockExtensionSystem() override;
  25
  26   content::BrowserContext* browser_context() { return browser_context_; }
  27
  28   void set_event_router(EventRouter* router) { event_router_ = router; }
  29
  30   // ExtensionSystem overrides:
  31   void InitForRegularProfile(bool extensions_enabled) override;
  32   ExtensionService* extension_service() override;
  33   RuntimeData* runtime_data() override;
  34   ManagementPolicy* management_policy() override;
  35   SharedUserScriptMaster* shared_user_script_master() override;
  36   StateStore* state_store() override;
  37   StateStore* rules_store() override;
  38   InfoMap* info_map() override;
  39   LazyBackgroundTaskQueue* lazy_background_task_queue() override;
  40   EventRouter* event_router() override;
  41   ErrorConsole* error_console() override;
  42   InstallVerifier* install_verifier() override;
  43   QuotaService* quota_service() override;
  44   const OneShotEvent& ready() const override;
  45   ContentVerifier* content_verifier() override;
  46   scoped_ptr<ExtensionSet> GetDependentExtensions(
  47       const Extension* extension) override;
  48   DeclarativeUserScriptMaster* GetDeclarativeUserScriptMasterByExtension(
  49       const ExtensionId& extension_id) override;
  50
  51  private:
  52   content::BrowserContext* browser_context_;
  53   OneShotEvent ready_;
  54
  55   // Not owned, defaults to null.
  56   EventRouter* event_router_;
  57
  58   DISALLOW_COPY_AND_ASSIGN(MockExtensionSystem);
  59 };
  60
  61 // A factory to create a MockExtensionSystem. Sample use:
  62 //
  63 // MockExtensionSystemFactory<MockExtensionSystemSubclass> factory;
  64 // TestExtensionsBrowserClient::set_extension_system_factory(factory);
  65 template <typename T>
  66 class MockExtensionSystemFactory : public ExtensionSystemProvider {
  67  public:
  68   MockExtensionSystemFactory()
  69       : ExtensionSystemProvider(
  70             "MockExtensionSystem",
  71             BrowserContextDependencyManager::GetInstance()) {
  72     DependsOn(ExtensionRegistryFactory::GetInstance());
  73   }
  74
  75   virtual ~MockExtensionSystemFactory() {}
  76
  77   // BrowserContextKeyedServiceFactory overrides:
  78   virtual KeyedService* BuildServiceInstanceFor(
  79       content::BrowserContext* context) const override {
  80     return new T(context);
  81   }
  82
  83   // ExtensionSystemProvider overrides:
  84   virtual ExtensionSystem* GetForBrowserContext(
  85       content::BrowserContext* context) override {
  86     return static_cast<ExtensionSystem*>(
  87         GetServiceForBrowserContext(context, true));
  88   }
  89
  90  private:
  91   DISALLOW_COPY_AND_ASSIGN(MockExtensionSystemFactory);
  92 };
  93
  94 }  // namespace extensions
  95
  96 #endif  // EXTENSIONS_BROWSER_MOCK_EXTENSION_SYSTEM_H_