| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/bind_helpers.h" | 8 #include "base/bind_helpers.h" |
| 9 #include "base/file_util.h" | 9 #include "base/file_util.h" |
| 10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
| 11 #include "base/message_loop.h" | 11 #include "base/message_loop.h" |
| 12 #include "base/message_loop_proxy.h" | 12 #include "base/message_loop_proxy.h" |
| 13 #include "base/synchronization/waitable_event.h" | 13 #include "base/synchronization/waitable_event.h" |
| 14 #include "base/threading/sequenced_worker_pool.h" |
| 14 #include "base/values.h" | 15 #include "base/values.h" |
| 15 #include "chrome/browser/extensions/extension_pref_store.h" | 16 #include "chrome/browser/extensions/extension_pref_store.h" |
| 16 #include "chrome/browser/extensions/extension_pref_value_map.h" | 17 #include "chrome/browser/extensions/extension_pref_value_map.h" |
| 17 #include "chrome/browser/extensions/extension_prefs.h" | 18 #include "chrome/browser/extensions/extension_prefs.h" |
| 18 #include "chrome/browser/prefs/pref_service.h" | 19 #include "chrome/browser/prefs/pref_service.h" |
| 19 #include "chrome/browser/prefs/pref_service_mock_builder.h" | 20 #include "chrome/browser/prefs/pref_service_mock_builder.h" |
| 20 #include "chrome/browser/prefs/pref_value_store.h" | 21 #include "chrome/browser/prefs/pref_value_store.h" |
| 21 #include "chrome/common/extensions/extension.h" | 22 #include "chrome/common/extensions/extension.h" |
| 22 #include "chrome/common/extensions/extension_manifest_constants.h" | 23 #include "chrome/common/extensions/extension_manifest_constants.h" |
| 23 #include "chrome/common/json_pref_store.h" | 24 #include "chrome/common/json_pref_store.h" |
| (...skipping 23 matching lines...) Expand all Loading... |
| 47 mutable base::Time currentTime; | 48 mutable base::Time currentTime; |
| 48 | 49 |
| 49 virtual base::Time GetCurrentTime() const { | 50 virtual base::Time GetCurrentTime() const { |
| 50 currentTime += base::TimeDelta::FromSeconds(10); | 51 currentTime += base::TimeDelta::FromSeconds(10); |
| 51 return currentTime; | 52 return currentTime; |
| 52 } | 53 } |
| 53 }; | 54 }; |
| 54 | 55 |
| 55 } // namespace | 56 } // namespace |
| 56 | 57 |
| 57 TestExtensionPrefs::TestExtensionPrefs() | 58 TestExtensionPrefs::TestExtensionPrefs( |
| 58 : pref_service_(NULL), | 59 base::SequencedWorkerPool* blocking_pool) : pref_service_(NULL), |
| 59 extensions_disabled_(false) { | 60 extensions_disabled_(false) { |
| 61 blocking_pool_ = blocking_pool; |
| 60 EXPECT_TRUE(temp_dir_.CreateUniqueTempDir()); | 62 EXPECT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| 61 preferences_file_ = temp_dir_.path().AppendASCII("Preferences"); | 63 preferences_file_ = temp_dir_.path().AppendASCII("Preferences"); |
| 62 extensions_dir_ = temp_dir_.path().AppendASCII("Extensions"); | 64 extensions_dir_ = temp_dir_.path().AppendASCII("Extensions"); |
| 63 EXPECT_TRUE(file_util::CreateDirectory(extensions_dir_)); | 65 EXPECT_TRUE(file_util::CreateDirectory(extensions_dir_)); |
| 64 | 66 |
| 65 RecreateExtensionPrefs(); | 67 RecreateExtensionPrefs(); |
| 66 } | 68 } |
| 67 | 69 |
| 68 TestExtensionPrefs::~TestExtensionPrefs() {} | 70 TestExtensionPrefs::~TestExtensionPrefs() { |
| 71 } |
| 69 | 72 |
| 70 void TestExtensionPrefs::RecreateExtensionPrefs() { | 73 void TestExtensionPrefs::RecreateExtensionPrefs() { |
| 71 // We persist and reload the PrefService's PrefStores because this process | 74 // We persist and reload the PrefService's PrefStores because this process |
| 72 // deletes all empty dictionaries. The ExtensionPrefs implementation | 75 // deletes all empty dictionaries. The ExtensionPrefs implementation |
| 73 // needs to be able to handle this situation. | 76 // needs to be able to handle this situation. |
| 74 if (pref_service_.get()) { | 77 if (pref_service_.get()) { |
| 75 // The PrefService writes its persistent file on the file thread, so we | 78 pref_service_->CommitPendingWrite(); |
| 76 // need to wait for any pending I/O to complete before creating a new | 79 blocking_pool_->FlushForTesting(); |
| 77 // PrefService. | |
| 78 base::WaitableEvent io_finished(false, false); | |
| 79 pref_service_-> CommitPendingWrite(); | |
| 80 EXPECT_TRUE(BrowserThread::PostTask( | |
| 81 BrowserThread::FILE, | |
| 82 FROM_HERE, | |
| 83 base::Bind(&base::WaitableEvent::Signal, | |
| 84 base::Unretained(&io_finished)))); | |
| 85 | |
| 86 // If the FILE thread is in fact the current thread (possible in testing | |
| 87 // scenarios), we have to ensure the task has a chance to run. If the FILE | |
| 88 // thread is a different thread, the test must ensure that thread is running | |
| 89 // (otherwise the Wait below will hang). | |
| 90 MessageLoop::current()->RunAllPending(); | |
| 91 | |
| 92 io_finished.Wait(); | |
| 93 } | 80 } |
| 94 | 81 |
| 95 extension_pref_value_map_.reset(new ExtensionPrefValueMap); | 82 extension_pref_value_map_.reset(new ExtensionPrefValueMap); |
| 96 PrefServiceMockBuilder builder; | 83 PrefServiceMockBuilder builder; |
| 97 builder.WithUserFilePrefs(preferences_file_); | 84 builder.WithUserFilePrefs(preferences_file_, blocking_pool_.get()); |
| 98 builder.WithExtensionPrefs( | 85 builder.WithExtensionPrefs( |
| 99 new ExtensionPrefStore(extension_pref_value_map_.get(), false)); | 86 new ExtensionPrefStore(extension_pref_value_map_.get(), false)); |
| 100 pref_service_.reset(builder.Create()); | 87 pref_service_.reset(builder.Create()); |
| 101 ExtensionPrefs::RegisterUserPrefs(pref_service_.get()); | 88 ExtensionPrefs::RegisterUserPrefs(pref_service_.get()); |
| 102 | 89 |
| 103 prefs_.reset(new MockExtensionPrefs(pref_service_.get(), | 90 prefs_.reset(new MockExtensionPrefs(pref_service_.get(), |
| 104 temp_dir_.path(), | 91 temp_dir_.path(), |
| 105 extension_pref_value_map_.get())); | 92 extension_pref_value_map_.get())); |
| 106 prefs_->Init(extensions_disabled_); | 93 prefs_->Init(extensions_disabled_); |
| 107 } | 94 } |
| (...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 158 PrefService* TestExtensionPrefs::CreateIncognitoPrefService() const { | 145 PrefService* TestExtensionPrefs::CreateIncognitoPrefService() const { |
| 159 return pref_service_->CreateIncognitoPrefService( | 146 return pref_service_->CreateIncognitoPrefService( |
| 160 new ExtensionPrefStore(extension_pref_value_map_.get(), true)); | 147 new ExtensionPrefStore(extension_pref_value_map_.get(), true)); |
| 161 } | 148 } |
| 162 | 149 |
| 163 void TestExtensionPrefs::set_extensions_disabled(bool extensions_disabled) { | 150 void TestExtensionPrefs::set_extensions_disabled(bool extensions_disabled) { |
| 164 extensions_disabled_ = extensions_disabled; | 151 extensions_disabled_ = extensions_disabled; |
| 165 } | 152 } |
| 166 | 153 |
| 167 } // namespace extensions | 154 } // namespace extensions |
| OLD | NEW |