Send screenshots back to the client for debugging

chrome/test/webdriver/chromedriver_tests.py

 #!/usr/bin/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.
 
 """Tests for ChromeDriver.
 
 If your test is testing a specific part of the WebDriver API, consider adding
 it to the appropriate place in the WebDriver tree instead.
@@ skipping 98 matching lines @@
       self.assertEquals(404, expected.code)
 
   def testCanStartChromeDriverOnSpecificPort(self):
     launcher = ChromeDriverLauncher(port=9520)
     self.assertEquals(9520, launcher.GetPort())
     driver = WebDriver(launcher.GetURL(), 'chrome', 'any')
     driver.quit()
     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

[kkania, 2011/03/07 18:15:17]

there are a lot of errors in this script; please fix

[Joe, 2011/03/11 22:12:48]

refreshed
On 2011/03/07 18:15:17, kkania wrote:

+    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"""
 
   SEARCH = "http://www.google.com/webhp?hl=en"
 
   def setUp(self):
     self._launcher = ChromeDriverLauncher()
     self._driver = WebDriver(self._launcher.GetURL(), 'chrome', 'any')
 
   def tearDown(self):
@@ skipping 146 matching lines @@
     self.assertEquals('', url_parts[0])
     self.assertEquals('wd', url_parts[1])
     self.assertEquals('hub', url_parts[2])
     self.assertEquals('session', url_parts[3])
     self.assertEquals(data['sessionId'], url_parts[4])
 
 
 if __name__ == '__main__':
   unittest.main(module='chromedriver_tests',
                 testRunner=GTestTextTestRunner(verbosity=1))