Create --disable-extensions-except switch

chrome/browser/extensions/extension_startup_browsertest.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 <stddef.h>
 #include <stdint.h>
 
 #include <vector>
 
 #include "base/base_switches.h"
 #include "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/path_service.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "build/build_config.h"
 #include "chrome/browser/chrome_notification_types.h"
+#include "chrome/browser/extensions/extension_browsertest.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_util.h"
 #include "chrome/browser/extensions/shared_user_script_master.h"
 #include "chrome/browser/prefs/chrome_pref_service_factory.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/testing_profile.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/notification_details.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/test/browser_test_utils.h"
 #include "extensions/browser/extension_registry.h"
 #include "extensions/browser/extension_system.h"
 #include "extensions/common/extension.h"
 #include "extensions/common/extension_set.h"
 #include "extensions/common/feature_switch.h"
 #include "net/base/filename_util.h"
 
 using extensions::FeatureSwitch;
+using extensions::ExtensionRegistry;
 
 // This file contains high-level startup tests for the extensions system. We've
 // had many silly bugs where command line flags did not get propagated correctly
 // into the services, so we didn't start correctly.
 
 class ExtensionStartupTestBase : public InProcessBrowserTest {
  public:
   ExtensionStartupTestBase() : unauthenticated_load_allowed_(true) {
     num_expected_extensions_ = 3;
   }
@@ skipping 249 matching lines @@
         .AppendASCII("extensions")
         .AppendASCII("app2");
     load_extensions_.push_back(fourth_extension_path.value());
   }
 };
 
 IN_PROC_BROWSER_TEST_F(ExtensionsLoadMultipleTest, Test) {
   WaitForServicesToStart(4, true);
   TestInjection(true, true);
 }
+
+class DisableExtensionsExceptBrowserTest : public ExtensionBrowserTest {
+ public:
+  DisableExtensionsExceptBrowserTest() {}
+
+  void SetUpCommandLine(base::CommandLine* command_line) override;
+
+  ExtensionRegistry* GetExtensionRegistry() {
+    return ExtensionRegistry::Get(browser()->profile());
+  }
+};
+
+void DisableExtensionsExceptBrowserTest::SetUpCommandLine(
+    base::CommandLine* command_line) {
+  ExtensionBrowserTest::SetUpCommandLine(command_line);
+#if defined(OS_WIN)

[Devlin, 2016/07/27 17:00:59]

This kind of platform-specific path logic is handled in base::FilePath I think. 
We should use that instead here so we don't have to worry about this.

[catmullings, 2016/08/04 22:59:31]

Done.

+  command_line->AppendSwitchASCII(switches::kDisableExtensionsExcept,
+                                  "chrome\\test\\data\\extensions\\app_dot_com_"
+                                  "app\\,"
+                                  "chrome\\test\\data\\extensions\\app\\");
+#else
+  command_line->AppendSwitchASCII(switches::kDisableExtensionsExcept,
+                                  "chrome/test/data/extensions/app_dot_com_app/"
+                                  ",chrome/test/data/extensions/app/");
+#endif
+}
+
+// Tests disabling all extensions except those listed
+// (--disable-extensions-except).
+IN_PROC_BROWSER_TEST_F(DisableExtensionsExceptBrowserTest,

[Devlin, 2016/07/27 17:01:00]

This test seems to be failing on the bots.

+                       DisableExtensionsExceptFlag) {
+  EXPECT_FALSE(extension_service()->extensions_enabled());
+
+  // Count the number of component extensions.
+  int component_extensions = 0;
+  for (const scoped_refptr<const extensions::Extension>& extension :
+       GetExtensionRegistry()->enabled_extensions()) {
+    component_extensions +=
+        extensions::Manifest::IsComponentLocation(extension->location()) ? 1
+                                                                         : 0;
+  }
+
+  // Number of enabled extensions should be the sum of the number of component
+  // extensions and the of number extensions installed with
+  // --disable-extensions-except.
+  size_t num_expected_extensions = component_extensions + 2;
+
+  EXPECT_EQ(num_expected_extensions,
+            GetExtensionRegistry()->enabled_extensions().size());
+
+  // Checks that the extensions loaded with the --disable-extensions-except flag
+  // are enabled.
+  bool is_app_dot_com_extension_enabled = false;
+  bool is_app_test_extension_enabled = false;
+  for (const scoped_refptr<const extensions::Extension>& extension :

[Devlin, 2016/07/27 17:01:00]

Given we're looping over all extensions already, it might be simpler to remove
the component_extensions count and just assert that every extension is either a
component or one of the expected extensions (while keeping in the assertions
that both expected extensions were added).

[catmullings, 2016/08/04 22:59:30]

Done.

+       GetExtensionRegistry()->enabled_extensions()) {
+    if (extension->name() == "App Dot Com: The App") {
+      is_app_dot_com_extension_enabled = true;
+    }
+
+    if (extension->name() == "App Test") {
+      is_app_test_extension_enabled = true;
+    }
+  }
+  EXPECT_TRUE(is_app_dot_com_extension_enabled);
+  EXPECT_TRUE(is_app_test_extension_enabled);
+}