Index: chrome/browser/extensions/component_migration_helper_unittest.cc |
diff --git a/chrome/browser/extensions/component_migration_helper_unittest.cc b/chrome/browser/extensions/component_migration_helper_unittest.cc |
index 02489928eddbc36fda053c2793679161a98c95d5..4b0ebb85a52a56be9f8a8a19c662ec90d38c3ca0 100644 |
--- a/chrome/browser/extensions/component_migration_helper_unittest.cc |
+++ b/chrome/browser/extensions/component_migration_helper_unittest.cc |
@@ -191,4 +191,22 @@ TEST_F(ComponentMigrationHelperTest, InstallUnregisteredExtension) { |
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 |