OLD | NEW |
| (Empty) |
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 #include "base/file_util.h" | |
6 #include "base/memory/ref_counted.h" | |
7 #include "base/memory/scoped_ptr.h" | |
8 #include "base/message_loop.h" | |
9 #include "base/message_loop_proxy.h" | |
10 #include "base/path_service.h" | |
11 #include "base/scoped_temp_dir.h" | |
12 #include "base/string_number_conversions.h" | |
13 #include "base/string_util.h" | |
14 #include "base/threading/thread.h" | |
15 #include "base/utf_string_conversions.h" | |
16 #include "base/values.h" | |
17 #include "chrome/common/chrome_paths.h" | |
18 #include "chrome/common/json_pref_store.h" | |
19 #include "chrome/common/pref_names.h" | |
20 #include "testing/gmock/include/gmock/gmock.h" | |
21 #include "testing/gtest/include/gtest/gtest.h" | |
22 | |
23 namespace { | |
24 | |
25 class MockPrefStoreObserver : public PrefStore::Observer { | |
26 public: | |
27 MOCK_METHOD1(OnPrefValueChanged, void (const std::string&)); | |
28 MOCK_METHOD1(OnInitializationCompleted, void (bool)); | |
29 }; | |
30 | |
31 class MockReadErrorDelegate : public PersistentPrefStore::ReadErrorDelegate { | |
32 public: | |
33 MOCK_METHOD1(OnError, void(PersistentPrefStore::PrefReadError)); | |
34 }; | |
35 | |
36 } // namespace | |
37 | |
38 class JsonPrefStoreTest : public testing::Test { | |
39 protected: | |
40 virtual void SetUp() { | |
41 message_loop_proxy_ = base::MessageLoopProxy::current(); | |
42 | |
43 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | |
44 | |
45 ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &data_dir_)); | |
46 data_dir_ = data_dir_.AppendASCII("pref_service"); | |
47 ASSERT_TRUE(file_util::PathExists(data_dir_)); | |
48 } | |
49 | |
50 // The path to temporary directory used to contain the test operations. | |
51 ScopedTempDir temp_dir_; | |
52 // The path to the directory where the test data is stored. | |
53 FilePath data_dir_; | |
54 // A message loop that we can use as the file thread message loop. | |
55 MessageLoop message_loop_; | |
56 scoped_refptr<base::MessageLoopProxy> message_loop_proxy_; | |
57 }; | |
58 | |
59 // Test fallback behavior for a nonexistent file. | |
60 TEST_F(JsonPrefStoreTest, NonExistentFile) { | |
61 FilePath bogus_input_file = data_dir_.AppendASCII("read.txt"); | |
62 ASSERT_FALSE(file_util::PathExists(bogus_input_file)); | |
63 scoped_refptr<JsonPrefStore> pref_store = | |
64 new JsonPrefStore(bogus_input_file, message_loop_proxy_.get()); | |
65 EXPECT_EQ(PersistentPrefStore::PREF_READ_ERROR_NO_FILE, | |
66 pref_store->ReadPrefs()); | |
67 EXPECT_FALSE(pref_store->ReadOnly()); | |
68 } | |
69 | |
70 // Test fallback behavior for an invalid file. | |
71 TEST_F(JsonPrefStoreTest, InvalidFile) { | |
72 FilePath invalid_file_original = data_dir_.AppendASCII("invalid.json"); | |
73 FilePath invalid_file = temp_dir_.path().AppendASCII("invalid.json"); | |
74 ASSERT_TRUE(file_util::CopyFile(invalid_file_original, invalid_file)); | |
75 scoped_refptr<JsonPrefStore> pref_store = | |
76 new JsonPrefStore(invalid_file, message_loop_proxy_.get()); | |
77 EXPECT_EQ(PersistentPrefStore::PREF_READ_ERROR_JSON_PARSE, | |
78 pref_store->ReadPrefs()); | |
79 EXPECT_FALSE(pref_store->ReadOnly()); | |
80 | |
81 // The file should have been moved aside. | |
82 EXPECT_FALSE(file_util::PathExists(invalid_file)); | |
83 FilePath moved_aside = temp_dir_.path().AppendASCII("invalid.bad"); | |
84 EXPECT_TRUE(file_util::PathExists(moved_aside)); | |
85 EXPECT_TRUE(file_util::TextContentsEqual(invalid_file_original, | |
86 moved_aside)); | |
87 } | |
88 | |
89 // This function is used to avoid code duplication while testing synchronous and | |
90 // asynchronous version of the JsonPrefStore loading. | |
91 void RunBasicJsonPrefStoreTest(JsonPrefStore *pref_store, | |
92 const FilePath& output_file, | |
93 const FilePath& golden_output_file) { | |
94 const char kNewWindowsInTabs[] = "tabs.new_windows_in_tabs"; | |
95 const char kMaxTabs[] = "tabs.max_tabs"; | |
96 const char kLongIntPref[] = "long_int.pref"; | |
97 | |
98 std::string cnn("http://www.cnn.com"); | |
99 | |
100 const Value* actual; | |
101 EXPECT_EQ(PrefStore::READ_OK, | |
102 pref_store->GetValue(prefs::kHomePage, &actual)); | |
103 std::string string_value; | |
104 EXPECT_TRUE(actual->GetAsString(&string_value)); | |
105 EXPECT_EQ(cnn, string_value); | |
106 | |
107 const char kSomeDirectory[] = "some_directory"; | |
108 | |
109 EXPECT_EQ(PrefStore::READ_OK, pref_store->GetValue(kSomeDirectory, &actual)); | |
110 FilePath::StringType path; | |
111 EXPECT_TRUE(actual->GetAsString(&path)); | |
112 EXPECT_EQ(FilePath::StringType(FILE_PATH_LITERAL("/usr/local/")), path); | |
113 FilePath some_path(FILE_PATH_LITERAL("/usr/sbin/")); | |
114 | |
115 pref_store->SetValue(kSomeDirectory, | |
116 Value::CreateStringValue(some_path.value())); | |
117 EXPECT_EQ(PrefStore::READ_OK, pref_store->GetValue(kSomeDirectory, &actual)); | |
118 EXPECT_TRUE(actual->GetAsString(&path)); | |
119 EXPECT_EQ(some_path.value(), path); | |
120 | |
121 // Test reading some other data types from sub-dictionaries. | |
122 EXPECT_EQ(PrefStore::READ_OK, | |
123 pref_store->GetValue(kNewWindowsInTabs, &actual)); | |
124 bool boolean = false; | |
125 EXPECT_TRUE(actual->GetAsBoolean(&boolean)); | |
126 EXPECT_TRUE(boolean); | |
127 | |
128 pref_store->SetValue(kNewWindowsInTabs, | |
129 Value::CreateBooleanValue(false)); | |
130 EXPECT_EQ(PrefStore::READ_OK, | |
131 pref_store->GetValue(kNewWindowsInTabs, &actual)); | |
132 EXPECT_TRUE(actual->GetAsBoolean(&boolean)); | |
133 EXPECT_FALSE(boolean); | |
134 | |
135 EXPECT_EQ(PrefStore::READ_OK, pref_store->GetValue(kMaxTabs, &actual)); | |
136 int integer = 0; | |
137 EXPECT_TRUE(actual->GetAsInteger(&integer)); | |
138 EXPECT_EQ(20, integer); | |
139 pref_store->SetValue(kMaxTabs, Value::CreateIntegerValue(10)); | |
140 EXPECT_EQ(PrefStore::READ_OK, pref_store->GetValue(kMaxTabs, &actual)); | |
141 EXPECT_TRUE(actual->GetAsInteger(&integer)); | |
142 EXPECT_EQ(10, integer); | |
143 | |
144 pref_store->SetValue(kLongIntPref, | |
145 Value::CreateStringValue( | |
146 base::Int64ToString(214748364842LL))); | |
147 EXPECT_EQ(PrefStore::READ_OK, pref_store->GetValue(kLongIntPref, &actual)); | |
148 EXPECT_TRUE(actual->GetAsString(&string_value)); | |
149 int64 value; | |
150 base::StringToInt64(string_value, &value); | |
151 EXPECT_EQ(214748364842LL, value); | |
152 | |
153 // Serialize and compare to expected output. | |
154 ASSERT_TRUE(file_util::PathExists(golden_output_file)); | |
155 pref_store->CommitPendingWrite(); | |
156 MessageLoop::current()->RunAllPending(); | |
157 EXPECT_TRUE(file_util::TextContentsEqual(golden_output_file, output_file)); | |
158 ASSERT_TRUE(file_util::Delete(output_file, false)); | |
159 } | |
160 | |
161 TEST_F(JsonPrefStoreTest, Basic) { | |
162 ASSERT_TRUE(file_util::CopyFile(data_dir_.AppendASCII("read.json"), | |
163 temp_dir_.path().AppendASCII("write.json"))); | |
164 | |
165 // Test that the persistent value can be loaded. | |
166 FilePath input_file = temp_dir_.path().AppendASCII("write.json"); | |
167 ASSERT_TRUE(file_util::PathExists(input_file)); | |
168 scoped_refptr<JsonPrefStore> pref_store = | |
169 new JsonPrefStore(input_file, message_loop_proxy_.get()); | |
170 ASSERT_EQ(PersistentPrefStore::PREF_READ_ERROR_NONE, pref_store->ReadPrefs()); | |
171 ASSERT_FALSE(pref_store->ReadOnly()); | |
172 | |
173 // The JSON file looks like this: | |
174 // { | |
175 // "homepage": "http://www.cnn.com", | |
176 // "some_directory": "/usr/local/", | |
177 // "tabs": { | |
178 // "new_windows_in_tabs": true, | |
179 // "max_tabs": 20 | |
180 // } | |
181 // } | |
182 | |
183 RunBasicJsonPrefStoreTest(pref_store, | |
184 input_file, | |
185 data_dir_.AppendASCII("write.golden.json")); | |
186 } | |
187 | |
188 TEST_F(JsonPrefStoreTest, BasicAsync) { | |
189 ASSERT_TRUE(file_util::CopyFile(data_dir_.AppendASCII("read.json"), | |
190 temp_dir_.path().AppendASCII("write.json"))); | |
191 | |
192 // Test that the persistent value can be loaded. | |
193 FilePath input_file = temp_dir_.path().AppendASCII("write.json"); | |
194 ASSERT_TRUE(file_util::PathExists(input_file)); | |
195 scoped_refptr<JsonPrefStore> pref_store = | |
196 new JsonPrefStore(input_file, message_loop_proxy_.get()); | |
197 | |
198 MockPrefStoreObserver mock_observer; | |
199 pref_store->AddObserver(&mock_observer); | |
200 | |
201 MockReadErrorDelegate *mock_error_delegate = new MockReadErrorDelegate; | |
202 pref_store->ReadPrefsAsync(mock_error_delegate); | |
203 | |
204 EXPECT_CALL(mock_observer, OnInitializationCompleted(true)).Times(1); | |
205 EXPECT_CALL(*mock_error_delegate, | |
206 OnError(PersistentPrefStore::PREF_READ_ERROR_NONE)).Times(0); | |
207 message_loop_.RunAllPending(); | |
208 pref_store->RemoveObserver(&mock_observer); | |
209 | |
210 ASSERT_FALSE(pref_store->ReadOnly()); | |
211 | |
212 // The JSON file looks like this: | |
213 // { | |
214 // "homepage": "http://www.cnn.com", | |
215 // "some_directory": "/usr/local/", | |
216 // "tabs": { | |
217 // "new_windows_in_tabs": true, | |
218 // "max_tabs": 20 | |
219 // } | |
220 // } | |
221 | |
222 RunBasicJsonPrefStoreTest(pref_store, | |
223 input_file, | |
224 data_dir_.AppendASCII("write.golden.json")); | |
225 } | |
226 | |
227 // Tests asynchronous reading of the file when there is no file. | |
228 TEST_F(JsonPrefStoreTest, AsyncNonExistingFile) { | |
229 FilePath bogus_input_file = data_dir_.AppendASCII("read.txt"); | |
230 ASSERT_FALSE(file_util::PathExists(bogus_input_file)); | |
231 scoped_refptr<JsonPrefStore> pref_store = | |
232 new JsonPrefStore(bogus_input_file, message_loop_proxy_.get()); | |
233 MockPrefStoreObserver mock_observer; | |
234 pref_store->AddObserver(&mock_observer); | |
235 | |
236 MockReadErrorDelegate *mock_error_delegate = new MockReadErrorDelegate; | |
237 pref_store->ReadPrefsAsync(mock_error_delegate); | |
238 | |
239 EXPECT_CALL(mock_observer, OnInitializationCompleted(true)).Times(1); | |
240 EXPECT_CALL(*mock_error_delegate, | |
241 OnError(PersistentPrefStore::PREF_READ_ERROR_NO_FILE)).Times(1); | |
242 message_loop_.RunAllPending(); | |
243 pref_store->RemoveObserver(&mock_observer); | |
244 | |
245 EXPECT_FALSE(pref_store->ReadOnly()); | |
246 } | |
247 | |
248 TEST_F(JsonPrefStoreTest, NeedsEmptyValue) { | |
249 FilePath pref_file = temp_dir_.path().AppendASCII("write.json"); | |
250 | |
251 ASSERT_TRUE(file_util::CopyFile( | |
252 data_dir_.AppendASCII("read.need_empty_value.json"), | |
253 pref_file)); | |
254 | |
255 // Test that the persistent value can be loaded. | |
256 ASSERT_TRUE(file_util::PathExists(pref_file)); | |
257 scoped_refptr<JsonPrefStore> pref_store = | |
258 new JsonPrefStore(pref_file, message_loop_proxy_.get()); | |
259 ASSERT_EQ(PersistentPrefStore::PREF_READ_ERROR_NONE, pref_store->ReadPrefs()); | |
260 ASSERT_FALSE(pref_store->ReadOnly()); | |
261 | |
262 // The JSON file looks like this: | |
263 // { | |
264 // "list": [ 1 ], | |
265 // "list_needs_empty_value": [ 2 ], | |
266 // "dict": { | |
267 // "dummy": true, | |
268 // }, | |
269 // "dict_needs_empty_value": { | |
270 // "dummy": true, | |
271 // }, | |
272 // } | |
273 | |
274 // Set flag to preserve empty values for the following keys. | |
275 pref_store->MarkNeedsEmptyValue("list_needs_empty_value"); | |
276 pref_store->MarkNeedsEmptyValue("dict_needs_empty_value"); | |
277 | |
278 // Set all keys to empty values. | |
279 pref_store->SetValue("list", new base::ListValue); | |
280 pref_store->SetValue("list_needs_empty_value", new base::ListValue); | |
281 pref_store->SetValue("dict", new base::DictionaryValue); | |
282 pref_store->SetValue("dict_needs_empty_value", new base::DictionaryValue); | |
283 | |
284 // Write to file. | |
285 pref_store->CommitPendingWrite(); | |
286 MessageLoop::current()->RunAllPending(); | |
287 | |
288 // Compare to expected output. | |
289 FilePath golden_output_file = | |
290 data_dir_.AppendASCII("write.golden.need_empty_value.json"); | |
291 ASSERT_TRUE(file_util::PathExists(golden_output_file)); | |
292 EXPECT_TRUE(file_util::TextContentsEqual(golden_output_file, pref_file)); | |
293 } | |
OLD | NEW |