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 "extensions/browser/extension_system.h"
   9 #include "extensions/common/one_shot_event.h"
  10
  11 class Profile;
  12 class TestingValueStore;
  13
  14 namespace base {
  15 class CommandLine;
  16 class FilePath;
  17 class Time;
  18 }
  19
  20 namespace content {
  21 class BrowserContext;
  22 }
  23
  24 namespace extensions {
  25 class ExtensionPrefs;
  26 class RuntimeData;
  27 class StandardManagementPolicyProvider;
  28
  29 // Test ExtensionSystem, for use with TestingProfile.
  30 class TestExtensionSystem : public ExtensionSystem {
  31  public:
  32   explicit TestExtensionSystem(Profile* profile);
  33   virtual ~TestExtensionSystem();
  34
  35   // KeyedService implementation.
  36   virtual void Shutdown() OVERRIDE;
  37
  38   // Creates an ExtensionPrefs with the testing profile and returns it.
  39   // Useful for tests that need to modify prefs before creating the
  40   // ExtensionService.
  41   ExtensionPrefs* CreateExtensionPrefs(const base::CommandLine* command_line,
  42                                        const base::FilePath& install_directory);
  43
  44   // Creates an ExtensionService initialized with the testing profile and
  45   // returns it, and creates ExtensionPrefs if it hasn't been created yet.
  46   ExtensionService* CreateExtensionService(
  47       const base::CommandLine* command_line,
  48       const base::FilePath& install_directory,
  49       bool autoupdate_enabled);
  50
  51   // Creates a ProcessManager. If not invoked, the ProcessManager is NULL.
  52   void CreateProcessManager();
  53
  54   // Allows the ProcessManager to be overriden, for example by a stub
  55   // implementation. Takes ownership of |manager|.
  56   void SetProcessManager(ProcessManager* manager);
  57
  58   void CreateSocketManager();
  59
  60   virtual void InitForRegularProfile(bool extensions_enabled) OVERRIDE {}
  61   void SetExtensionService(ExtensionService* service);
  62   virtual ExtensionService* extension_service() OVERRIDE;
  63   virtual RuntimeData* runtime_data() OVERRIDE;
  64   virtual ManagementPolicy* management_policy() OVERRIDE;
  65   virtual UserScriptMaster* user_script_master() OVERRIDE;
  66   virtual ProcessManager* process_manager() OVERRIDE;
  67   virtual StateStore* state_store() OVERRIDE;
  68   virtual StateStore* rules_store() OVERRIDE;
  69   TestingValueStore* value_store() { return value_store_; }
  70   virtual InfoMap* info_map() OVERRIDE;
  71   virtual LazyBackgroundTaskQueue* lazy_background_task_queue() OVERRIDE;
  72   void SetEventRouter(scoped_ptr<EventRouter> event_router);
  73   virtual EventRouter* event_router() OVERRIDE;
  74   virtual ExtensionWarningService* warning_service() OVERRIDE;
  75   virtual Blacklist* blacklist() OVERRIDE;
  76   virtual ErrorConsole* error_console() OVERRIDE;
  77   virtual InstallVerifier* install_verifier() OVERRIDE;
  78   virtual QuotaService* quota_service() OVERRIDE;
  79   virtual const OneShotEvent& ready() const OVERRIDE;
  80   virtual ContentVerifier* content_verifier() OVERRIDE;
  81
  82   void SetReady() {
  83     LOG(INFO) << "SetReady()";
  84     ready_.Signal();
  85   }
  86
  87   // Factory method for tests to use with SetTestingProfile.
  88   static KeyedService* Build(content::BrowserContext* profile);
  89
  90  protected:
  91   Profile* profile_;
  92
  93  private:
  94   scoped_ptr<StateStore> state_store_;
  95   // A pointer to the TestingValueStore owned by |state_store_|.
  96   TestingValueStore* value_store_;
  97   scoped_ptr<Blacklist> blacklist_;
  98   scoped_ptr<StandardManagementPolicyProvider>
  99       standard_management_policy_provider_;
 100   scoped_ptr<ManagementPolicy> management_policy_;
 101   scoped_ptr<RuntimeData> runtime_data_;
 102   scoped_ptr<ExtensionService> extension_service_;
 103   scoped_ptr<ProcessManager> process_manager_;
 104   scoped_refptr<InfoMap> info_map_;
 105   scoped_ptr<EventRouter> event_router_;
 106   scoped_ptr<ErrorConsole> error_console_;
 107   scoped_ptr<InstallVerifier> install_verifier_;
 108   scoped_ptr<QuotaService> quota_service_;
 109   OneShotEvent ready_;
 110 };
 111
 112 }  // namespace extensions
 113
 114 #endif  // CHROME_BROWSER_EXTENSIONS_TEST_EXTENSION_SYSTEM_H_