Expose gallery watching functionality to the public mediaGalleries API.

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

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.

[Lei Zhang, 2014/08/25 22:45:21]

nit: "(c) 2012" -> 2014, ditto in
chrome/test/data/extensions/api_test/media_galleries/gallerywatch/test.js

[Oren Blasberg, 2014/08/26 00:40:49]

Done.

 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 //
-// MediaGalleriesPrivate gallery watch API browser tests.
+// MediaGalleries gallery watch API browser tests.
 
 #include "base/file_util.h"
 #include "base/files/file_path.h"
 #include "base/files/file_path_watcher.h"
 #include "base/path_service.h"
 #include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "build/build_config.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/extensions/extension_apitest.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_test_message_listener.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/common/chrome_paths.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "extensions/browser/extension_system.h"
 #include "extensions/common/extension.h"
 #include "extensions/common/switches.h"
 
 namespace {
 
 // Id of test extension from
 // chrome/test/data/extensions/api_test/|kTestExtensionPath|
 const char kTestExtensionId[] = "gceegfkgibmgpfopknlcgleimclbknie";
-const char kTestExtensionPath[] = "media_galleries_private/gallerywatch";
+const char kTestExtensionPath[] = "media_galleries/gallerywatch";
 
 // JS commands.
 const char kGetAllWatchedGalleryIdsCmd[] = "getAllWatchedGalleryIds()";
 const char kGetMediaFileSystemsCmd[] = "getMediaFileSystems()";
 const char kSetupWatchOnValidGalleriesCmd[] = "setupWatchOnValidGalleries()";
+const char kSetupWatchOnUnlistenedValidGalleriesCmd[] =
+    "setupWatchOnUnlistenedValidGalleries()";
 const char kAddGalleryChangedListenerCmd[] = "addGalleryChangedListener()";
+const char kAddCheckingGalleryChangedListenerCmd[] =
+    "addCheckingGalleryChangedListener()";
 const char kRemoveAllGalleryWatchCmd[] = "removeAllGalleryWatch()";
 const char kRemoveGalleryChangedListenerCmd[] =
     "removeGalleryChangedListener()";
 const char kRemoveGalleryWatchCmd[] = "removeGalleryWatch()";
 const char kSetupWatchOnInvalidGalleryCmd[] = "setupWatchOnInvalidGallery()";
 
 // And JS reply messages.
 const char kAddGalleryWatchOK[] = "add_gallery_watch_ok";
 const char kGetAllGalleryWatchOK[] = "get_all_gallery_watch_ok";
 const char kGetMediaFileSystemsCallbackOK[] =
     "get_media_file_systems_callback_ok";
 const char kGetMediaFileSystemsOK[] = "get_media_file_systems_ok";
 const char kAddGalleryChangedListenerOK[] = "add_gallery_changed_listener_ok";
 const char kRemoveAllGalleryWatchOK[] = "remove_all_gallery_watch_ok";
 const char kRemoveGalleryChangedListenerOK[] =
     "remove_gallery_changed_listener_ok";
 const char kRemoveGalleryWatchOK[] = "remove_gallery_watch_ok";
+const char kOnGalleryChangedCheckingOK[] = "on_gallery_changed_checking_ok";
 
 // Test reply messages.
 const char kNoGalleryWatchesInstalled[] = "gallery_watchers_does_not_exists";
 const char kAddGalleryWatchRequestFailed[] = "add_watch_request_failed";
+const char kAddGalleryWatchRequestRuntimeError[] =
+    "add_watch_request_runtime_error";
 const char kAddGalleryWatchRequestSucceeded[] = "add_watch_request_succeeded";
 const char kGalleryChangedEventReceived[] = "gallery_changed_event_received";
 const char kGalleryWatchesCheck[] = "gallery_watcher_checks";
 
 }  // namespace
 
-
 ///////////////////////////////////////////////////////////////////////////////
-//                 MediaGalleriesPrivateGalleryWatchApiTest                  //
+//                 MediaGalleriesGalleryWatchApiTest                         //
 ///////////////////////////////////////////////////////////////////////////////
 
-class MediaGalleriesPrivateGalleryWatchApiTest : public ExtensionApiTest {
+class MediaGalleriesGalleryWatchApiTest : public ExtensionApiTest {
  public:
-  MediaGalleriesPrivateGalleryWatchApiTest()
-      : extension_(NULL),
-        background_host_(NULL) {
-  }
-  virtual ~MediaGalleriesPrivateGalleryWatchApiTest() {}
+  MediaGalleriesGalleryWatchApiTest()
+      : extension_(NULL), background_host_(NULL) {}
+  virtual ~MediaGalleriesGalleryWatchApiTest() {}
 
  protected:
   // ExtensionApiTest overrides.
   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     ExtensionApiTest::SetUpCommandLine(command_line);
     command_line->AppendSwitchASCII(
-        extensions::switches::kWhitelistedExtensionID,
-        kTestExtensionId);
+        extensions::switches::kWhitelistedExtensionID, kTestExtensionId);
   }
   virtual void SetUpOnMainThread() OVERRIDE {
     ExtensionApiTest::SetUpOnMainThread();
     ensure_media_directories_exists_.reset(new EnsureMediaDirectoriesExists);
     extension_ = LoadExtension(test_data_dir_.AppendASCII(kTestExtensionPath));
     GetBackgroundHostForTestExtension();
     CreateTestGallery();
     FetchMediaGalleriesList();
   }
   virtual void TearDownOnMainThread() OVERRIDE {
@@ skipping 12 matching lines @@
     ExtensionTestMessageListener listener(ok_message, false);
     background_host_->GetMainFrame()->ExecuteJavaScript(
         base::ASCIIToUTF16(js_command));
     EXPECT_TRUE(listener.WaitUntilSatisfied());
   }
 
   bool AddNewFileInTestGallery() {
     base::FilePath gallery_file =
         test_gallery_.path().Append(FILE_PATH_LITERAL("test1.txt"));
     std::string content("new content");
-    int write_size = base::WriteFile(gallery_file, content.c_str(),
-                                     content.length());
+    int write_size =
+        base::WriteFile(gallery_file, content.c_str(), content.length());
     return (write_size == static_cast<int>(content.length()));
   }
 
   void SetupGalleryWatches() {
-    std::string expected_result = GalleryWatchesSupported() ?
-        kAddGalleryWatchRequestSucceeded : kAddGalleryWatchRequestFailed;
+    std::string expected_result = GalleryWatchesSupported()
+                                      ? kAddGalleryWatchRequestSucceeded
+                                      : kAddGalleryWatchRequestFailed;
 
     ExtensionTestMessageListener add_gallery_watch_finished(
-        expected_result, false  /* no reply */);
+        expected_result, false /* no reply */);
     ExecuteCmdAndCheckReply(kSetupWatchOnValidGalleriesCmd, kAddGalleryWatchOK);
     EXPECT_TRUE(add_gallery_watch_finished.WaitUntilSatisfied());
   }
 
  private:
   void GetBackgroundHostForTestExtension() {
     ASSERT_TRUE(extension_);
     extensions::ExtensionSystem* extension_system =
         extensions::ExtensionSystem::Get(browser()->profile());
-    background_host_ =
-        extension_system->process_manager()->GetBackgroundHostForExtension(
-            extension_->id())->render_view_host();
+    background_host_ = extension_system->process_manager()
+                           ->GetBackgroundHostForExtension(extension_->id())
+                           ->render_view_host();
     ASSERT_TRUE(background_host_);
   }
 
   void CreateTestGallery() {
     MediaGalleriesPreferences* preferences =
         g_browser_process->media_file_system_registry()->GetPreferences(
             browser()->profile());
     base::RunLoop runloop;
     preferences->EnsureInitialized(runloop.QuitClosure());
     runloop.Run();
 
     ASSERT_TRUE(test_gallery_.CreateUniqueTempDir());
     MediaGalleryPrefInfo gallery_info;
-    ASSERT_FALSE(preferences->LookUpGalleryByPath(test_gallery_.path(),
-                                                  &gallery_info));
-    MediaGalleryPrefId id = preferences->AddGallery(
-        gallery_info.device_id,
-        gallery_info.path,
-        MediaGalleryPrefInfo::kAutoDetected,
-        gallery_info.volume_label,
-        gallery_info.vendor_name,
-        gallery_info.model_name,
-        gallery_info.total_size_in_bytes,
-        gallery_info.last_attach_time,
-        0, 0, 0);
+    ASSERT_FALSE(
+        preferences->LookUpGalleryByPath(test_gallery_.path(), &gallery_info));
+    MediaGalleryPrefId id =
+        preferences->AddGallery(gallery_info.device_id,
+                                gallery_info.path,
+                                MediaGalleryPrefInfo::kAutoDetected,
+                                gallery_info.volume_label,
+                                gallery_info.vendor_name,
+                                gallery_info.model_name,
+                                gallery_info.total_size_in_bytes,
+                                gallery_info.last_attach_time,
+                                0,
+                                0,
+                                0);
 
     preferences->SetGalleryPermissionForExtension(*extension_, id, true);
   }
 
   void FetchMediaGalleriesList() {
     ExtensionTestMessageListener get_media_systems_finished(
-        kGetMediaFileSystemsCallbackOK, false  /* no reply */);
+        kGetMediaFileSystemsCallbackOK, false /* no reply */);
     ExecuteCmdAndCheckReply(kGetMediaFileSystemsCmd, kGetMediaFileSystemsOK);
     EXPECT_TRUE(get_media_systems_finished.WaitUntilSatisfied());
   }
 
   scoped_ptr<EnsureMediaDirectoriesExists> ensure_media_directories_exists_;
 
   base::ScopedTempDir test_gallery_;
 
   const extensions::Extension* extension_;
 
   content::RenderViewHost* background_host_;
 
-  DISALLOW_COPY_AND_ASSIGN(MediaGalleriesPrivateGalleryWatchApiTest);
+  DISALLOW_COPY_AND_ASSIGN(MediaGalleriesGalleryWatchApiTest);
 };
 
 // Crashing on OSX.
 #if defined(OS_MACOSX)
 #define MAYBE_BasicGalleryWatch DISABLED_BasicGalleryWatch

[Lei Zhang, 2014/08/25 22:45:21]

Maybe you should enable your new tests on OS X?
DISABLED_RemoveListenerAndModifyGallery below as well...

[Oren Blasberg, 2014/08/26 00:40:49]

I discussed w/ Tommy. I will do this in a follow up CL immediately after this
one. That way if those tests end up flaky we can handle them independently of
rolling back this entire CL.

 #else
 #define MAYBE_BasicGalleryWatch BasicGalleryWatch
 #endif
-IN_PROC_BROWSER_TEST_F(MediaGalleriesPrivateGalleryWatchApiTest,
+IN_PROC_BROWSER_TEST_F(MediaGalleriesGalleryWatchApiTest,
                        MAYBE_BasicGalleryWatch) {
-  SetupGalleryWatches();
-
   // Add gallery watch listener.
   ExecuteCmdAndCheckReply(kAddGalleryChangedListenerCmd,
                           kAddGalleryChangedListenerOK);
 
+  SetupGalleryWatches();
+
   // Modify gallery contents.
   ExtensionTestMessageListener gallery_change_event_received(
-      kGalleryChangedEventReceived, false  /* no reply */);
+      kGalleryChangedEventReceived, false /* no reply */);
+
   ASSERT_TRUE(AddNewFileInTestGallery());
   if (GalleryWatchesSupported())
     EXPECT_TRUE(gallery_change_event_received.WaitUntilSatisfied());
 
   // Remove gallery watch listener.
   ExecuteCmdAndCheckReply(kRemoveGalleryChangedListenerCmd,
                           kRemoveGalleryChangedListenerOK);
 
   // Remove gallery watch request.
   if (GalleryWatchesSupported())
     ExecuteCmdAndCheckReply(kRemoveGalleryWatchCmd, kRemoveGalleryWatchOK);
 }
 
+IN_PROC_BROWSER_TEST_F(MediaGalleriesGalleryWatchApiTest,
+                       CorrectResponseOnModifyingWatchedGallery) {
+  if (!GalleryWatchesSupported())
+    return;
+
+  // Add gallery watch listener.
+  ExecuteCmdAndCheckReply(kAddCheckingGalleryChangedListenerCmd,
+                          kAddGalleryChangedListenerOK);
+
+  ExecuteCmdAndCheckReply(kSetupWatchOnValidGalleriesCmd, kAddGalleryWatchOK);
+
+  // Modify gallery contents; expect correct details.
+  ExtensionTestMessageListener got_correct_details(
+      kOnGalleryChangedCheckingOK, false);
+  ASSERT_TRUE(AddNewFileInTestGallery());
+  EXPECT_TRUE(got_correct_details.WaitUntilSatisfied());
+}
+
 // http://crbug.com/390979
-IN_PROC_BROWSER_TEST_F(MediaGalleriesPrivateGalleryWatchApiTest,
+IN_PROC_BROWSER_TEST_F(MediaGalleriesGalleryWatchApiTest,
                        DISABLED_RemoveListenerAndModifyGallery) {
   if (!GalleryWatchesSupported())
     return;
 
-  SetupGalleryWatches();
-
   // Add a gallery watch listener.
   ExecuteCmdAndCheckReply(kAddGalleryChangedListenerCmd,
                           kAddGalleryChangedListenerOK);
+  SetupGalleryWatches();
+
   // Modify gallery contents.
   ExtensionTestMessageListener gallery_change_event_received(
-      kGalleryChangedEventReceived, false  /* no reply */);
+      kGalleryChangedEventReceived, false /* no reply */);
   ASSERT_TRUE(AddNewFileInTestGallery());
   EXPECT_TRUE(gallery_change_event_received.WaitUntilSatisfied());
 
   // Remove gallery watch listener.
   ExecuteCmdAndCheckReply(kRemoveGalleryChangedListenerCmd,
                           kRemoveGalleryChangedListenerOK);
 
   // No listener, modify gallery contents.
   ASSERT_TRUE(AddNewFileInTestGallery());
 
   // Remove gallery watch.
   ExecuteCmdAndCheckReply(kRemoveGalleryWatchCmd, kRemoveGalleryWatchOK);
 }
 
-IN_PROC_BROWSER_TEST_F(MediaGalleriesPrivateGalleryWatchApiTest,
+IN_PROC_BROWSER_TEST_F(MediaGalleriesGalleryWatchApiTest,
                        SetupGalleryWatchWithoutListeners) {
   if (!GalleryWatchesSupported())
     return;
 
-  SetupGalleryWatches();
-
-  // No listeners, modify gallery contents.
-  ExtensionTestMessageListener gallery_change_event_received(
-      kGalleryChangedEventReceived, false  /* no reply */);
-  ASSERT_TRUE(AddNewFileInTestGallery());
-
-  // Remove gallery watch.
-  ExecuteCmdAndCheckReply(kRemoveGalleryWatchCmd, kRemoveGalleryWatchOK);
+  ExecuteCmdAndCheckReply(kSetupWatchOnUnlistenedValidGalleriesCmd,
+                          kAddGalleryWatchRequestRuntimeError);
 }
 
-IN_PROC_BROWSER_TEST_F(MediaGalleriesPrivateGalleryWatchApiTest,
+IN_PROC_BROWSER_TEST_F(MediaGalleriesGalleryWatchApiTest,
                        SetupGalleryChangedListenerWithoutWatchers) {
   // Add gallery watch listener.
   ExecuteCmdAndCheckReply(kAddGalleryChangedListenerCmd,
                           kAddGalleryChangedListenerOK);
 
   // Modify gallery contents. Listener should not get called because add watch
   // request was not called.
   ExtensionTestMessageListener gallery_change_event_received(
-      kGalleryChangedEventReceived, false  /* no reply */);
+      kGalleryChangedEventReceived, false /* no reply */);
   ASSERT_TRUE(AddNewFileInTestGallery());
 
   // Remove gallery watch listener.
   ExecuteCmdAndCheckReply(kRemoveGalleryChangedListenerCmd,
                           kRemoveGalleryChangedListenerOK);
 }
 
-IN_PROC_BROWSER_TEST_F(MediaGalleriesPrivateGalleryWatchApiTest,
+IN_PROC_BROWSER_TEST_F(MediaGalleriesGalleryWatchApiTest,
                        SetupWatchOnInvalidGallery) {
+  // Add gallery watch listener.
+  ExecuteCmdAndCheckReply(kAddGalleryChangedListenerCmd,
+                          kAddGalleryChangedListenerOK);
   // Set up a invalid gallery watch.
-  ExtensionTestMessageListener invalid_gallery_watch_request_finished(
-      kAddGalleryWatchRequestFailed, false  /* no reply */);
-  ExecuteCmdAndCheckReply(kSetupWatchOnInvalidGalleryCmd, kAddGalleryWatchOK);
-  EXPECT_TRUE(invalid_gallery_watch_request_finished.WaitUntilSatisfied());
+  ExecuteCmdAndCheckReply(
+      kSetupWatchOnInvalidGalleryCmd, kAddGalleryWatchRequestFailed);
 }
 
-IN_PROC_BROWSER_TEST_F(MediaGalleriesPrivateGalleryWatchApiTest,
-                       GetAllGalleryWatch) {
+IN_PROC_BROWSER_TEST_F(MediaGalleriesGalleryWatchApiTest, GetAllGalleryWatch) {
+  // Add gallery watch listener.
+  ExecuteCmdAndCheckReply(kAddGalleryChangedListenerCmd,
+                          kAddGalleryChangedListenerOK);
+
   // Gallery watchers are not yet added.
-  // chrome.mediaGalleriesPrivate.getAllGalleryWatch should return an empty
+  // chrome.mediaGalleries.getAllGalleryWatch should return an empty
   // list.
   ExtensionTestMessageListener initial_get_all_check_finished(
-      kNoGalleryWatchesInstalled, false  /* no reply */);
+      kNoGalleryWatchesInstalled, false /* no reply */);
   ExecuteCmdAndCheckReply(kGetAllWatchedGalleryIdsCmd, kGetAllGalleryWatchOK);
   EXPECT_TRUE(initial_get_all_check_finished.WaitUntilSatisfied());
 
   if (!GalleryWatchesSupported())
     return;
 
   SetupGalleryWatches();
 
-  // chrome.mediaGalleriesPrivate.getAllGalleryWatch should return the
+  // chrome.mediaGalleries.getAllGalleryWatch should return the
   // gallery identifiers.
   ExtensionTestMessageListener get_all_watched_galleries_finished(
-      kGalleryWatchesCheck, false  /* no reply */);
+      kGalleryWatchesCheck, false /* no reply */);
   ExecuteCmdAndCheckReply(kGetAllWatchedGalleryIdsCmd, kGetAllGalleryWatchOK);
   EXPECT_TRUE(get_all_watched_galleries_finished.WaitUntilSatisfied());
 
   // Remove gallery watch request.
   ExecuteCmdAndCheckReply(kRemoveGalleryWatchCmd, kRemoveGalleryWatchOK);
 
   // Gallery watchers removed.
-  // chrome.mediaGalleriesPrivate.getAllGalleryWatch() should return an empty
+  // chrome.mediaGalleries.getAllGalleryWatch() should return an empty
   // list.
   ExtensionTestMessageListener final_get_all_check_finished(
-      kNoGalleryWatchesInstalled, false  /* no reply */);
+      kNoGalleryWatchesInstalled, false /* no reply */);
   ExecuteCmdAndCheckReply(kGetAllWatchedGalleryIdsCmd, kGetAllGalleryWatchOK);
   EXPECT_TRUE(final_get_all_check_finished.WaitUntilSatisfied());
 }
 
-IN_PROC_BROWSER_TEST_F(MediaGalleriesPrivateGalleryWatchApiTest,
+IN_PROC_BROWSER_TEST_F(MediaGalleriesGalleryWatchApiTest,
                        RemoveAllGalleryWatch) {
+  // Add gallery watch listener.
+  ExecuteCmdAndCheckReply(kAddGalleryChangedListenerCmd,
+                          kAddGalleryChangedListenerOK);
+
   if (!GalleryWatchesSupported())
     return;
 
   SetupGalleryWatches();
 
-  // chrome.mediaGalleriesPrivate.getAllGalleryWatch should return the watched
+  // chrome.mediaGalleries.getAllGalleryWatch should return the watched
   // gallery identifiers.
   ExtensionTestMessageListener get_all_watched_galleries_finished(
-      kGalleryWatchesCheck, false  /* no reply */);
+      kGalleryWatchesCheck, false /* no reply */);
   ExecuteCmdAndCheckReply(kGetAllWatchedGalleryIdsCmd, kGetAllGalleryWatchOK);
   EXPECT_TRUE(get_all_watched_galleries_finished.WaitUntilSatisfied());
 
   // Remove all gallery watchers.
   ExecuteCmdAndCheckReply(kRemoveAllGalleryWatchCmd, kRemoveAllGalleryWatchOK);
 
-  // Gallery watchers removed. chrome.mediaGalleriesPrivate.getAllGalleryWatch
+  // Gallery watchers removed. chrome.mediaGalleries.getAllGalleryWatch
   // should return an empty list.
   ExtensionTestMessageListener final_get_all_check_finished(
-      kNoGalleryWatchesInstalled, false  /* no reply */);
+      kNoGalleryWatchesInstalled, false /* no reply */);
   ExecuteCmdAndCheckReply(kGetAllWatchedGalleryIdsCmd, kGetAllGalleryWatchOK);
   EXPECT_TRUE(final_get_all_check_finished.WaitUntilSatisfied());
 }