[Extensions] Show enterprise policy icon in the context menu when unremovable

chrome/browser/extensions/extension_context_menu_model_unittest.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 "chrome/browser/extensions/extension_context_menu_model.h"
 
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/extensions/api/extension_action/extension_action_api.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_service_test_base.h"
@@ skipping 140 matching lines @@
   service()->AddExtension(extension.get());
 
   scoped_ptr<Browser> browser = CreateBrowser(profile());
   scoped_refptr<ExtensionContextMenuModel> menu(
       new ExtensionContextMenuModel(extension.get(), browser.get()));
 
   ExtensionSystem* system = ExtensionSystem::Get(profile());
   system->management_policy()->UnregisterAllProviders();
 
   // Uninstallation should be, by default, enabled.
-  ASSERT_TRUE(menu->IsCommandIdEnabled(ExtensionContextMenuModel::UNINSTALL));
+  EXPECT_TRUE(menu->IsCommandIdEnabled(ExtensionContextMenuModel::UNINSTALL));
 
   TestManagementPolicyProvider policy_provider(
       TestManagementPolicyProvider::PROHIBIT_MODIFY_STATUS);
   system->management_policy()->RegisterProvider(&policy_provider);
 
   // If there's a policy provider that requires the extension stay enabled, then
   // uninstallation should be disabled.
-  ASSERT_FALSE(menu->IsCommandIdEnabled(ExtensionContextMenuModel::UNINSTALL));
+  EXPECT_FALSE(menu->IsCommandIdEnabled(ExtensionContextMenuModel::UNINSTALL));
+  int uninstall_index =
+      menu->GetIndexOfCommandId(ExtensionContextMenuModel::UNINSTALL);
+  // There should also be an icon to visually indicate why uninstallation is
+  // forbidden.
+  gfx::Image icon;
+  EXPECT_TRUE(menu->GetIconAt(uninstall_index, &icon));
+  EXPECT_FALSE(icon.IsEmpty());
 
   // Don't leave |policy_provider| dangling.
   system->management_policy()->UnregisterProvider(&policy_provider);
 }
 
 // Tests the context menu for a component extension.
 TEST_F(ExtensionContextMenuModelTest, ComponentExtensionContextMenu) {
   InitializeEmptyExtensionService();
 
   std::string name("component");
@@ skipping 185 matching lines @@
              browser_action.get(),
              browser.get(),
              ExtensionContextMenuModel::TRANSITIVELY_VISIBLE,
              nullptr);
   index = GetCommandIndex(menu, visibility_command);
   EXPECT_NE(-1, index);
   EXPECT_EQ(redesign_keep_string, menu->GetLabelAt(index));
 }
 
 }  // namespace extensions