src/chrome/browser/extensions/test_extension_system.h

   1 // Copyright (c) 2012 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 CHROME_BROWSER_EXTENSIONS_TEST_EXTENSION_SYSTEM_H_
   6 #define CHROME_BROWSER_EXTENSIONS_TEST_EXTENSION_SYSTEM_H_
   7
   8 #include "base/memory/scoped_vector.h"
   9 #include "extensions/browser/extension_system.h"
  10 #include "extensions/common/one_shot_event.h"
  11
  12 class Profile;
  13 class TestingValueStore;
  14
  15 namespace base {
  16 class CommandLine;
  17 class FilePath;
  18 class Time;
  19 }
  20
  21 namespace content {
  22 class BrowserContext;
  23 }
  24
  25 namespace extensions {
  26 class DeclarativeUserScriptMaster;
  27 class ExtensionPrefs;
  28 class RuntimeData;
  29 class SharedUserScriptMaster;
  30 class StandardManagementPolicyProvider;
  31
  32 // Test ExtensionSystem, for use with TestingProfile.
  33 class TestExtensionSystem : public ExtensionSystem {
  34  public:
  35   explicit TestExtensionSystem(Profile* profile);
  36   ~TestExtensionSystem() override;
  37
  38   // KeyedService implementation.
  39   void Shutdown() override;
  40
  41   // Creates an ExtensionPrefs with the testing profile and returns it.
  42   // Useful for tests that need to modify prefs before creating the
  43   // ExtensionService.
  44   ExtensionPrefs* CreateExtensionPrefs(const base::CommandLine* command_line,
  45                                        const base::FilePath& install_directory);
  46
  47   // Creates an ExtensionService initialized with the testing profile and
  48   // returns it, and creates ExtensionPrefs if it hasn't been created yet.
  49   ExtensionService* CreateExtensionService(
  50       const base::CommandLine* command_line,
  51       const base::FilePath& install_directory,
  52       bool autoupdate_enabled);
  53
  54   void CreateSocketManager();
  55
  56   void InitForRegularProfile(bool extensions_enabled) override {}
  57   void SetExtensionService(ExtensionService* service);
  58   ExtensionService* extension_service() override;
  59   RuntimeData* runtime_data() override;
  60   ManagementPolicy* management_policy() override;
  61   SharedUserScriptMaster* shared_user_script_master() override;
  62   StateStore* state_store() override;
  63   StateStore* rules_store() override;
  64   TestingValueStore* value_store() { return value_store_; }
  65   InfoMap* info_map() override;
  66   LazyBackgroundTaskQueue* lazy_background_task_queue() override;
  67   void SetEventRouter(scoped_ptr<EventRouter> event_router);
  68   EventRouter* event_router() override;
  69   ErrorConsole* error_console() override;
  70   InstallVerifier* install_verifier() override;
  71   QuotaService* quota_service() override;
  72   const OneShotEvent& ready() const override;
  73   ContentVerifier* content_verifier() override;
  74   scoped_ptr<ExtensionSet> GetDependentExtensions(
  75       const Extension* extension) override;
  76   DeclarativeUserScriptMaster* GetDeclarativeUserScriptMasterByExtension(
  77       const ExtensionId& extension_id) override;
  78
  79   // Note that you probably want to use base::RunLoop().RunUntilIdle() right
  80   // after this to run all the accumulated tasks.
  81   void SetReady() { ready_.Signal(); }
  82
  83   // Factory method for tests to use with SetTestingProfile.
  84   static KeyedService* Build(content::BrowserContext* profile);
  85
  86  protected:
  87   Profile* profile_;
  88
  89  private:
  90   scoped_ptr<StateStore> state_store_;
  91   // A pointer to the TestingValueStore owned by |state_store_|.
  92   TestingValueStore* value_store_;
  93   ScopedVector<DeclarativeUserScriptMaster> declarative_user_script_masters_;
  94   scoped_ptr<ManagementPolicy> management_policy_;
  95   scoped_ptr<RuntimeData> runtime_data_;
  96   scoped_ptr<ExtensionService> extension_service_;
  97   scoped_refptr<InfoMap> info_map_;
  98   scoped_ptr<EventRouter> event_router_;
  99   scoped_ptr<ErrorConsole> error_console_;
 100   scoped_ptr<InstallVerifier> install_verifier_;
 101   scoped_ptr<QuotaService> quota_service_;
 102   OneShotEvent ready_;
 103 };
 104
 105 }  // namespace extensions
 106
 107 #endif  // CHROME_BROWSER_EXTENSIONS_TEST_EXTENSION_SYSTEM_H_