Disable an extension when it is upgraded to a version that requires more

chrome/browser/extensions/extension_browsertests_misc.cc

 // Copyright (c) 2009 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 "base/ref_counted.h"
 #include "chrome/browser/browser.h"
 #include "chrome/browser/browser_list.h"
 #include "chrome/browser/renderer_host/render_view_host.h"
 #include "chrome/browser/extensions/extension_browsertest.h"
 #include "chrome/browser/extensions/extension_host.h"
@@ skipping 133 matching lines @@
 // Tests that installing and uninstalling extensions don't crash with an
 // incognito window open.
 // This test is disabled. see bug 16106.
 IN_PROC_BROWSER_TEST_F(ExtensionBrowserTest, Incognito) {
   // Open an incognito window to the extensions management page.  We just
   // want to make sure that we don't crash while playing with extensions when
   // this guy is around.
   Browser::OpenURLOffTheRecord(browser()->profile(),
                                GURL(chrome::kChromeUIExtensionsURL));
 
-  ASSERT_TRUE(InstallExtension(test_data_dir_.AppendASCII("good.crx")));
+  ASSERT_TRUE(InstallExtension(test_data_dir_.AppendASCII("good.crx"), 1));
   UninstallExtension("ldnnhddmnhbkjipkidpdiheffobcpfmf");
 }
 
 // Tests that we can load extension pages into the tab area and they can call
 // extension APIs.
 IN_PROC_BROWSER_TEST_F(ExtensionBrowserTest, TabContents) {
   ASSERT_TRUE(LoadExtension(
       test_data_dir_.AppendASCII("good").AppendASCII("Extensions")
                     .AppendASCII("behllobkkfkfnphdnhnkndlbkcpglgmj")
                     .AppendASCII("1.0.0.0")));
@@ skipping 243 matching lines @@
   ui_test_utils::ExecuteJavaScriptAndExtractBool(
       host->render_view_host(), L"", L"testDisconnect()", &result);
   EXPECT_TRUE(result);
 
   // Test disconnect is fired on tab close.
   result = false;
   ui_test_utils::ExecuteJavaScriptAndExtractBool(
       host->render_view_host(), L"", L"testDisconnectOnClose()", &result);
   EXPECT_TRUE(result);
 }
+
+// Tests the process of updating an extension to one that requires higher
+// permissions.
+IN_PROC_BROWSER_TEST_F(ExtensionBrowserTest, UpdatePermissions) {
+  TabContents* contents = browser()->GetSelectedTabContents();
+  ASSERT_TRUE(contents);
+
+  // Install the initial version, which should happen just fine.
+  ASSERT_TRUE(InstallExtension(
+      test_data_dir_.AppendASCII("permissions-low-v1.crx"), 1));
+  DCHECK_EQ(0, contents->infobar_delegate_count());
+
+  // Upgrade to a version that wants more permissions. We should disable the
+  // extension and prompt the user to reenable.
+  ASSERT_TRUE(InstallExtension(
+      test_data_dir_.AppendASCII("permissions-high-v2.crx"), -1));
+  EXPECT_EQ(1, contents->infobar_delegate_count());
+
+  ExtensionsService* service = browser()->profile()->GetExtensionsService();
+  EXPECT_EQ(0u, service->extensions()->size());
+  ASSERT_EQ(1u, service->disabled_extensions()->size());
+
+  // Now try reenabling it, which should also dismiss the infobar.
+  service->EnableExtension(service->disabled_extensions()->at(0)->id());
+  EXPECT_EQ(0, contents->infobar_delegate_count());
+  EXPECT_EQ(1u, service->extensions()->size());
+  EXPECT_EQ(0u, service->disabled_extensions()->size());
+}
+
+// Tests that we can uninstall a disabled extension.
+IN_PROC_BROWSER_TEST_F(ExtensionBrowserTest, UninstallDisabled) {
+  // Install and upgrade, so that we have a disabled extension.
+  ASSERT_TRUE(InstallExtension(
+      test_data_dir_.AppendASCII("permissions-low-v1.crx"), 1));
+  ASSERT_TRUE(InstallExtension(
+      test_data_dir_.AppendASCII("permissions-high-v2.crx"), -1));
+
+  ExtensionsService* service = browser()->profile()->GetExtensionsService();
+  EXPECT_EQ(0u, service->extensions()->size());
+  ASSERT_EQ(1u, service->disabled_extensions()->size());
+
+  // Now try uninstalling it.
+  UninstallExtension(service->disabled_extensions()->at(0)->id());
+  EXPECT_EQ(0u, service->extensions()->size());
+  EXPECT_EQ(0u, service->disabled_extensions()->size());
+}