src/chrome/browser/extensions/extension_service_unittest.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_EXTENSION_SERVICE_UNITTEST_H_
   6 #define CHROME_BROWSER_EXTENSIONS_EXTENSION_SERVICE_UNITTEST_H_
   7
   8 #include "base/at_exit.h"
   9 #include "base/files/file_path.h"
  10 #include "base/files/scoped_temp_dir.h"
  11 #include "base/memory/ref_counted.h"
  12 #include "base/memory/scoped_ptr.h"
  13 #include "base/message_loop/message_loop.h"
  14 #include "chrome/browser/extensions/extension_service.h"
  15 #include "chrome/common/extensions/feature_switch.h"
  16 #include "content/public/test/test_browser_thread_bundle.h"
  17 #include "testing/gtest/include/gtest/gtest.h"
  18
  19 #if defined(OS_CHROMEOS)
  20 #include "chrome/browser/chromeos/login/user_manager.h"
  21 #include "chrome/browser/chromeos/settings/cros_settings.h"
  22 #include "chrome/browser/chromeos/settings/device_settings_service.h"
  23 #endif
  24
  25 class TestingProfile;
  26
  27 namespace extensions {
  28 class ManagementPolicy;
  29 }
  30
  31 class ExtensionServiceTestBase : public testing::Test {
  32  public:
  33   struct ExtensionServiceInitParams {
  34     base::FilePath profile_path;
  35     base::FilePath pref_file;
  36     base::FilePath extensions_install_dir;
  37     bool autoupdate_enabled;
  38     bool is_first_run;
  39     bool profile_is_managed;
  40
  41     ExtensionServiceInitParams();
  42   };
  43
  44   ExtensionServiceTestBase();
  45   virtual ~ExtensionServiceTestBase();
  46
  47   void InitializeExtensionService(const ExtensionServiceInitParams& params);
  48
  49   void InitializeInstalledExtensionService(
  50       const base::FilePath& prefs_file,
  51       const base::FilePath& source_install_dir);
  52
  53   void InitializeGoodInstalledExtensionService();
  54
  55   void InitializeEmptyExtensionService();
  56
  57   void InitializeExtensionProcessManager();
  58
  59   void InitializeExtensionServiceWithUpdater();
  60
  61   void InitializeExtensionSyncService();
  62
  63   static void SetUpTestCase();
  64
  65   virtual void SetUp() OVERRIDE;
  66   virtual void TearDown() OVERRIDE;
  67
  68   void set_extensions_enabled(bool enabled) {
  69     service_->set_extensions_enabled(enabled);
  70   }
  71
  72  protected:
  73   ExtensionServiceInitParams CreateDefaultInitParams();
  74
  75   // Destroying at_exit_manager_ will delete all LazyInstances, so it must come
  76   // after thread_bundle_ in the destruction order.
  77   base::ShadowingAtExitManager at_exit_manager_;
  78   content::TestBrowserThreadBundle thread_bundle_;
  79   base::ScopedTempDir temp_dir_;
  80   scoped_ptr<TestingProfile> profile_;
  81   base::FilePath extensions_install_dir_;
  82   base::FilePath data_dir_;
  83   // Managed by extensions::ExtensionSystemFactory.
  84   ExtensionService* service_;
  85   extensions::ManagementPolicy* management_policy_;
  86   scoped_ptr<ExtensionSyncService> extension_sync_service_;
  87   size_t expected_extensions_count_;
  88
  89 #if defined OS_CHROMEOS
  90   chromeos::ScopedTestDeviceSettingsService test_device_settings_service_;
  91   chromeos::ScopedTestCrosSettings test_cros_settings_;
  92   chromeos::ScopedTestUserManager test_user_manager_;
  93 #endif
  94 };
  95
  96 #endif  // CHROME_BROWSER_EXTENSIONS_EXTENSION_SERVICE_UNITTEST_H_