Chromium Code Reviews| Index: chrome/browser/media_galleries/fileapi/picasa/picasa_file_util_unittest.cc |
| diff --git a/chrome/browser/media_galleries/fileapi/picasa/picasa_file_util_unittest.cc b/chrome/browser/media_galleries/fileapi/picasa/picasa_file_util_unittest.cc |
| index cd19e3bb25eccc2463315701ad6a285bddf05126..bf5d21d29739a5365dbde69f43cc721dee567d20 100644 |
| --- a/chrome/browser/media_galleries/fileapi/picasa/picasa_file_util_unittest.cc |
| +++ b/chrome/browser/media_galleries/fileapi/picasa/picasa_file_util_unittest.cc |
| @@ -8,12 +8,14 @@ |
| #include "base/bind_helpers.h" |
| #include "base/file_util.h" |
| +#include "base/files/file_path.h" |
| #include "base/files/scoped_temp_dir.h" |
| #include "base/memory/scoped_vector.h" |
| #include "base/message_loop/message_loop.h" |
| #include "base/message_loop/message_loop_proxy.h" |
| #include "base/run_loop.h" |
| #include "base/strings/stringprintf.h" |
| +#include "base/synchronization/waitable_event.h" |
| #include "base/time/time.h" |
| #include "chrome/browser/media_galleries/fileapi/media_file_system_backend.h" |
| #include "chrome/browser/media_galleries/fileapi/media_path_filter.h" |
| @@ -151,31 +153,6 @@ void ReadDirectoryTestHelper(fileapi::FileSystemOperationRunner* runner, |
| } // namespace |
| -class TestPicasaDataProvider : public PicasaDataProvider { |
| - public: |
| - TestPicasaDataProvider() |
| - : PicasaDataProvider(base::FilePath(FILE_PATH_LITERAL("Fake"))), |
| - initialized_(false) { |
| - } |
| - |
| - virtual ~TestPicasaDataProvider() {} |
| - |
| - virtual void RefreshData(const base::Closure& ready_callback) OVERRIDE { |
| - DCHECK(initialized_); |
| - ready_callback.Run(); |
| - } |
| - |
| - void Init(const std::vector<AlbumInfo>& albums, |
| - const std::vector<AlbumInfo>& folders) { |
| - UniquifyNames(albums, &album_map_); |
| - UniquifyNames(folders, &folder_map_); |
| - initialized_ = true; |
| - } |
| - |
| - private: |
| - bool initialized_; |
| -}; |
| - |
| class TestPicasaFileUtil : public PicasaFileUtil { |
| public: |
| TestPicasaFileUtil(chrome::MediaPathFilter* media_path_filter, |
| @@ -227,8 +204,15 @@ class PicasaFileUtilTest : public testing::Test { |
| scoped_refptr<quota::SpecialStoragePolicy> storage_policy = |
| new quota::MockSpecialStoragePolicy(); |
| + base::WaitableEvent event(true, false); |
|
vandebo (ex-Chrome)
2013/08/22 17:48:02
Abstract this into void RunOnMediaTaskRunner(const
tommycli
2013/08/22 22:32:51
Done.
|
| + chrome::MediaFileSystemBackend::MediaTaskRunner()->PostTask( |
| + FROM_HERE, |
| + base::Bind(&PicasaFileUtilTest::SetUpOnMediaTaskRunner, |
| + base::Unretained(this), |
| + &event)); |
| + event.Wait(); |
| + |
| media_path_filter_.reset(new chrome::MediaPathFilter()); |
| - picasa_data_provider_.reset(new TestPicasaDataProvider()); |
| ScopedVector<fileapi::FileSystemBackend> additional_providers; |
| additional_providers.push_back(new TestMediaFileSystemBackend( |
| @@ -247,7 +231,38 @@ class PicasaFileUtilTest : public testing::Test { |
| fileapi::CreateAllowFileAccessOptions()); |
| } |
| + virtual void TearDown() OVERRIDE { |
| + base::WaitableEvent event(true, false); |
|
vandebo (ex-Chrome)
2013/08/22 17:48:02
Why not use a runloop like in the browser test?
tommycli
2013/08/22 22:32:51
I believe this was because Runloops can only be us
vandebo (ex-Chrome)
2013/08/23 16:42:41
If calling on the wrong thread is the only issue,
tommycli
2013/08/26 22:06:17
Done.
|
| + chrome::MediaFileSystemBackend::MediaTaskRunner()->PostTask( |
| + FROM_HERE, |
| + base::Bind(&PicasaFileUtilTest::TearDownOnMediaTaskRunner, |
| + base::Unretained(this), |
| + &event)); |
| + event.Wait(); |
| + } |
| + |
| protected: |
| + void SetUpOnMediaTaskRunner(base::WaitableEvent* event) { |
| + picasa_data_provider_.reset(new PicasaDataProvider(base::FilePath())); |
| + event->Signal(); |
| + } |
| + |
| + void TearDownOnMediaTaskRunner(base::WaitableEvent* event) { |
| + picasa_data_provider_.reset(); |
| + event->Signal(); |
| + } |
| + |
| + void SetupDataProvider(PicasaDataProvider* picasa_data_provider, |
| + const std::vector<AlbumInfo>& albums, |
| + const std::vector<AlbumInfo>& folders) { |
| + PicasaDataProvider::UniquifyNames(albums, |
| + &picasa_data_provider->album_map_); |
| + PicasaDataProvider::UniquifyNames(folders, |
| + &picasa_data_provider->folder_map_); |
| + picasa_data_provider->state_ = |
| + PicasaDataProvider::ALBUMS_IMAGES_FRESH_STATE; |
| + } |
| + |
| // |test_folders| must be in alphabetical order for easy verification |
| void SetupFolders(ScopedVector<TestFolder>* test_folders) { |
| std::vector<AlbumInfo> folders; |
| @@ -257,7 +272,9 @@ class PicasaFileUtilTest : public testing::Test { |
| ASSERT_TRUE(test_folder->Init()); |
| folders.push_back(test_folder->folder_info()); |
| } |
| - picasa_data_provider_->Init(std::vector<AlbumInfo>(), folders); |
| + |
| + SetupDataProvider( |
| + picasa_data_provider_.get(), std::vector<AlbumInfo>(), folders); |
| } |
| void VerifyFolderDirectoryList(const ScopedVector<TestFolder>& test_folders) { |
| @@ -337,8 +354,8 @@ class PicasaFileUtilTest : public testing::Test { |
| base::ScopedTempDir profile_dir_; |
| scoped_refptr<fileapi::FileSystemContext> file_system_context_; |
| + scoped_ptr<PicasaDataProvider> picasa_data_provider_; |
| scoped_ptr<chrome::MediaPathFilter> media_path_filter_; |
| - scoped_ptr<TestPicasaDataProvider> picasa_data_provider_; |
| DISALLOW_COPY_AND_ASSIGN(PicasaFileUtilTest); |
| }; |