| OLD | NEW | 
|---|
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be | 
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. | 
| 4 | 4 | 
| 5 #include "chrome/browser/extensions/test_extension_prefs.h" | 5 #include "chrome/browser/extensions/test_extension_prefs.h" | 
| 6 | 6 | 
| 7 #include "base/file_util.h" | 7 #include "base/file_util.h" | 
| 8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" | 
| 9 #include "base/message_loop_proxy.h" | 9 #include "base/message_loop_proxy.h" | 
| 10 #include "base/scoped_ptr.h" | 10 #include "base/scoped_ptr.h" | 
| 11 #include "base/values.h" | 11 #include "base/values.h" | 
| 12 #include "chrome/browser/browser_thread.h" | 12 #include "chrome/browser/browser_thread.h" | 
| 13 #include "chrome/browser/extensions/extension_pref_store.h" | 13 #include "chrome/browser/extensions/extension_pref_store.h" | 
|  | 14 #include "chrome/browser/extensions/extension_pref_value_map.h" | 
| 14 #include "chrome/browser/extensions/extension_prefs.h" | 15 #include "chrome/browser/extensions/extension_prefs.h" | 
| 15 #include "chrome/browser/prefs/pref_service.h" | 16 #include "chrome/browser/prefs/pref_service.h" | 
| 16 #include "chrome/browser/prefs/pref_service_mock_builder.h" | 17 #include "chrome/browser/prefs/pref_service_mock_builder.h" | 
| 17 #include "chrome/browser/prefs/pref_value_store.h" | 18 #include "chrome/browser/prefs/pref_value_store.h" | 
| 18 #include "chrome/common/extensions/extension.h" | 19 #include "chrome/common/extensions/extension.h" | 
| 19 #include "chrome/common/extensions/extension_constants.h" | 20 #include "chrome/common/extensions/extension_constants.h" | 
| 20 #include "chrome/common/json_pref_store.h" | 21 #include "chrome/common/json_pref_store.h" | 
| 21 #include "testing/gtest/include/gtest/gtest.h" | 22 #include "testing/gtest/include/gtest/gtest.h" | 
| 22 | 23 | 
| 23 namespace { | 24 namespace { | 
| 24 | 25 | 
| 25 // Mock ExtensionPrefs class with artificial clock to guarantee that no two | 26 // Mock ExtensionPrefs class with artificial clock to guarantee that no two | 
| 26 // extensions get the same installation time stamp and we can reliably | 27 // extensions get the same installation time stamp and we can reliably | 
| 27 // assert the installation order in the tests below. | 28 // assert the installation order in the tests below. | 
| 28 class MockExtensionPrefs : public ExtensionPrefs { | 29 class MockExtensionPrefs : public ExtensionPrefs { | 
| 29  public: | 30  public: | 
| 30   MockExtensionPrefs(PrefService* prefs, | 31   MockExtensionPrefs(PrefService* prefs, | 
| 31                      const FilePath& root_dir_, | 32                      const FilePath& root_dir, | 
| 32                      ExtensionPrefStore* pref_store) | 33                      ExtensionPrefValueMap* extension_pref_value_map) | 
| 33     : ExtensionPrefs(prefs, root_dir_, pref_store), | 34     : ExtensionPrefs(prefs, root_dir, extension_pref_value_map), | 
| 34       currentTime(base::Time::Now()) {} | 35       currentTime(base::Time::Now()) {} | 
| 35   ~MockExtensionPrefs() {} | 36   ~MockExtensionPrefs() {} | 
| 36 | 37 | 
| 37  protected: | 38  protected: | 
| 38   mutable base::Time currentTime; | 39   mutable base::Time currentTime; | 
| 39 | 40 | 
| 40   virtual base::Time GetCurrentTime() const { | 41   virtual base::Time GetCurrentTime() const { | 
| 41     currentTime += base::TimeDelta::FromSeconds(10); | 42     currentTime += base::TimeDelta::FromSeconds(10); | 
| 42     return currentTime; | 43     return currentTime; | 
| 43   } | 44   } | 
| (...skipping 19 matching lines...) Expand all  Loading... | 
| 63   if (pref_service_.get()) { | 64   if (pref_service_.get()) { | 
| 64     // The PrefService writes its persistent file on the file thread, so we | 65     // The PrefService writes its persistent file on the file thread, so we | 
| 65     // need to wait for any pending I/O to complete before creating a new | 66     // need to wait for any pending I/O to complete before creating a new | 
| 66     // PrefService. | 67     // PrefService. | 
| 67     MessageLoop file_loop; | 68     MessageLoop file_loop; | 
| 68     BrowserThread file_thread(BrowserThread::FILE, &file_loop); | 69     BrowserThread file_thread(BrowserThread::FILE, &file_loop); | 
| 69     pref_service_->SavePersistentPrefs(); | 70     pref_service_->SavePersistentPrefs(); | 
| 70     file_loop.RunAllPending(); | 71     file_loop.RunAllPending(); | 
| 71   } | 72   } | 
| 72 | 73 | 
| 73   ExtensionPrefStore* pref_store = new ExtensionPrefStore; | 74   extension_pref_value_map_.reset(new ExtensionPrefValueMap); | 
| 74   PrefServiceMockBuilder builder; | 75   PrefServiceMockBuilder builder; | 
| 75   builder.WithUserFilePrefs(preferences_file_); | 76   builder.WithUserFilePrefs(preferences_file_); | 
| 76   builder.WithExtensionPrefs(pref_store); | 77   builder.WithExtensionPrefs( | 
|  | 78       new ExtensionPrefStore(extension_pref_value_map_.get(), false)); | 
| 77   pref_service_.reset(builder.Create()); | 79   pref_service_.reset(builder.Create()); | 
| 78   ExtensionPrefs::RegisterUserPrefs(pref_service_.get()); | 80   ExtensionPrefs::RegisterUserPrefs(pref_service_.get()); | 
| 79   prefs_.reset(new MockExtensionPrefs(pref_service_.get(), temp_dir_.path(), | 81 | 
| 80                                       pref_store)); | 82   prefs_.reset(new MockExtensionPrefs(pref_service_.get(), | 
|  | 83                                       temp_dir_.path(), | 
|  | 84                                       extension_pref_value_map_.get())); | 
| 81 } | 85 } | 
| 82 | 86 | 
| 83 scoped_refptr<Extension> TestExtensionPrefs::AddExtension(std::string name) { | 87 scoped_refptr<Extension> TestExtensionPrefs::AddExtension(std::string name) { | 
| 84   DictionaryValue dictionary; | 88   DictionaryValue dictionary; | 
| 85   dictionary.SetString(extension_manifest_keys::kName, name); | 89   dictionary.SetString(extension_manifest_keys::kName, name); | 
| 86   dictionary.SetString(extension_manifest_keys::kVersion, "0.1"); | 90   dictionary.SetString(extension_manifest_keys::kVersion, "0.1"); | 
| 87   return AddExtensionWithManifest(dictionary, Extension::INTERNAL); | 91   return AddExtensionWithManifest(dictionary, Extension::INTERNAL); | 
| 88 } | 92 } | 
| 89 | 93 | 
| 90 scoped_refptr<Extension> TestExtensionPrefs::AddExtensionWithManifest( | 94 scoped_refptr<Extension> TestExtensionPrefs::AddExtensionWithManifest( | 
| (...skipping 12 matching lines...) Expand all  Loading... | 
| 103   const bool kInitialIncognitoEnabled = false; | 107   const bool kInitialIncognitoEnabled = false; | 
| 104   prefs_->OnExtensionInstalled(extension, Extension::ENABLED, | 108   prefs_->OnExtensionInstalled(extension, Extension::ENABLED, | 
| 105                                kInitialIncognitoEnabled); | 109                                kInitialIncognitoEnabled); | 
| 106   return extension; | 110   return extension; | 
| 107 } | 111 } | 
| 108 | 112 | 
| 109 std::string TestExtensionPrefs::AddExtensionAndReturnId(std::string name) { | 113 std::string TestExtensionPrefs::AddExtensionAndReturnId(std::string name) { | 
| 110   scoped_refptr<Extension> extension(AddExtension(name)); | 114   scoped_refptr<Extension> extension(AddExtension(name)); | 
| 111   return extension->id(); | 115   return extension->id(); | 
| 112 } | 116 } | 
|  | 117 | 
|  | 118 PrefService* TestExtensionPrefs::CreateIncognitoPrefService() const { | 
|  | 119   return pref_service_->CreateIncognitoPrefService( | 
|  | 120       new ExtensionPrefStore(extension_pref_value_map_.get(), true)); | 
|  | 121 } | 
| OLD | NEW | 
|---|