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
12class Profile;
13class TestingValueStore;
14
15namespace base {
16class CommandLine;
17class FilePath;
18class Time;
19}
20
21namespace content {
22class BrowserContext;
23}
24
25namespace extensions {
26class DeclarativeUserScriptMaster;
27class ExtensionPrefs;
28class RuntimeData;
29class SharedUserScriptMaster;
30class StandardManagementPolicyProvider;
31
32// Test ExtensionSystem, for use with TestingProfile.
33class TestExtensionSystem : public ExtensionSystem {
34 public:
35  explicit TestExtensionSystem(Profile* profile);
36  virtual ~TestExtensionSystem();
37
38  // KeyedService implementation.
39  virtual 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  // Creates a ProcessManager. If not invoked, the ProcessManager is NULL.
55  void CreateProcessManager();
56
57  // Allows the ProcessManager to be overriden, for example by a stub
58  // implementation. Takes ownership of |manager|.
59  void SetProcessManager(ProcessManager* manager);
60
61  void CreateSocketManager();
62
63  virtual void InitForRegularProfile(bool extensions_enabled) OVERRIDE {}
64  void SetExtensionService(ExtensionService* service);
65  virtual ExtensionService* extension_service() OVERRIDE;
66  virtual RuntimeData* runtime_data() OVERRIDE;
67  virtual ManagementPolicy* management_policy() OVERRIDE;
68  virtual SharedUserScriptMaster* shared_user_script_master() OVERRIDE;
69  virtual ProcessManager* process_manager() OVERRIDE;
70  virtual StateStore* state_store() OVERRIDE;
71  virtual StateStore* rules_store() OVERRIDE;
72  TestingValueStore* value_store() { return value_store_; }
73  virtual InfoMap* info_map() OVERRIDE;
74  virtual LazyBackgroundTaskQueue* lazy_background_task_queue() OVERRIDE;
75  void SetEventRouter(scoped_ptr<EventRouter> event_router);
76  virtual EventRouter* event_router() OVERRIDE;
77  virtual WarningService* warning_service() OVERRIDE;
78  virtual Blacklist* blacklist() OVERRIDE;
79  virtual ErrorConsole* error_console() OVERRIDE;
80  virtual InstallVerifier* install_verifier() OVERRIDE;
81  virtual QuotaService* quota_service() OVERRIDE;
82  virtual const OneShotEvent& ready() const OVERRIDE;
83  virtual ContentVerifier* content_verifier() OVERRIDE;
84  virtual scoped_ptr<ExtensionSet> GetDependentExtensions(
85      const Extension* extension) OVERRIDE;
86  virtual DeclarativeUserScriptMaster*
87      GetDeclarativeUserScriptMasterByExtension(
88          const ExtensionId& extension_id) OVERRIDE;
89
90  // Note that you probably want to use base::RunLoop().RunUntilIdle() right
91  // after this to run all the accumulated tasks.
92  void SetReady() { ready_.Signal(); }
93
94  // Factory method for tests to use with SetTestingProfile.
95  static KeyedService* Build(content::BrowserContext* profile);
96
97 protected:
98  Profile* profile_;
99
100 private:
101  scoped_ptr<StateStore> state_store_;
102  // A pointer to the TestingValueStore owned by |state_store_|.
103  TestingValueStore* value_store_;
104  ScopedVector<DeclarativeUserScriptMaster> declarative_user_script_masters_;
105  scoped_ptr<Blacklist> blacklist_;
106  scoped_ptr<ManagementPolicy> management_policy_;
107  scoped_ptr<RuntimeData> runtime_data_;
108  scoped_ptr<ExtensionService> extension_service_;
109  scoped_ptr<ProcessManager> process_manager_;
110  scoped_refptr<InfoMap> info_map_;
111  scoped_ptr<EventRouter> event_router_;
112  scoped_ptr<ErrorConsole> error_console_;
113  scoped_ptr<InstallVerifier> install_verifier_;
114  scoped_ptr<QuotaService> quota_service_;
115  OneShotEvent ready_;
116};
117
118}  // namespace extensions
119
120#endif  // CHROME_BROWSER_EXTENSIONS_TEST_EXTENSION_SYSTEM_H_
121