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 c7d7b205af62b626fd6d2d5a204c5bc6ac9f57a3..be4bfa18300d185f0c188dd78008b4f9a861e4ec 100755
--- a/chrome/test/webdriver/chromedriver_tests.py
+++ b/chrome/test/webdriver/chromedriver_tests.py
@@ -116,6 +116,29 @@ class BasicTest(unittest.TestCase):
     launcher.Kill()
 
 
+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(), 'chrome', 'any')
+
+  def tearDown(self):
+    self._driver.quit()
+    self._launcher.Kill()
+
+  def testScreenCapture
+    self._driver_.get(self.EARCH)
+    s1 = self._driver.get_screenshot_as_base64()
+    self._driver.get(self.NEWS)
+    s2 = self._driver.get_screenshot_as_base64()
+    self.assertTrue(len(s1) > 0 && (len)s2 > 0)
+    self.assertTrue(len(s1) != len(s2))
+
+
 class CookieTest(unittest.TestCase):
   """Cookie test for the json webdriver protocol"""