Adding tests for plugins section in content settings....

functional/plugins.py

 #!/usr/bin/python
 # Copyright (c) 2010 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 re
 
 import pyauto_functional  # Must be imported before pyauto
@@ skipping 121 matching lines @@
       if 'Shockwave Flash' == plugin_name:
         continue  # cannot reload file:// flash URL - crbug.com/47249
       # Enable
       self._TogglePlugin(plugin_name)
       self.GetBrowserWindow(0).GetTab(0).Reload()
       self.assertTrue([x for x in self.GetBrowserInfo()['child_processes']
                        if x['type'] == 'Plug-in' and
                        re.search(plugin_name, x['name'])])
       self.assertTrue(self._IsEnabled(plugin_name), plugin_name)
 
+  def testBlockAllPlugins(self):
+    """Verify that all the plugins can be blocked.
+    Verifying by checking that flash plugin was blocked.
+    """
+    flash_url = self.GetFileURLForPath(os.path.join(
+        self.DataDir(), 'plugin', 'flash-clicktoplay.html'))
+    self.NavigateToURL(flash_url)
+    flash_pid = self._GetPluginPID('Shockwave Flash')
+    self.assertTrue(flash_pid, msg='No plugin process for Shockwave Flash')
+    # Killing the flash process as it takes a while before the plugin
+    # process is terminated even though there are no tabs using it.
+    self.Kill(flash_pid)
+    self.assertTrue(self.WaitUntil(
+        lambda: self._GetPluginPID('Shockwave Flash') is None),
+        msg='Expected Shockwave Flash plugin to die after killing')
+
+    # Set the preference to block all plugins.
+    self.SetPrefs(pyauto.kDefaultContentSettings, {'plugins': 2})
+
+    self.GetBrowserWindow(0).GetTab(0).Reload()
+    self.assertFalse(self._GetPluginPID('Shockwave Flash'),
+                     msg='Plug-in not blocked.')
+
+  def testAllowPluginException(self):
+    """Verify that plugins can be allowed on a domain by adding
+    an exception(s)."""
+    # Set the preference to block all plugins.
+    self.SetPrefs(pyauto.kDefaultContentSettings, {'plugins': 2})
+
+    flash_url = self.GetFileURLForPath(os.path.join(
+        self.DataDir(), 'plugin', 'flash-clicktoplay.html'))
+    self.NavigateToURL(flash_url)
+    # Check that plugins are blocked.
+    self.assertFalse(self._GetPluginPID('Shockwave Flash'),
+                     msg='Plug-in not blocked.')
+
+    # Add an exception to allow plugins on hulu.com.
+    self.SetPrefs(pyauto.kContentSettingsPatterns,
+                 {'[*.]hulu.com': {'plugins': 1}})
+    self.AppendTab(pyauto.GURL('http://www.hulu.com'))
+    self.assertTrue(self._GetPluginPID('Shockwave Flash'),
+                    msg='No plugin process for Shockwave Flash')
+
+  def testBlockPluginException(self):
+    """Verify that plugins can be blocked on a domain by adding
+    an exception(s)."""
+    # We are using the same live site in order to detect if the web page
+    # is using shockwave flash process
+    self.NavigateToURL('http://www.hulu.com')
+    pid = self._GetPluginPID('Shockwave Flash')
+    self.assertTrue(pid, msg='No plugin process for Shockwave Flash')
+    self.Kill(pid)
+    self.assertTrue(self.WaitUntil(
+        lambda: self._GetPluginPID('Shockwave Flash') is None),
+        msg='Expected Shockwave Flash plugin to die after killing')
+
+    # Add an exception to block plugins on hulu.com.
+    self.SetPrefs(pyauto.kContentSettingsPatterns,
+                 {'[*.]hulu.com': {'plugins': 2}})
+    self.GetBrowserWindow(0).GetTab(0).Reload()
+    self.assertFalse(self._GetPluginPID('Shockwave Flash'),
+                     msg='Plug-in not blocked.')
+
 
 if __name__ == '__main__':
   pyauto_functional.Main()