Show extension icons next to their top-level context menu items....

chrome/browser/extensions/extension_menu_manager_unittest.cc

 // Copyright (c) 2010 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 <vector>
 
 #include "base/json/json_reader.h"
 #include "base/path_service.h"
 #include "base/scoped_temp_dir.h"
 #include "base/values.h"
 #include "chrome/browser/extensions/extension_menu_manager.h"
 #include "chrome/browser/extensions/extension_message_service.h"
+#include "chrome/browser/extensions/test_extension_prefs.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/extensions/extension_constants.h"
 #include "chrome/common/notification_service.h"
 #include "chrome/test/testing_profile.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "webkit/glue/context_menu.h"
 
 using testing::_;
 using testing::AtLeast;
 using testing::Return;
 using testing::SaveArg;
 
 // Base class for tests.
 class ExtensionMenuManagerTest : public testing::Test {
  public:
   ExtensionMenuManagerTest() {}
   ~ExtensionMenuManagerTest() {}
 
-  // Returns a test item with some default values you can override if you want
-  // to by passing in |properties| (currently just extension_id). Caller owns
-  // the return value and is responsible for freeing it.
-  static ExtensionMenuItem* CreateTestItem(DictionaryValue* properties) {
-    std::string extension_id = "0123456789";  // A default dummy value.
-    if (properties && properties->HasKey(L"extension_id"))
-      EXPECT_TRUE(properties->GetString(L"extension_id", &extension_id));
-
+  // Returns a test item.
+  static ExtensionMenuItem* CreateTestItem(Extension* extension) {
     ExtensionMenuItem::Type type = ExtensionMenuItem::NORMAL;
     ExtensionMenuItem::ContextList contexts(ExtensionMenuItem::ALL);
     ExtensionMenuItem::ContextList enabled_contexts = contexts;
-    std::string title = "test";
+    return new ExtensionMenuItem(extension->id(), "test", false, type, contexts,
+                                 enabled_contexts);
+  }
 
-    return new ExtensionMenuItem(extension_id, title, false, type, contexts,
-                                 enabled_contexts);
+  // Creates and returns a test Extension. The caller does *not* own the return
+  // value.
+  Extension* AddExtension(std::string name) {
+    return prefs_.AddExtension(name);
   }
 
  protected:
   ExtensionMenuManager manager_;
+  TestExtensionPrefs prefs_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(ExtensionMenuManagerTest);
 };
 
 // Tests adding, getting, and removing items.
 TEST_F(ExtensionMenuManagerTest, AddGetRemoveItems) {
+  Extension* extension = AddExtension("test");
+
   // Add a new item, make sure you can get it back.
-  ExtensionMenuItem* item1 = CreateTestItem(NULL);
+  ExtensionMenuItem* item1 = CreateTestItem(extension);
   ASSERT_TRUE(item1 != NULL);
-  int id1 = manager_.AddContextItem(item1);  // Ownership transferred.
+  int id1 = manager_.AddContextItem(extension, item1);
   ASSERT_GT(id1, 0);
   ASSERT_EQ(item1, manager_.GetItemById(id1));
   const ExtensionMenuItem::List* items =
       manager_.MenuItems(item1->extension_id());
   ASSERT_EQ(1u, items->size());
   ASSERT_EQ(item1, items->at(0));
 
   // Add a second item, make sure it comes back too.
-  ExtensionMenuItem* item2 = CreateTestItem(NULL);
-  int id2 = manager_.AddContextItem(item2);  // Ownership transferred.
+  ExtensionMenuItem* item2 = CreateTestItem(extension);
+  int id2 = manager_.AddContextItem(extension, item2);
   ASSERT_GT(id2, 0);
   ASSERT_NE(id1, id2);
   ASSERT_EQ(item2, manager_.GetItemById(id2));
   items = manager_.MenuItems(item2->extension_id());
   ASSERT_EQ(2u, items->size());
   ASSERT_EQ(item1, items->at(0));
   ASSERT_EQ(item2, items->at(1));
 
   // Try adding item 3, then removing it.
-  ExtensionMenuItem* item3 = CreateTestItem(NULL);
+  ExtensionMenuItem* item3 = CreateTestItem(extension);
   std::string extension_id = item3->extension_id();
-  int id3 = manager_.AddContextItem(item3);  // Ownership transferred.
+  int id3 = manager_.AddContextItem(extension, item3);
   ASSERT_GT(id3, 0);
   ASSERT_EQ(item3, manager_.GetItemById(id3));
   ASSERT_EQ(3u, manager_.MenuItems(extension_id)->size());
   ASSERT_TRUE(manager_.RemoveContextMenuItem(id3));
   ASSERT_EQ(NULL, manager_.GetItemById(id3));
   ASSERT_EQ(2u, manager_.MenuItems(extension_id)->size());
 
   // Make sure removing a non-existent item returns false.
   ASSERT_FALSE(manager_.RemoveContextMenuItem(5));
 }
 
 // Test adding/removing child items.
 TEST_F(ExtensionMenuManagerTest, ChildFunctions) {
-  DictionaryValue properties;
-  properties.SetString(L"extension_id", "1111");
-  ExtensionMenuItem* item1 = CreateTestItem(&properties);
+  Extension* extension1 = AddExtension("1111");
+  Extension* extension2 = AddExtension("2222");
+  Extension* extension3 = AddExtension("3333");
 
-  properties.SetString(L"extension_id", "2222");
-  ExtensionMenuItem* item2 = CreateTestItem(&properties);
-  ExtensionMenuItem* item2_child = CreateTestItem(&properties);
-  ExtensionMenuItem* item2_grandchild = CreateTestItem(&properties);
+  ExtensionMenuItem* item1 = CreateTestItem(extension1);
+  ExtensionMenuItem* item2 = CreateTestItem(extension2);
+  ExtensionMenuItem* item2_child = CreateTestItem(extension2);
+  ExtensionMenuItem* item2_grandchild = CreateTestItem(extension2);
 
   // This third item we expect to fail inserting, so we use a scoped_ptr to make
   // sure it gets deleted.
-  properties.SetString(L"extension_id", "3333");
-  scoped_ptr<ExtensionMenuItem> item3(CreateTestItem(&properties));
+  scoped_ptr<ExtensionMenuItem> item3(CreateTestItem(extension3));
 
   // Add in the first two items.
-  int id1 = manager_.AddContextItem(item1);  // Ownership transferred.
-  int id2 = manager_.AddContextItem(item2);  // Ownership transferred.
+  int id1 = manager_.AddContextItem(extension1, item1);
+  int id2 = manager_.AddContextItem(extension2, item2);
 
   ASSERT_NE(id1, id2);
 
   // Try adding item3 as a child of item2 - this should fail because item3 has
   // a different extension id.
   ASSERT_EQ(0, manager_.AddChildItem(id2, item3.get()));
 
   // Add item2_child as a child of item2.
   int id2_child = manager_.AddChildItem(id2, item2_child);
   ASSERT_GT(id2_child, 0);
@@ skipping 18 matching lines @@
 
   // Remove child2_item.
   ASSERT_TRUE(manager_.RemoveContextMenuItem(id2_child));
   ASSERT_EQ(1u, manager_.MenuItems(item2->extension_id())->size());
   ASSERT_EQ(item2, manager_.MenuItems(item2->extension_id())->at(0));
   ASSERT_EQ(0, item2->child_count());
 }
 
 // Tests changing parents.
 TEST_F(ExtensionMenuManagerTest, ChangeParent) {
+  Extension* extension1 = AddExtension("1111");
+
   // First create two items and add them both to the manager.
-  ExtensionMenuItem* item1 = CreateTestItem(NULL);
-  ExtensionMenuItem* item2 = CreateTestItem(NULL);
+  ExtensionMenuItem* item1 = CreateTestItem(extension1);
+  ExtensionMenuItem* item2 = CreateTestItem(extension1);
 
-  int id1 = manager_.AddContextItem(item1);
+  int id1 = manager_.AddContextItem(extension1, item1);
   ASSERT_GT(id1, 0);
-  int id2 = manager_.AddContextItem(item2);
+  int id2 = manager_.AddContextItem(extension1, item2);
   ASSERT_GT(id2, 0);
 
   const ExtensionMenuItem::List* items =
       manager_.MenuItems(item1->extension_id());
   ASSERT_EQ(2u, items->size());
   ASSERT_EQ(item1, items->at(0));
   ASSERT_EQ(item2, items->at(1));
 
   // Now create a third item, initially add it as a child of item1, then move
   // it to be a child of item2.
-  ExtensionMenuItem* item3 = CreateTestItem(NULL);
+  ExtensionMenuItem* item3 = CreateTestItem(extension1);
 
   int id3 = manager_.AddChildItem(id1, item3);
   ASSERT_GT(id3, 0);
   ASSERT_EQ(1, item1->child_count());
   ASSERT_EQ(item3, item1->children()[0]);
 
   ASSERT_TRUE(manager_.ChangeParent(id3, id2));
   ASSERT_EQ(0, item1->child_count());
   ASSERT_EQ(1, item2->child_count());
   ASSERT_EQ(item3, item2->children()[0]);
@@ skipping 30 matching lines @@
   // Move item2 to be a top-level item.
   ASSERT_TRUE(manager_.ChangeParent(id2, 0));
   items = manager_.MenuItems(item1->extension_id());
   ASSERT_EQ(2u, items->size());
   ASSERT_EQ(item1, items->at(0));
   ASSERT_EQ(item2, items->at(1));
   ASSERT_EQ(1, item1->child_count());
   ASSERT_EQ(item3, item1->children()[0]);
 
   // Make sure you can't move a node to be a child of another extension's item.
-  DictionaryValue properties;
-  properties.SetString(L"extension_id", "4444");
-  ExtensionMenuItem* item4 = CreateTestItem(&properties);
-  int id4 = manager_.AddContextItem(item4);
+  Extension* extension2 = AddExtension("2222");
+  ExtensionMenuItem* item4 = CreateTestItem(extension2);
+  int id4 = manager_.AddContextItem(extension2, item4);
   ASSERT_GT(id4, 0);
   ASSERT_FALSE(manager_.ChangeParent(id4, id1));
   ASSERT_FALSE(manager_.ChangeParent(id1, id4));
 
   // Make sure you can't make an item be it's own parent.
   ASSERT_FALSE(manager_.ChangeParent(id1, id1));
 }
 
 // Tests that we properly remove an extension's menu item when that extension is
 // unloaded.
 TEST_F(ExtensionMenuManagerTest, ExtensionUnloadRemovesMenuItems) {
-  ScopedTempDir temp_dir;
-  ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
-
   NotificationService* notifier = NotificationService::current();
   ASSERT_TRUE(notifier != NULL);
 
   // Create a test extension.
-  DictionaryValue extension_properties;
-  extension_properties.SetString(extension_manifest_keys::kVersion, "1");
-  extension_properties.SetString(extension_manifest_keys::kName, "Test");
-  Extension extension(temp_dir.path().AppendASCII("extension"));
-  std::string errors;
-  ASSERT_TRUE(extension.InitFromValue(extension_properties,
-                                      false,  // No public key required.
-                                      &errors)) << errors;
+  Extension* extension1 = AddExtension("1111");
 
   // Create an ExtensionMenuItem and put it into the manager.
-  DictionaryValue item_properties;
-  item_properties.SetString(L"extension_id", extension.id());
-  ExtensionMenuItem* item1 = CreateTestItem(&item_properties);
-  ASSERT_EQ(extension.id(), item1->extension_id());
-  int id1 = manager_.AddContextItem(item1);  // Ownership transferred.
+  ExtensionMenuItem* item1 = CreateTestItem(extension1);
+  ASSERT_EQ(extension1->id(), item1->extension_id());
+  int id1 = manager_.AddContextItem(extension1, item1);
   ASSERT_GT(id1, 0);
-  ASSERT_EQ(1u, manager_.MenuItems(extension.id())->size());
+  ASSERT_EQ(1u, manager_.MenuItems(extension1->id())->size());
 
   // Create a menu item with a different extension id and add it to the manager.
-  std::string alternate_extension_id = "0000";
-  item_properties.SetString(L"extension_id", alternate_extension_id);
-  ExtensionMenuItem* item2 = CreateTestItem(&item_properties);
+  Extension* extension2 = AddExtension("2222");
+  ExtensionMenuItem* item2 = CreateTestItem(extension2);
   ASSERT_NE(item1->extension_id(), item2->extension_id());
-  int id2 = manager_.AddContextItem(item2);  // Ownership transferred.
+  int id2 = manager_.AddContextItem(extension2, item2);
   ASSERT_GT(id2, 0);
 
   // Notify that the extension was unloaded, and make sure the right item is
   // gone.
   notifier->Notify(NotificationType::EXTENSION_UNLOADED,
                    Source<Profile>(NULL),
-                   Details<Extension>(&extension));
-  ASSERT_EQ(NULL, manager_.MenuItems(extension.id()));
-  ASSERT_EQ(1u, manager_.MenuItems(alternate_extension_id)->size());
+                   Details<Extension>(extension1));
+  ASSERT_EQ(NULL, manager_.MenuItems(extension1->id()));
+  ASSERT_EQ(1u, manager_.MenuItems(extension2->id())->size());
   ASSERT_TRUE(manager_.GetItemById(id1) == NULL);
   ASSERT_TRUE(manager_.GetItemById(id2) != NULL);
 }
 
 // A mock message service for tests of ExtensionMenuManager::ExecuteCommand.
 class MockExtensionMessageService : public ExtensionMessageService {
  public:
   explicit MockExtensionMessageService(Profile* profile) :
       ExtensionMessageService(profile) {}
 
@@ skipping 15 matching lines @@
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockTestingProfile);
 };
 
 // Tests the RemoveAll functionality.
 TEST_F(ExtensionMenuManagerTest, RemoveAll) {
   // Try removing all items for an extension id that doesn't have any items.
   manager_.RemoveAllContextItems("CCCC");
 
-  // Add 2 top-level and one child item for extension id AAAA.
-  DictionaryValue properties;
-  properties.SetString(L"extension_id", "AAAA");
-  ExtensionMenuItem* item1 = CreateTestItem(&properties);
-  ExtensionMenuItem* item2 = CreateTestItem(&properties);
-  ExtensionMenuItem* item3 = CreateTestItem(&properties);
-  int id1 = manager_.AddContextItem(item1);
-  int id2 = manager_.AddContextItem(item2);
+  // Add 2 top-level and one child item for extension 1.
+  Extension* extension1 = AddExtension("1111");
+  ExtensionMenuItem* item1 = CreateTestItem(extension1);
+  ExtensionMenuItem* item2 = CreateTestItem(extension1);
+  ExtensionMenuItem* item3 = CreateTestItem(extension1);
+  int id1 = manager_.AddContextItem(extension1, item1);
+  int id2 = manager_.AddContextItem(extension1, item2);
   EXPECT_GT(id1, 0);
   EXPECT_GT(id2, 0);
   int id3 = manager_.AddChildItem(id1, item3);
   EXPECT_GT(id3, 0);
 
-  // Add one top-level item for extension id BBBB.
-  properties.SetString(L"extension_id", "BBBB");
-  ExtensionMenuItem* item4 = CreateTestItem(&properties);
-  manager_.AddContextItem(item4);
+  // Add one top-level item for extension 2.
+  Extension* extension2 = AddExtension("2222");
+  ExtensionMenuItem* item4 = CreateTestItem(extension2);
+  manager_.AddContextItem(extension2, item4);
 
-  EXPECT_EQ(2u, manager_.MenuItems("AAAA")->size());
-  EXPECT_EQ(1u, manager_.MenuItems("BBBB")->size());
+  EXPECT_EQ(2u, manager_.MenuItems(extension1->id())->size());
+  EXPECT_EQ(1u, manager_.MenuItems(extension2->id())->size());
 
-  // Remove the BBBB item.
-  manager_.RemoveAllContextItems("BBBB");
-  EXPECT_EQ(2u, manager_.MenuItems("AAAA")->size());
-  EXPECT_EQ(NULL, manager_.MenuItems("BBBB"));
+  // Remove extension2's item.
+  manager_.RemoveAllContextItems(extension2->id());
+  EXPECT_EQ(2u, manager_.MenuItems(extension1->id())->size());
+  EXPECT_EQ(NULL, manager_.MenuItems(extension2->id()));
 
-  // Remove the AAAA items.
-  manager_.RemoveAllContextItems("AAAA");
-  EXPECT_EQ(NULL, manager_.MenuItems("AAAA"));
+  // Remove extension1's items.
+  manager_.RemoveAllContextItems(extension1->id());
+  EXPECT_EQ(NULL, manager_.MenuItems(extension1->id()));
 }
 
 TEST_F(ExtensionMenuManagerTest, ExecuteCommand) {
   MessageLoopForUI message_loop;
   ChromeThread ui_thread(ChromeThread::UI, &message_loop);
 
   MockTestingProfile profile;
 
   scoped_refptr<MockExtensionMessageService> mock_message_service =
       new MockExtensionMessageService(&profile);
 
   ContextMenuParams params;
   params.media_type = WebKit::WebContextMenuData::MediaTypeImage;
   params.src_url = GURL("http://foo.bar/image.png");
   params.page_url = GURL("http://foo.bar");
   params.selection_text = L"Hello World";
   params.is_editable = false;
 
-  ExtensionMenuItem* item = CreateTestItem(NULL);
-  int id = manager_.AddContextItem(item);  // Ownership transferred.
+  Extension* extension = AddExtension("test");
+  ExtensionMenuItem* item = CreateTestItem(extension);
+  int id = manager_.AddContextItem(extension, item);
   ASSERT_GT(id, 0);
 
   EXPECT_CALL(profile, GetExtensionMessageService())
       .Times(1)
       .WillOnce(Return(mock_message_service.get()));
 
   EXPECT_CALL(profile, IsOffTheRecord())
       .Times(AtLeast(1))
       .WillRepeatedly(Return(false));
 
@@ skipping 36 matching lines @@
   ASSERT_EQ(params.page_url.spec(), tmp);
 
   std::wstring wide_tmp;
   ASSERT_TRUE(info->GetString(L"selectionText", &wide_tmp));
   ASSERT_EQ(params.selection_text, wide_tmp);
 
   bool bool_tmp = true;
   ASSERT_TRUE(info->GetBoolean(L"editable", &bool_tmp));
   ASSERT_EQ(params.is_editable, bool_tmp);
 }