Chromium Code Reviews| Index: apps/app_restore_service_browsertest.cc |
| diff --git a/apps/app_restore_service_browsertest.cc b/apps/app_restore_service_browsertest.cc |
| index ce800a446d4ff3c2b65333c8ef290050b6608ef8..996b2f1fc1f155dc3aacaf8e3dfe6227e575410e 100644 |
| --- a/apps/app_restore_service_browsertest.cc |
| +++ b/apps/app_restore_service_browsertest.cc |
| @@ -4,7 +4,7 @@ |
| #include "apps/app_restore_service.h" |
| #include "apps/app_restore_service_factory.h" |
| -#include "chrome/browser/extensions/api/file_handlers/app_file_handler_util.h" |
| +#include "apps/saved_files_service.h" |
| #include "chrome/browser/extensions/api/file_system/file_system_api.h" |
| #include "chrome/browser/extensions/extension_prefs.h" |
| #include "chrome/browser/extensions/extension_test_message_listener.h" |
| @@ -12,7 +12,6 @@ |
| #include "chrome/common/extensions/extension.h" |
| #include "content/public/test/test_utils.h" |
| -using extensions::app_file_handler_util::SavedFileEntry; |
| using extensions::Extension; |
| using extensions::ExtensionPrefs; |
| using extensions::ExtensionSystem; |
| @@ -78,21 +77,15 @@ IN_PROC_BROWSER_TEST_F(PlatformAppBrowserTest, FileAccessIsSavedToPrefs) { |
| ASSERT_TRUE(extension); |
| file_written_listener.WaitUntilSatisfied(); |
| - ExtensionPrefs* extension_prefs = |
| - ExtensionPrefs::Get(browser()->profile()); |
| + SavedFilesService* saved_files_service = SavedFilesService::Get(profile()); |
| - // Record the file entries in prefs because when the app gets suspended it |
| - // will have them all cleared. |
|
Matt Giuca
2013/05/22 08:26:14
Why is this comment removed?
Sam McNally
2013/05/23 03:47:28
koz said to.
|
| - std::vector<SavedFileEntry> file_entries; |
| - extensions::app_file_handler_util::GetSavedFileEntries( |
| - extension_prefs, extension->id(), &file_entries); |
| + std::vector<SavedFileEntry> file_entries = |
| + saved_files_service->GetAllFileEntries(extension->id()); |
| // One for the read-only file entry and one for the writable file entry. |
| ASSERT_EQ(2u, file_entries.size()); |
| extension_suspended.Wait(); |
| - file_entries.clear(); |
| - extensions::app_file_handler_util::GetSavedFileEntries( |
| - extension_prefs, extension->id(), &file_entries); |
| + file_entries = saved_files_service->GetAllFileEntries(extension->id()); |
| // File entries should be cleared when the extension is suspended. |
| ASSERT_TRUE(file_entries.empty()); |
| } |
| @@ -124,12 +117,9 @@ IN_PROC_BROWSER_TEST_F(PlatformAppBrowserTest, FileAccessIsRestored) { |
| ExtensionPrefs* extension_prefs = |
| ExtensionPrefs::Get(browser()->profile()); |
| - // Record the file entries in prefs because when the app gets suspended it |
| - // will have them all cleared. |
| - std::vector<SavedFileEntry> file_entries; |
| - extensions::app_file_handler_util::GetSavedFileEntries(extension_prefs, |
| - extension->id(), |
| - &file_entries); |
| + SavedFilesService* saved_files_service = SavedFilesService::Get(profile()); |
| + std::vector<SavedFileEntry> file_entries = |
| + saved_files_service->GetAllFileEntries(extension->id()); |
| extension_suspended.Wait(); |
| // Simulate a restart by populating the preferences as if the browser didn't |
| @@ -137,8 +127,8 @@ IN_PROC_BROWSER_TEST_F(PlatformAppBrowserTest, FileAccessIsRestored) { |
| extension_prefs->SetExtensionRunning(extension->id(), true); |
| for (std::vector<SavedFileEntry>::const_iterator it = file_entries.begin(); |
| it != file_entries.end(); ++it) { |
| - extensions::app_file_handler_util::AddSavedFileEntry( |
| - extension_prefs, extension->id(), it->id, it->path, it->writable); |
| + saved_files_service->RetainFileUntilAppSuspend( |
| + extension->id(), it->id, it->path, it->writable); |
| } |
| apps::AppRestoreServiceFactory::GetForProfile(browser()->profile())-> |