[Media Router] Adds a contextual menu item to remove the MR component action.

chrome/browser/extensions/component_migration_helper_unittest.cc

 // Copyright 2016 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 <string>
 #include <vector>
 
 #include "base/memory/ref_counted.h"
 #include "base/prefs/pref_registry_simple.h"
 #include "base/values.h"
@@ skipping 173 matching lines @@
       registry()->enabled_extensions().Contains(migrated_extension_a_->id()));
 }
 
 TEST_F(ComponentMigrationHelperTest, InstallUnregisteredExtension) {
   service()->AddExtension(unregistered_extension_.get());
   EXPECT_FALSE(IsTestComponentActionEnabled());
   EXPECT_TRUE(
       registry()->enabled_extensions().Contains(unregistered_extension_->id()));
 }
 
+TEST_F(ComponentMigrationHelperTest, RemoveComponentAction) {
+  mock_helper_->SetTestComponentActionPref(true);
+
+  EXPECT_CALL(mock_delegate_, HasComponentAction(kTestActionId))
+      .WillOnce(Return(false));
+  EXPECT_CALL(mock_delegate_, AddComponentAction(kTestActionId));
+
+  mock_helper_->OnFeatureEnabled(kTestActionId);
+  EXPECT_TRUE(IsTestComponentActionEnabled());
+
+  EXPECT_CALL(mock_delegate_, HasComponentAction(kTestActionId))
+      .WillOnce(Return(true));
+  EXPECT_CALL(mock_delegate_, RemoveComponentAction(kTestActionId));
+
+  mock_helper_->OnActionRemoved(kTestActionId);
+  EXPECT_FALSE(IsTestComponentActionEnabled());
+}
+
 }  // namespace extensions