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(); |
akalin
2012/10/18 23:52:24
i think you can replace the FlushForTesting call w
zel
2012/10/19 01:20:32
Done.
| |
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 |