Add extension policy tests

chrome/test/functional/policy.py

 #!/usr/bin/env python
 # Copyright (c) 2011 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.
 
 import logging
+import os
 
 import pyauto_functional  # must come before pyauto.
 import policy_base
 import pyauto
 from pyauto_errors import JSONInterfaceError
 
 
 class PolicyTest(policy_base.PolicyTestBase):
   """Tests that the effects of policies are being enforced as expected."""
 
@@ skipping 161 matching lines @@
     self.SetPolicies(policy)
     self.NavigateToURL('about:blank')
     self.assertTrue(self.IsJavascriptEnabled())
 
   def testDisable3DAPIs(self):
     """Tests the policy that disables the 3D APIs."""
     self.assertFalse(self.GetPrefsInfo().Prefs(pyauto.kDisable3DAPIs))
     self.assertTrue(self.IsWebGLEnabled())
 
     self.SetPolicies({
-        'Disable3DAPIs': True
+      'Disable3DAPIs': True
     })
     self.assertTrue(self.GetPrefsInfo().Prefs(pyauto.kDisable3DAPIs))
     # The Disable3DAPIs policy only applies updated values to new renderers.
     self.RestartRenderer()
     self.assertFalse(self.IsWebGLEnabled())
 
+  # Needed for extension tests
+  GOOD_CRX_ID = 'ldnnhddmnhbkjipkidpdiheffobcpfmf'
+  ADBLOCK_CRX_ID = 'dojnnbeimaimaojcialkkgajdnefpgcn'
+  SCREEN_CAPTURE_CRX_ID = 'cpngackimfmofbokmjmljamhdncknpmg'

[frankf, 2012/03/14 01:41:39]

_GOOD_CRX_ID

+
+  def _BuildCRXPath(self, crx_file_name):
+    """Returns the complete path to a crx_file in the data directory.
+
+    Args:
+      crx_file_name: The file name of the extension
+
+    Returns:
+      The full path to the crx in the data directory
+    """
+    return os.path.abspath(os.path.join(self.DataDir(), 'extensions',
+                                        crx_file_name))
+
+  def _AttemptExtensionInstallThatShouldFail(self, crx_file_name):
+    """Attempts to install an extension, raises an exception if installed.
+
+    Args:
+      crx_file_name: The file name of the extension
+    """
+    install_failed = True
+    try:
+      self.InstallExtension(self._BuildCRXPath(crx_file_name))
+      install_failed = False
+    except pyauto_errors.JSONInterfaceError as e:
+      self.assertEqual(e[0], 'Extension could not be installed',
+                       msg='The extension failed to install which is expected. '
+                       ' However it failed due to an unexpected reason: %s'
+                       % e[0])
+    self.assertTrue(install_failed, msg='The extension %s did not throw an '
+                    'exception when installation was attempted.  This most '
+                    'likely means it succeeded, which it should not.')
+
+  def _AttemptExtensionInstallThatShouldPass(self, crx_file_name):
+    """Attempts to install an extension, raises an exception if not installed.
+
+    Args:
+      crx_file_name: The file name of the extension
+    """
+    ext_id = self.InstallExtension(self._BuildCRXPath(crx_file_name))
+    self.assertTrue(self._CheckForExtensionByID(ext_id),
+                    msg='The %s extension was not install even though '
+                    'it should have been.' % crx_file_name)
+
+  def _CheckForExtensionByID(self, extension_id):
+    """Returns if an extension is installed.
+
+    Args:
+      extension_id: The id of the extension
+
+    Returns:
+      True if the extension is installed; False otherwise
+    """
+    all_extensions = [extension['id'] for extension in self.GetExtensionsInfo()]
+    return extension_id in all_extensions
+
+  def _RemoveTestingExtensions(self):
+    """Temporary method that cleans state for extension test.
+
+    Currently the tear down method for policy_base does not clear the user
+    state.  See crosbug.com/27227.
+    """
+    if self._CheckForExtensionByID(PolicyTest.SCREEN_CAPTURE_CRX_ID):

[frankf, 2012/03/14 01:41:39]

self._SCREEN_CAPTURE_CRX_ID

+      self.UninstallExtensionById(PolicyTest.SCREEN_CAPTURE_CRX_ID)
+    if self._CheckForExtensionByID(PolicyTest.GOOD_CRX_ID):
+      self.UninstallExtensionById(PolicyTest.GOOD_CRX_ID)
+      self.NavigateToURL('chrome:extensions')
+    if self._CheckForExtensionByID(PolicyTest.ADBLOCK_CRX_ID):
+      self.UninstallExtensionById(PolicyTest.ADBLOCK_CRX_ID)
+    # There is an issue where if you uninstall and reinstall and extension
+    # quickly with self.InstallExtension, the reinstall fails.  This is a hack
+    # to fix it.  Bug coming soon.
+    self.NavigateToURL('chrome:extensions')
+
+  def testExtensionInstallPopulatedBlacklist(self):
+    """Verify blacklisted extensions cannot be installed."""
+    # TODO(krisr): Remove this when crosbug.com/27227 is fixed.
+    self._RemoveTestingExtensions()
+    # Blacklist good.crx
+    self.SetPolicies({
+      'ExtensionInstallBlacklist': [PolicyTest.GOOD_CRX_ID]
+    })
+    self._AttemptExtensionInstallThatShouldFail('good.crx')
+    # Check adblock is installed.
+    self._AttemptExtensionInstallThatShouldPass('adblock.crx')
+
+  def testExtensionInstallFailWithGlobalBlacklist(self):
+    """Verify no extensions can be installed when all are blacklisted."""
+    # TODO(krisr): Remove this when crosbug.com/27227 is fixed.
+    self._RemoveTestingExtensions()
+    # Block installs of all extensions
+    self.SetPolicies({
+      'ExtensionInstallBlacklist': ['*']
+    })
+    self._AttemptExtensionInstallThatShouldFail('good.crx')
+    self._AttemptExtensionInstallThatShouldFail('adblock.crx')
+
+  def testExtensionInstallWithGlobalBlacklistAndWhitelistedExtension(self):
+    """Verify whitelisted extension is installed when all are blacklisted."""
+    # TODO(krisr): Remove this when crosbug.com/27227 is fixed.
+    self._RemoveTestingExtensions()
+    # Block installs of all extensions, but whitelist adblock.crx
+    self.SetPolicies({
+      'ExtensionInstallBlacklist': ['*'],
+      'ExtensionInstallWhitelist': [PolicyTest.ADBLOCK_CRX_ID]
+    })
+    self._AttemptExtensionInstallThatShouldFail('good.crx')
+    self._AttemptExtensionInstallThatShouldPass('adblock.crx')
+
+  # TODO(krisr): Enable this test once we figure out why it isn't downloading
+  # the extension.
+  def testExtensionInstallFromForceList(self):
+    """Verify force install extensions are installed."""
+    # TODO(krisr): Remove this when crosbug.com/27227 is fixed.
+    self._RemoveTestingExtensions()
+    # Force an extension download from the webstore.
+    self.SetPolicies({
+      'ExtensionInstallForcelist': [PolicyTest.SCREEN_CAPTURE_CRX_ID],
+    })
+    # Give the system 30 seconds to go get this extension.  We are not sure how
+    # long it will take the policy to take affect and download the extension.
+    self.assertTrue(self.WaitUntil(lambda:
+      self._CheckForExtensionByID(PolicyTest.SCREEN_CAPTURE_CRX_ID),
+      expect_retval=True),
+      msg='The force install extension was never installed.')
+
 
 if __name__ == '__main__':
   pyauto_functional.Main()