| 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/prefs/json_pref_store.h" | 13 #include "base/prefs/json_pref_store.h" |
| 14 #include "base/run_loop.h" |
| 15 #include "base/sequenced_task_runner.h" |
| 14 #include "base/synchronization/waitable_event.h" | 16 #include "base/synchronization/waitable_event.h" |
| 15 #include "base/values.h" | 17 #include "base/values.h" |
| 16 #include "chrome/browser/extensions/extension_pref_store.h" | 18 #include "chrome/browser/extensions/extension_pref_store.h" |
| 17 #include "chrome/browser/extensions/extension_pref_value_map.h" | 19 #include "chrome/browser/extensions/extension_pref_value_map.h" |
| 18 #include "chrome/browser/extensions/extension_prefs.h" | 20 #include "chrome/browser/extensions/extension_prefs.h" |
| 19 #include "chrome/browser/prefs/pref_service.h" | 21 #include "chrome/browser/prefs/pref_service.h" |
| 20 #include "chrome/browser/prefs/pref_service_mock_builder.h" | 22 #include "chrome/browser/prefs/pref_service_mock_builder.h" |
| 21 #include "chrome/browser/prefs/pref_value_store.h" | 23 #include "chrome/browser/prefs/pref_value_store.h" |
| 22 #include "chrome/common/extensions/extension.h" | 24 #include "chrome/common/extensions/extension.h" |
| 23 #include "chrome/common/extensions/extension_manifest_constants.h" | 25 #include "chrome/common/extensions/extension_manifest_constants.h" |
| 24 #include "content/public/browser/browser_thread.h" | 26 #include "content/public/browser/browser_thread.h" |
| 25 #include "sync/api/string_ordinal.h" | 27 #include "sync/api/string_ordinal.h" |
| 26 #include "testing/gtest/include/gtest/gtest.h" | 28 #include "testing/gtest/include/gtest/gtest.h" |
| 27 | 29 |
| 28 using content::BrowserThread; | 30 using content::BrowserThread; |
| 29 | 31 |
| 30 namespace extensions { | 32 namespace extensions { |
| 31 | 33 |
| 32 namespace { | 34 namespace { |
| 33 | 35 |
| 36 void DoNothing() {} |
| 37 |
| 34 // Mock ExtensionPrefs class with artificial clock to guarantee that no two | 38 // Mock ExtensionPrefs class with artificial clock to guarantee that no two |
| 35 // extensions get the same installation time stamp and we can reliably | 39 // extensions get the same installation time stamp and we can reliably |
| 36 // assert the installation order in the tests below. | 40 // assert the installation order in the tests below. |
| 37 class MockExtensionPrefs : public ExtensionPrefs { | 41 class MockExtensionPrefs : public ExtensionPrefs { |
| 38 public: | 42 public: |
| 39 MockExtensionPrefs(PrefService* prefs, | 43 MockExtensionPrefs(PrefService* prefs, |
| 40 const FilePath& root_dir, | 44 const FilePath& root_dir, |
| 41 ExtensionPrefValueMap* extension_pref_value_map) | 45 ExtensionPrefValueMap* extension_pref_value_map) |
| 42 : ExtensionPrefs(prefs, root_dir, extension_pref_value_map), | 46 : ExtensionPrefs(prefs, root_dir, extension_pref_value_map), |
| 43 currentTime(base::Time::Now()) {} | 47 currentTime(base::Time::Now()) {} |
| 44 ~MockExtensionPrefs() {} | 48 ~MockExtensionPrefs() {} |
| 45 | 49 |
| 46 protected: | 50 protected: |
| 47 mutable base::Time currentTime; | 51 mutable base::Time currentTime; |
| 48 | 52 |
| 49 virtual base::Time GetCurrentTime() const { | 53 virtual base::Time GetCurrentTime() const { |
| 50 currentTime += base::TimeDelta::FromSeconds(10); | 54 currentTime += base::TimeDelta::FromSeconds(10); |
| 51 return currentTime; | 55 return currentTime; |
| 52 } | 56 } |
| 53 }; | 57 }; |
| 54 | 58 |
| 55 } // namespace | 59 } // namespace |
| 56 | 60 |
| 57 TestExtensionPrefs::TestExtensionPrefs() | 61 TestExtensionPrefs::TestExtensionPrefs( |
| 58 : pref_service_(NULL), | 62 base::SequencedTaskRunner* task_runner) : pref_service_(NULL), |
| 59 extensions_disabled_(false) { | 63 task_runner_(task_runner), |
| 64 extensions_disabled_(false) { |
| 60 EXPECT_TRUE(temp_dir_.CreateUniqueTempDir()); | 65 EXPECT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| 61 preferences_file_ = temp_dir_.path().AppendASCII("Preferences"); | 66 preferences_file_ = temp_dir_.path().AppendASCII("Preferences"); |
| 62 extensions_dir_ = temp_dir_.path().AppendASCII("Extensions"); | 67 extensions_dir_ = temp_dir_.path().AppendASCII("Extensions"); |
| 63 EXPECT_TRUE(file_util::CreateDirectory(extensions_dir_)); | 68 EXPECT_TRUE(file_util::CreateDirectory(extensions_dir_)); |
| 64 | 69 |
| 65 RecreateExtensionPrefs(); | 70 RecreateExtensionPrefs(); |
| 66 } | 71 } |
| 67 | 72 |
| 68 TestExtensionPrefs::~TestExtensionPrefs() {} | 73 TestExtensionPrefs::~TestExtensionPrefs() { |
| 74 } |
| 69 | 75 |
| 70 void TestExtensionPrefs::RecreateExtensionPrefs() { | 76 void TestExtensionPrefs::RecreateExtensionPrefs() { |
| 71 // We persist and reload the PrefService's PrefStores because this process | 77 // We persist and reload the PrefService's PrefStores because this process |
| 72 // deletes all empty dictionaries. The ExtensionPrefs implementation | 78 // deletes all empty dictionaries. The ExtensionPrefs implementation |
| 73 // needs to be able to handle this situation. | 79 // needs to be able to handle this situation. |
| 74 if (pref_service_.get()) { | 80 if (pref_service_.get()) { |
| 75 // The PrefService writes its persistent file on the file thread, so we | 81 // Commit a pending write (which posts a task to task_runner_) and wait for |
| 76 // need to wait for any pending I/O to complete before creating a new | 82 // it to finish. |
| 77 // PrefService. | 83 pref_service_->CommitPendingWrite(); |
| 78 base::WaitableEvent io_finished(false, false); | 84 base::RunLoop run_loop; |
| 79 pref_service_-> CommitPendingWrite(); | 85 ASSERT_TRUE( |
| 80 EXPECT_TRUE(BrowserThread::PostTask( | 86 task_runner_->PostTaskAndReply( |
| 81 BrowserThread::FILE, | 87 FROM_HERE, |
| 82 FROM_HERE, | 88 base::Bind(&DoNothing), |
| 83 base::Bind(&base::WaitableEvent::Signal, | 89 run_loop.QuitClosure())); |
| 84 base::Unretained(&io_finished)))); | 90 run_loop.Run(); |
| 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 } | 91 } |
| 94 | 92 |
| 95 extension_pref_value_map_.reset(new ExtensionPrefValueMap); | 93 extension_pref_value_map_.reset(new ExtensionPrefValueMap); |
| 96 PrefServiceMockBuilder builder; | 94 PrefServiceMockBuilder builder; |
| 97 builder.WithUserFilePrefs(preferences_file_); | 95 builder.WithUserFilePrefs(preferences_file_, task_runner_); |
| 98 builder.WithExtensionPrefs( | 96 builder.WithExtensionPrefs( |
| 99 new ExtensionPrefStore(extension_pref_value_map_.get(), false)); | 97 new ExtensionPrefStore(extension_pref_value_map_.get(), false)); |
| 100 pref_service_.reset(builder.Create()); | 98 pref_service_.reset(builder.Create()); |
| 101 ExtensionPrefs::RegisterUserPrefs(pref_service_.get()); | 99 ExtensionPrefs::RegisterUserPrefs(pref_service_.get()); |
| 102 | 100 |
| 103 prefs_.reset(new MockExtensionPrefs(pref_service_.get(), | 101 prefs_.reset(new MockExtensionPrefs(pref_service_.get(), |
| 104 temp_dir_.path(), | 102 temp_dir_.path(), |
| 105 extension_pref_value_map_.get())); | 103 extension_pref_value_map_.get())); |
| 106 prefs_->Init(extensions_disabled_); | 104 prefs_->Init(extensions_disabled_); |
| 107 } | 105 } |
| (...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 158 PrefService* TestExtensionPrefs::CreateIncognitoPrefService() const { | 156 PrefService* TestExtensionPrefs::CreateIncognitoPrefService() const { |
| 159 return pref_service_->CreateIncognitoPrefService( | 157 return pref_service_->CreateIncognitoPrefService( |
| 160 new ExtensionPrefStore(extension_pref_value_map_.get(), true)); | 158 new ExtensionPrefStore(extension_pref_value_map_.get(), true)); |
| 161 } | 159 } |
| 162 | 160 |
| 163 void TestExtensionPrefs::set_extensions_disabled(bool extensions_disabled) { | 161 void TestExtensionPrefs::set_extensions_disabled(bool extensions_disabled) { |
| 164 extensions_disabled_ = extensions_disabled; | 162 extensions_disabled_ = extensions_disabled; |
| 165 } | 163 } |
| 166 | 164 |
| 167 } // namespace extensions | 165 } // namespace extensions |
| OLD | NEW |