Media Galleries API Picasa: End-to-end browsertest.

chrome/browser/extensions/api/media_galleries/media_galleries_apitest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/command_line.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/apps/app_browsertest_util.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/extensions/extension_apitest.h"
 #include "chrome/browser/extensions/extension_process_manager.h"
 #include "chrome/browser/extensions/extension_system.h"
 #include "chrome/browser/media_galleries/media_file_system_registry.h"
 #include "chrome/browser/media_galleries/media_galleries_preferences.h"
 #include "chrome/browser/media_galleries/media_galleries_test_util.h"
 #include "chrome/browser/storage_monitor/storage_info.h"
 #include "chrome/browser/storage_monitor/storage_monitor.h"
 #include "chrome/browser/ui/browser.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/test/test_utils.h"
 #include "extensions/common/switches.h"
 
+#if defined(OS_WIN) || defined(OS_MACOSX)
+#include "chrome/browser/media_galleries/fileapi/picasa_finder.h"
+#include "chrome/common/media_galleries/picasa_test_util.h"
+#include "chrome/common/media_galleries/picasa_types.h"
+#include "chrome/common/media_galleries/pmp_test_util.h"
+#endif
+
 using extensions::PlatformAppBrowserTest;
 
 namespace {
 
 // Dummy device properties.
 const char kDeviceId[] = "testDeviceId";
 const char kDeviceName[] = "foobar";
 #if defined(FILE_PATH_USES_DRIVE_LETTERS)
 base::FilePath::CharType kDevicePath[] = FILE_PATH_LITERAL("C:\\qux");
 #else
 base::FilePath::CharType kDevicePath[] = FILE_PATH_LITERAL("/qux");
 #endif
 
 const char kTestGalleries[] = "testGalleries(%d)";
 
-}  // namespace
-
 // This function is to ensure at least one (fake) media gallery exists for
 // testing platforms with no default media galleries, such as CHROMEOS.
 void MakeFakeMediaGalleryForTest(Profile* profile, const base::FilePath& path) {
   base::RunLoop runloop;
   StorageMonitor::GetInstance()->EnsureInitialized(runloop.QuitClosure());
   runloop.Run();
 
   MediaGalleriesPreferences* preferences =
       g_browser_process->media_file_system_registry()->GetPreferences(profile);
 
   MediaGalleryPrefInfo gallery_info;
   ASSERT_FALSE(preferences->LookUpGalleryByPath(path, &gallery_info));
   preferences->AddGallery(gallery_info.device_id,
                           gallery_info.path,
                           false /* user_added */,
                           gallery_info.volume_label,
                           gallery_info.vendor_name,
                           gallery_info.model_name,
                           gallery_info.total_size_in_bytes,
                           gallery_info.last_attach_time);
 
   content::RunAllPendingInMessageLoop();
 }
 
+}  // namespace
+
 class MediaGalleriesPlatformAppBrowserTest : public PlatformAppBrowserTest {
  protected:
   // Since ExtensionTestMessageListener does not work with RunPlatformAppTest(),
   // This helper method can be used to run additional media gallery tests.
   void RunSecondTestPhase(const string16& command) {
     const extensions::Extension* extension = GetSingleLoadedExtension();
     extensions::ExtensionHost* host =
         extensions::ExtensionSystem::Get(browser()->profile())->
             process_manager()->GetBackgroundHostForExtension(extension->id());
     ASSERT_TRUE(host);
@@ skipping 11 matching lines @@
         StorageInfo(device_id_, string16(), kDevicePath,
                     ASCIIToUTF16(kDeviceName), string16(), string16(), 0));
     content::RunAllPendingInMessageLoop();
   }
 
   void DetachFakeDevice() {
     StorageMonitor::GetInstance()->receiver()->ProcessDetach(device_id_);
     content::RunAllPendingInMessageLoop();
   }
 
+#if defined(OS_WIN) || defined(OS_MACOSX)
+  // This initializes the storage monitor, MediaGalleriesPreferences, and
+  // removes all non-Picasa media galleries for a test.
+  void SetMediaGalleriesPreferencesForPicasaTest(Profile* profile) {

[vandebo (ex-Chrome), 2013/09/17 21:18:06]

I think the js equivalent (picking out the picasa gallery) would be easy and
easier to maintain.

[tommycli, 2013/09/18 15:25:37]

Done.

+    base::RunLoop runloop;
+    StorageMonitor::GetInstance()->EnsureInitialized(runloop.QuitClosure());
+    runloop.Run();
+
+    MediaGalleriesPreferences* preferences =
+        g_browser_process->media_file_system_registry()->GetPreferences(
+            profile);
+
+    content::RunAllPendingInMessageLoop();
+
+    // Remove all non-Picasa galleries for ease of testing.
+    MediaGalleriesPrefInfoMap known_galleries = preferences->known_galleries();
+    for (MediaGalleriesPrefInfoMap::const_iterator it = known_galleries.begin();
+         it != known_galleries.end();
+         ++it) {
+      if (!StorageInfo::IsPicasaDevice(it->second.device_id))
+        preferences->ForgetGalleryById(it->first);
+    }
+
+    content::RunAllPendingInMessageLoop();
+  }
+#endif
+
  private:
   std::string device_id_;
 };
 
 IN_PROC_BROWSER_TEST_F(MediaGalleriesPlatformAppBrowserTest,
                        MediaGalleriesNoAccess) {
   EnsureMediaDirectoriesExists media_directories;
   ASSERT_TRUE(RunPlatformAppTest("api_test/media_galleries/no_access"))
       << message_;
   RunSecondTestPhase(base::UTF8ToUTF16(base::StringPrintf(
@@ skipping 58 matching lines @@
 
   DetachFakeDevice();
 }
 
 IN_PROC_BROWSER_TEST_F(MediaGalleriesPlatformAppBrowserTest,
                        GetFilesystemMetadata) {
   EnsureMediaDirectoriesExists media_directories;
   ASSERT_TRUE(RunPlatformAppTest("api_test/media_galleries/metadata"))
       << message_;
 }
+
+#if defined(OS_WIN) || defined(OS_MACOSX)
+IN_PROC_BROWSER_TEST_F(MediaGalleriesPlatformAppBrowserTest, Picasa) {
+  EnsureMediaDirectoriesExists media_directories;
+
+  // Create Picasa metadata database directories.
+  base::FilePath picasa_database_path =
+      picasa::PicasaFinder::GetPicasaDatabasePath();

[vandebo (ex-Chrome), 2013/09/17 21:18:06]

Would it make sense to just call the regular PicasaFinder method here?

[tommycli, 2013/09/18 15:25:37]

Changed some stuff here.

+  base::FilePath picasa_temp_dir_path =
+      picasa_database_path.DirName().AppendASCII(picasa::kPicasaTempDirName);
+  ASSERT_TRUE(file_util::CreateDirectory(picasa_database_path));
+  ASSERT_TRUE(file_util::CreateDirectory(picasa_temp_dir_path));
+
+  // Create fake folder directories.
+  base::ScopedTempDir folders_root;
+  ASSERT_TRUE(folders_root.CreateUniqueTempDir());
+  base::FilePath fake_folder_1 = folders_root.path().AppendASCII("folder1");
+  base::FilePath fake_folder_2 = folders_root.path().AppendASCII("folder2");
+  ASSERT_TRUE(file_util::CreateDirectory(fake_folder_1));
+  ASSERT_TRUE(file_util::CreateDirectory(fake_folder_2));
+
+  // Write folder and album contents.
+  picasa::WriteTestAlbumTable(
+      picasa_database_path, fake_folder_1, fake_folder_2);
+  picasa::WriteTestAlbumsImagesIndex(fake_folder_1, fake_folder_2);
+  picasa::WriteJPEGHeader(fake_folder_1.AppendASCII("InBoth.jpg"));
+  picasa::WriteJPEGHeader(fake_folder_1.AppendASCII("InSecondAlbumOnly.jpg"));
+  picasa::WriteJPEGHeader(fake_folder_2.AppendASCII("InFirstAlbumOnly.jpg"));
+
+  SetMediaGalleriesPreferencesForPicasaTest(browser()->profile());
+
+  ASSERT_TRUE(RunPlatformAppTest("api_test/media_galleries/picasa"))
+      << message_;
+}
+#endif  // defined(OS_WIN) || defined(OS_MACOSX)