screenshot disabling policy tests

chrome/test/functional/policy.py

 #!/usr/bin/env python
 # 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.
 
 import logging
 import os
 
 import pyauto_functional  # must come before pyauto.
 import policy_base
@@ skipping 670 matching lines @@
 
     # With the policy set, the cookie should be gone after a restart.
     self.SetUserPolicy({
       'ClearSiteDataOnExit': True
     })
     self.RestartBrowser(clear_profile=False)
     self.assertFalse(
         self.GetCookie(pyauto.GURL(cookie_url)),
         msg='Cookie present on ' + cookie_url + '.');
 
+  def testDisableScreenshotFile(self):
+    """Verify screenshot keyboard shortcut can be disabled by policy."""
+    if not self.IsChromeOS():
+      return  # Screenshot accelerator available only on ChromeOS
+
+    download_dir = self.GetDownloadDirectory().value()
+
+    def FilesCount():
+      return len(os.listdir(download_dir))
+
+    # Make sure screenshots actually go to downloads directory.
+    expected_count = FilesCount() + 1
+    self.assertTrue(self.ApplyAshAccelerator(pyauto.TAKE_SCREENSHOT))
+    self.assertTrue(self.WaitUntil(FilesCount, expect_retval=expected_count))
+
+    # Real testing here: check if disabling screenshots works.
+    policy = {
+      'DisableScreenshots': True
+    }
+    self.SetUserPolicy(policy)
+    self.assertTrue(self.ApplyAshAccelerator(TAKE_SCREENSHOT))
+    self.assertFalse(self.WaitUntil(lambda: files_count() != expected_count,
+                                    timeout=5, expect_retval=True))
+
+  def testDisableScreenshotFeedback(self):
+    """Verify screenshot in feedback form can be disabled by policy."""
+
+    def SafeWaitUntilNavigationCompletes(tab_index):
+      self.assertTrue(self.WaitUntil(lambda: self.GetTabCount() > tab_index,
+                                     expect_retval=True))
+      self.WaitUntilNavigationCompletes(tab_index=tab_index)
+
+    # Make sure screenshot is normally displayed in feedback form.
+    self.assertTrue(self.ApplyAccelerator(pyauto.IDC_FEEDBACK))
+    SafeWaitUntilNavigationCompletes(1)
+    self.WaitForDomNode('//img[starts-with(@id, "current-screenshots")]',
+                        tab_index=1)
+
+    # The feedback tab is a singleton tab, so close it first before trying to
+    # reopen. Otherwise it might not reload.
+    self.GetBrowserWindow(0).GetTab(1).Close(True)
+
+    # Real testing here: check if disabling screenshots works.
+    policy = {
+      'DisableScreenshots': True
+    }
+    self.SetUserPolicy(policy)
+    self.assertTrue(self.ApplyAccelerator(pyauto.IDC_FEEDBACK))
+    SafeWaitUntilNavigationCompletes(1)
+    self.assertRaises(pyauto.JSONInterfaceError, lambda: self.WaitForDomNode(
+        '//img[starts-with(@id, "current-screenshots")]', tab_index=1,
+        timeout=3))
+
 
 if __name__ == '__main__':
   pyauto_functional.Main()