Send screenshots back to the client for debugging

chrome/test/webdriver/chromedriver_tests.py

Index: chrome/test/webdriver/chromedriver_tests.py
diff --git a/chrome/test/webdriver/chromedriver_tests.py b/chrome/test/webdriver/chromedriver_tests.py
index 55a7bd37ec7dcb538199b9e0a80b18170d727122..bf86d94b22bba3a7e633bf2771516a767c722a13 100755
--- a/chrome/test/webdriver/chromedriver_tests.py
+++ b/chrome/test/webdriver/chromedriver_tests.py
@@ -148,6 +148,29 @@ class CookieTest(unittest.TestCase):
     self.assertEqual(cookie_dict, None)
 
 
+class ScreenshotTest(unittest.TestCase):
+  """Tests to verify screenshot retrieval"""
+
+  SEARCH = "http://www.google.com/webhp?hl=en"
+  NEWS = "http://www.google.com/news"
+
+  def setUp(self):
+    self._launcher = ChromeDriverLauncher()
+    self._driver = WebDriver(self._launcher.GetURL(), {})
+
+  def tearDown(self):
+    self._driver.quit()
+    self._launcher.Kill()
+
+  def testScreenCapture(self):
+    self._driver.get(self.SEARCH)
+    s1 = self._driver.get_screenshot_as_base64()
+    self._driver.get(self.NEWS)
+    s2 = self._driver.get_screenshot_as_base64()
+    self.assertTrue(len(s1) > 0 and len(s2) > 0)
+    self.assertTrue(len(s1) != len(s2))
+
+
 class SessionTest(unittest.TestCase):
   """Tests dealing with WebDriver sessions."""