OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 <stdint.h> | 5 #include <stdint.h> |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/files/file_util.h" | 8 #include "base/files/file_util.h" |
9 #include "base/files/scoped_temp_dir.h" | 9 #include "base/files/scoped_temp_dir.h" |
10 #include "base/run_loop.h" | 10 #include "base/run_loop.h" |
(...skipping 20 matching lines...) Expand all Loading... |
31 : thread_bundle_(TestBrowserThreadBundle::IO_MAINLOOP), success_(false) {} | 31 : thread_bundle_(TestBrowserThreadBundle::IO_MAINLOOP), success_(false) {} |
32 | 32 |
33 // Callback to provide when reading a single notification from the database. | 33 // Callback to provide when reading a single notification from the database. |
34 void DidReadNotificationData(bool success, | 34 void DidReadNotificationData(bool success, |
35 const NotificationDatabaseData& database_data) { | 35 const NotificationDatabaseData& database_data) { |
36 success_ = success; | 36 success_ = success; |
37 database_data_ = database_data; | 37 database_data_ = database_data; |
38 } | 38 } |
39 | 39 |
40 // Callback to provide when writing a notification to the database. | 40 // Callback to provide when writing a notification to the database. |
41 void DidWriteNotificationData(bool success, int64_t notification_id) { | 41 void DidWriteNotificationData(bool success, |
| 42 const std::string& notification_id) { |
42 success_ = success; | 43 success_ = success; |
43 notification_id_ = notification_id; | 44 notification_id_ = notification_id; |
44 } | 45 } |
45 | 46 |
46 // Callback to provide when deleting notification data from the database. | 47 // Callback to provide when deleting notification data from the database. |
47 void DidDeleteNotificationData(bool success) { success_ = success; } | 48 void DidDeleteNotificationData(bool success) { success_ = success; } |
48 | 49 |
49 // Callback to provide when registering a Service Worker with a Service | 50 // Callback to provide when registering a Service Worker with a Service |
50 // Worker Context. Will write the registration id to |store_registration_id|. | 51 // Worker Context. Will write the registration id to |store_registration_id|. |
51 void DidRegisterServiceWorker(int64_t* store_registration_id, | 52 void DidRegisterServiceWorker(int64_t* store_registration_id, |
(...skipping 52 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
104 // Returns whether the last invoked callback finished successfully. | 105 // Returns whether the last invoked callback finished successfully. |
105 bool success() const { return success_; } | 106 bool success() const { return success_; } |
106 | 107 |
107 // Returns the NotificationDatabaseData associated with the last invoked | 108 // Returns the NotificationDatabaseData associated with the last invoked |
108 // ReadNotificationData callback. | 109 // ReadNotificationData callback. |
109 const NotificationDatabaseData& database_data() const { | 110 const NotificationDatabaseData& database_data() const { |
110 return database_data_; | 111 return database_data_; |
111 } | 112 } |
112 | 113 |
113 // Returns the notification id of the notification last written. | 114 // Returns the notification id of the notification last written. |
114 int64_t notification_id() const { return notification_id_; } | 115 const std::string& notification_id() const { return notification_id_; } |
115 | 116 |
116 private: | 117 private: |
117 TestBrowserThreadBundle thread_bundle_; | 118 TestBrowserThreadBundle thread_bundle_; |
118 TestBrowserContext browser_context_; | 119 TestBrowserContext browser_context_; |
119 | 120 |
120 bool success_; | 121 bool success_; |
121 NotificationDatabaseData database_data_; | 122 NotificationDatabaseData database_data_; |
122 int64_t notification_id_; | 123 std::string notification_id_; |
123 }; | 124 }; |
124 | 125 |
125 TEST_F(PlatformNotificationContextTest, ReadNonExistentNotification) { | 126 TEST_F(PlatformNotificationContextTest, ReadNonExistentNotification) { |
126 scoped_refptr<PlatformNotificationContextImpl> context = | 127 scoped_refptr<PlatformNotificationContextImpl> context = |
127 CreatePlatformNotificationContext(); | 128 CreatePlatformNotificationContext(); |
128 | 129 |
129 context->ReadNotificationData( | 130 context->ReadNotificationData( |
130 42 /* notification_id */, GURL("https://example.com"), | 131 "invalid-notification-id", GURL("https://example.com"), |
131 base::Bind(&PlatformNotificationContextTest::DidReadNotificationData, | 132 base::Bind(&PlatformNotificationContextTest::DidReadNotificationData, |
132 base::Unretained(this))); | 133 base::Unretained(this))); |
133 | 134 |
134 base::RunLoop().RunUntilIdle(); | 135 base::RunLoop().RunUntilIdle(); |
135 | 136 |
136 // The read operation should have failed, as it does not exist. | 137 // The read operation should have failed, as it does not exist. |
137 ASSERT_FALSE(success()); | 138 ASSERT_FALSE(success()); |
138 } | 139 } |
139 | 140 |
140 TEST_F(PlatformNotificationContextTest, WriteReadNotification) { | 141 TEST_F(PlatformNotificationContextTest, WriteReadNotification) { |
141 scoped_refptr<PlatformNotificationContextImpl> context = | 142 scoped_refptr<PlatformNotificationContextImpl> context = |
142 CreatePlatformNotificationContext(); | 143 CreatePlatformNotificationContext(); |
143 | 144 |
144 GURL origin("https://example.com"); | 145 GURL origin("https://example.com"); |
145 NotificationDatabaseData notification_database_data; | 146 NotificationDatabaseData notification_database_data; |
146 notification_database_data.origin = origin; | 147 notification_database_data.origin = origin; |
147 | 148 |
148 context->WriteNotificationData( | 149 context->WriteNotificationData( |
149 origin, notification_database_data, | 150 origin, notification_database_data, |
150 base::Bind(&PlatformNotificationContextTest::DidWriteNotificationData, | 151 base::Bind(&PlatformNotificationContextTest::DidWriteNotificationData, |
151 base::Unretained(this))); | 152 base::Unretained(this))); |
152 | 153 |
153 base::RunLoop().RunUntilIdle(); | 154 base::RunLoop().RunUntilIdle(); |
154 | 155 |
155 // The write operation should have succeeded with a notification id. | 156 // The write operation should have succeeded with a notification id. |
156 ASSERT_TRUE(success()); | 157 ASSERT_TRUE(success()); |
157 EXPECT_GT(notification_id(), 0); | 158 EXPECT_FALSE(notification_id().empty()); |
158 | 159 |
159 context->ReadNotificationData( | 160 context->ReadNotificationData( |
160 notification_id(), origin, | 161 notification_id(), origin, |
161 base::Bind(&PlatformNotificationContextTest::DidReadNotificationData, | 162 base::Bind(&PlatformNotificationContextTest::DidReadNotificationData, |
162 base::Unretained(this))); | 163 base::Unretained(this))); |
163 | 164 |
164 base::RunLoop().RunUntilIdle(); | 165 base::RunLoop().RunUntilIdle(); |
165 | 166 |
166 // The read operation should have succeeded, with the right notification. | 167 // The read operation should have succeeded, with the right notification. |
167 ASSERT_TRUE(success()); | 168 ASSERT_TRUE(success()); |
(...skipping 18 matching lines...) Expand all Loading... |
186 notification_database_data.notification_data.tag = tag; | 187 notification_database_data.notification_data.tag = tag; |
187 | 188 |
188 // Write the first notification with the given |tag|. | 189 // Write the first notification with the given |tag|. |
189 context->WriteNotificationData( | 190 context->WriteNotificationData( |
190 origin, notification_database_data, | 191 origin, notification_database_data, |
191 base::Bind(&PlatformNotificationContextTest::DidWriteNotificationData, | 192 base::Bind(&PlatformNotificationContextTest::DidWriteNotificationData, |
192 base::Unretained(this))); | 193 base::Unretained(this))); |
193 | 194 |
194 base::RunLoop().RunUntilIdle(); | 195 base::RunLoop().RunUntilIdle(); |
195 | 196 |
196 int64_t read_notification_id = notification_id(); | 197 std::string read_notification_id = notification_id(); |
197 | 198 |
198 // The write operation should have succeeded with a notification id. | 199 // The write operation should have succeeded with a notification id. |
199 ASSERT_TRUE(success()); | 200 ASSERT_TRUE(success()); |
200 EXPECT_GT(read_notification_id, 0); | 201 EXPECT_FALSE(read_notification_id.empty()); |
201 | 202 |
202 notification_database_data.notification_data.title = | 203 notification_database_data.notification_data.title = |
203 base::ASCIIToUTF16("Second"); | 204 base::ASCIIToUTF16("Second"); |
204 | 205 |
205 // Write the second notification with the given |tag|. | 206 // Write the second notification with the given |tag|. |
206 context->WriteNotificationData( | 207 context->WriteNotificationData( |
207 origin, notification_database_data, | 208 origin, notification_database_data, |
208 base::Bind(&PlatformNotificationContextTest::DidWriteNotificationData, | 209 base::Bind(&PlatformNotificationContextTest::DidWriteNotificationData, |
209 base::Unretained(this))); | 210 base::Unretained(this))); |
210 | 211 |
211 base::RunLoop().RunUntilIdle(); | 212 base::RunLoop().RunUntilIdle(); |
212 | 213 |
213 ASSERT_TRUE(success()); | 214 ASSERT_TRUE(success()); |
214 ASSERT_NE(notification_id(), read_notification_id); | 215 ASSERT_FALSE(notification_id().empty()); |
215 ASSERT_GT(notification_id(), 0); | 216 ASSERT_EQ(notification_id(), read_notification_id); |
216 | 217 |
217 // Reading the notifications should only yield the second, replaced one. | 218 // Reading the notifications should only yield the second, replaced one. |
218 std::vector<NotificationDatabaseData> notification_database_datas; | 219 std::vector<NotificationDatabaseData> notification_database_datas; |
219 context->ReadAllNotificationDataForServiceWorkerRegistration( | 220 context->ReadAllNotificationDataForServiceWorkerRegistration( |
220 origin, kFakeServiceWorkerRegistrationId, | 221 origin, kFakeServiceWorkerRegistrationId, |
221 base::Bind(&PlatformNotificationContextTest::DidReadAllNotificationDatas, | 222 base::Bind(&PlatformNotificationContextTest::DidReadAllNotificationDatas, |
222 base::Unretained(this), ¬ification_database_datas)); | 223 base::Unretained(this), ¬ification_database_datas)); |
223 | 224 |
224 base::RunLoop().RunUntilIdle(); | 225 base::RunLoop().RunUntilIdle(); |
225 | 226 |
226 // The read operation should have succeeded, with the right notification. | 227 // The read operation should have succeeded, with the right notification. |
227 ASSERT_TRUE(success()); | 228 ASSERT_TRUE(success()); |
228 | 229 |
229 ASSERT_EQ(1u, notification_database_datas.size()); | 230 ASSERT_EQ(1u, notification_database_datas.size()); |
230 | 231 |
231 EXPECT_EQ(tag, notification_database_datas[0].notification_data.tag); | 232 EXPECT_EQ(tag, notification_database_datas[0].notification_data.tag); |
232 EXPECT_EQ(base::ASCIIToUTF16("Second"), | 233 EXPECT_EQ(base::ASCIIToUTF16("Second"), |
233 notification_database_datas[0].notification_data.title); | 234 notification_database_datas[0].notification_data.title); |
234 } | 235 } |
235 | 236 |
236 TEST_F(PlatformNotificationContextTest, DeleteInvalidNotification) { | 237 TEST_F(PlatformNotificationContextTest, DeleteInvalidNotification) { |
237 scoped_refptr<PlatformNotificationContextImpl> context = | 238 scoped_refptr<PlatformNotificationContextImpl> context = |
238 CreatePlatformNotificationContext(); | 239 CreatePlatformNotificationContext(); |
239 | 240 |
240 context->DeleteNotificationData( | 241 context->DeleteNotificationData( |
241 42 /* notification_id */, GURL("https://example.com"), | 242 "invalid-notification-id", GURL("https://example.com"), |
242 base::Bind(&PlatformNotificationContextTest::DidDeleteNotificationData, | 243 base::Bind(&PlatformNotificationContextTest::DidDeleteNotificationData, |
243 base::Unretained(this))); | 244 base::Unretained(this))); |
244 | 245 |
245 base::RunLoop().RunUntilIdle(); | 246 base::RunLoop().RunUntilIdle(); |
246 | 247 |
247 // The notification may not have existed, but since the goal of deleting data | 248 // The notification may not have existed, but since the goal of deleting data |
248 // is to make sure that it's gone, the goal has been satisfied. As such, | 249 // is to make sure that it's gone, the goal has been satisfied. As such, |
249 // deleting a non-existent notification is considered to be a success. | 250 // deleting a non-existent notification is considered to be a success. |
250 EXPECT_TRUE(success()); | 251 EXPECT_TRUE(success()); |
251 } | 252 } |
252 | 253 |
253 TEST_F(PlatformNotificationContextTest, DeleteNotification) { | 254 TEST_F(PlatformNotificationContextTest, DeleteNotification) { |
254 scoped_refptr<PlatformNotificationContextImpl> context = | 255 scoped_refptr<PlatformNotificationContextImpl> context = |
255 CreatePlatformNotificationContext(); | 256 CreatePlatformNotificationContext(); |
256 | 257 |
257 GURL origin("https://example.com"); | 258 GURL origin("https://example.com"); |
258 NotificationDatabaseData notification_database_data; | 259 NotificationDatabaseData notification_database_data; |
259 | 260 |
260 context->WriteNotificationData( | 261 context->WriteNotificationData( |
261 origin, notification_database_data, | 262 origin, notification_database_data, |
262 base::Bind(&PlatformNotificationContextTest::DidWriteNotificationData, | 263 base::Bind(&PlatformNotificationContextTest::DidWriteNotificationData, |
263 base::Unretained(this))); | 264 base::Unretained(this))); |
264 | 265 |
265 base::RunLoop().RunUntilIdle(); | 266 base::RunLoop().RunUntilIdle(); |
266 | 267 |
267 // The write operation should have succeeded with a notification id. | 268 // The write operation should have succeeded with a notification id. |
268 ASSERT_TRUE(success()); | 269 ASSERT_TRUE(success()); |
269 EXPECT_GT(notification_id(), 0); | 270 EXPECT_FALSE(notification_id().empty()); |
270 | 271 |
271 context->DeleteNotificationData( | 272 context->DeleteNotificationData( |
272 notification_id(), origin, | 273 notification_id(), origin, |
273 base::Bind(&PlatformNotificationContextTest::DidDeleteNotificationData, | 274 base::Bind(&PlatformNotificationContextTest::DidDeleteNotificationData, |
274 base::Unretained(this))); | 275 base::Unretained(this))); |
275 | 276 |
276 base::RunLoop().RunUntilIdle(); | 277 base::RunLoop().RunUntilIdle(); |
277 | 278 |
278 // The notification existed, so it should have been removed successfully. | 279 // The notification existed, so it should have been removed successfully. |
279 ASSERT_TRUE(success()); | 280 ASSERT_TRUE(success()); |
(...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
323 | 324 |
324 // Create a notification for that Service Worker registration. | 325 // Create a notification for that Service Worker registration. |
325 notification_context->WriteNotificationData( | 326 notification_context->WriteNotificationData( |
326 origin, notification_database_data, | 327 origin, notification_database_data, |
327 base::Bind(&PlatformNotificationContextTest::DidWriteNotificationData, | 328 base::Bind(&PlatformNotificationContextTest::DidWriteNotificationData, |
328 base::Unretained(this))); | 329 base::Unretained(this))); |
329 | 330 |
330 base::RunLoop().RunUntilIdle(); | 331 base::RunLoop().RunUntilIdle(); |
331 | 332 |
332 ASSERT_TRUE(success()); | 333 ASSERT_TRUE(success()); |
333 EXPECT_GT(notification_id(), 0); | 334 EXPECT_FALSE(notification_id().empty()); |
334 | 335 |
335 ServiceWorkerStatusCode unregister_status; | 336 ServiceWorkerStatusCode unregister_status; |
336 | 337 |
337 // Now drop the Service Worker registration which owns that notification. | 338 // Now drop the Service Worker registration which owns that notification. |
338 embedded_worker_test_helper->context()->UnregisterServiceWorker( | 339 embedded_worker_test_helper->context()->UnregisterServiceWorker( |
339 origin, | 340 origin, |
340 base::Bind(&PlatformNotificationContextTest::DidUnregisterServiceWorker, | 341 base::Bind(&PlatformNotificationContextTest::DidUnregisterServiceWorker, |
341 base::Unretained(this), &unregister_status)); | 342 base::Unretained(this), &unregister_status)); |
342 | 343 |
343 base::RunLoop().RunUntilIdle(); | 344 base::RunLoop().RunUntilIdle(); |
(...skipping 19 matching lines...) Expand all Loading... |
363 | 364 |
364 context->WriteNotificationData( | 365 context->WriteNotificationData( |
365 origin, notification_database_data, | 366 origin, notification_database_data, |
366 base::Bind(&PlatformNotificationContextTest::DidWriteNotificationData, | 367 base::Bind(&PlatformNotificationContextTest::DidWriteNotificationData, |
367 base::Unretained(this))); | 368 base::Unretained(this))); |
368 | 369 |
369 base::RunLoop().RunUntilIdle(); | 370 base::RunLoop().RunUntilIdle(); |
370 | 371 |
371 // The write operation should have succeeded with a notification id. | 372 // The write operation should have succeeded with a notification id. |
372 ASSERT_TRUE(success()); | 373 ASSERT_TRUE(success()); |
373 EXPECT_GT(notification_id(), 0); | 374 EXPECT_FALSE(notification_id().empty()); |
374 | 375 |
375 // Call the OnStorageWiped override from the ServiceWorkerContextObserver, | 376 // Call the OnStorageWiped override from the ServiceWorkerContextObserver, |
376 // which indicates that the database should go away entirely. | 377 // which indicates that the database should go away entirely. |
377 context->OnStorageWiped(); | 378 context->OnStorageWiped(); |
378 | 379 |
379 // Verify that reading notification data fails because the data does not | 380 // Verify that reading notification data fails because the data does not |
380 // exist anymore. Deliberately omit RunUntilIdle(), since this is unlikely to | 381 // exist anymore. Deliberately omit RunUntilIdle(), since this is unlikely to |
381 // be the case when OnStorageWiped gets called in production. | 382 // be the case when OnStorageWiped gets called in production. |
382 context->ReadNotificationData( | 383 context->ReadNotificationData( |
383 notification_id(), origin, | 384 notification_id(), origin, |
(...skipping 12 matching lines...) Expand all Loading... |
396 // Manually construct the PlatformNotificationContextImpl because this test | 397 // Manually construct the PlatformNotificationContextImpl because this test |
397 // requires the database to be created on the filesystem. | 398 // requires the database to be created on the filesystem. |
398 scoped_refptr<PlatformNotificationContextImpl> context( | 399 scoped_refptr<PlatformNotificationContextImpl> context( |
399 new PlatformNotificationContextImpl(database_dir.path(), | 400 new PlatformNotificationContextImpl(database_dir.path(), |
400 browser_context(), nullptr)); | 401 browser_context(), nullptr)); |
401 | 402 |
402 OverrideTaskRunnerForTesting(context.get()); | 403 OverrideTaskRunnerForTesting(context.get()); |
403 | 404 |
404 // Trigger a read-operation to force creating the database. | 405 // Trigger a read-operation to force creating the database. |
405 context->ReadNotificationData( | 406 context->ReadNotificationData( |
406 42 /* notification_id */, GURL("https://example.com"), | 407 "invalid-notification-id", GURL("https://example.com"), |
407 base::Bind(&PlatformNotificationContextTest::DidReadNotificationData, | 408 base::Bind(&PlatformNotificationContextTest::DidReadNotificationData, |
408 base::Unretained(this))); | 409 base::Unretained(this))); |
409 | 410 |
410 base::RunLoop().RunUntilIdle(); | 411 base::RunLoop().RunUntilIdle(); |
411 | 412 |
412 EXPECT_FALSE(IsDirectoryEmpty(database_dir.path())); | 413 EXPECT_FALSE(IsDirectoryEmpty(database_dir.path())); |
413 EXPECT_FALSE(success()); | 414 EXPECT_FALSE(success()); |
414 | 415 |
415 // Blow away the database by faking a Service Worker Context wipe-out. | 416 // Blow away the database by faking a Service Worker Context wipe-out. |
416 context->OnStorageWiped(); | 417 context->OnStorageWiped(); |
(...skipping 60 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
477 ASSERT_EQ(10u, notification_database_datas.size()); | 478 ASSERT_EQ(10u, notification_database_datas.size()); |
478 | 479 |
479 for (int i = 0; i < 10; ++i) { | 480 for (int i = 0; i < 10; ++i) { |
480 EXPECT_EQ(origin, notification_database_datas[i].origin); | 481 EXPECT_EQ(origin, notification_database_datas[i].origin); |
481 EXPECT_EQ(kFakeServiceWorkerRegistrationId, | 482 EXPECT_EQ(kFakeServiceWorkerRegistrationId, |
482 notification_database_datas[i].service_worker_registration_id); | 483 notification_database_datas[i].service_worker_registration_id); |
483 } | 484 } |
484 } | 485 } |
485 | 486 |
486 } // namespace content | 487 } // namespace content |
OLD | NEW |