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