This CL adds client.toDataURL which gets the contents...

tests/selenium/selenium_utilities.py

 #!/usr/bin/python2.4
 # Copyright 2009, Google Inc.
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions are
 # met:
 #
 #     * Redistributions of source code must retain the above copyright
 # notice, this list of conditions and the following disclaimer.
@@ skipping 21 matching lines @@
 """Utility scripts for selenium.
 
 A collection of utility scripts for selenium test cases to use.
 """
 
 
 import os
 import re
 import time
 import unittest
+import base64
 import gflags
 import selenium_constants
 
 
 FLAGS = gflags.FLAGS
 SUFFIXES = ["small", "medium", "large"]
 
 
 def IsValidTestType(test_type):
   """Returns True if test_type is a "small", "medium" or "large"."""
@@ skipping 115 matching lines @@
                        (selenium_constants.RESIZE_WIDTH,
                         selenium_constants.RESIZE_HEIGHT))
 
   # Execute screenshot capture code
 
   # Replace all backslashes with forward slashes so it is parsed correctly
   # by Javascript
   full_path = filename.replace("\\", "/")
 
   # Attempt to take a screenshot of the display buffer
-  eval_string = ("%s.saveScreen('%s')" % (client, full_path))
+  eval_string = ("%s.toDataURL()" % client)
+
 
   # Set Post render call back to take screenshot
   script = ["window.g_selenium_post_render = false;",
             "window.g_selenium_save_screen_result = false;",
             "var frameCount = 0;",
             "%s.setPostRenderCallback(function() {" % client,
             "  ++frameCount;",
             "  if (frameCount >= 3) {",
             "    %s.clearPostRenderCallback();" % client,
             "    window.g_selenium_save_screen_result = %s;" % eval_string,
             "    window.g_selenium_post_render = true;",
             "  } else {",
             "    %s.render()" % client,
             "  }",
             "})",
             "%s.render()" % client]
   session.run_script("\n".join(script))
   # Wait for screenshot to be taken.
   session.wait_for_condition("window.g_selenium_post_render", 20000)
 
   # Get result
-  success = session.get_eval("window.g_selenium_save_screen_result")
-
-  if success == u"true":
+  data_url = session.get_eval("window.g_selenium_save_screen_result")
+  expected_header = "data:image/png;base64,"
+  if data_url.startswith(expected_header):
+    png = base64.b64decode(data_url[len(expected_header):])
+    file = open(full_path + ".png", 'wb')
+    file.write(png)
+    file.close()
     print "Saved screenshot %s." % full_path
     return True
 
   return False
 
 
 class SeleniumTestCase(unittest.TestCase):
   """Wrapper for TestCase for selenium."""
 
   def __init__(self, name, session, browser, test_type=None, sample_path=None,
@@ skipping 83 matching lines @@
       # if they are animated we need to stop the animation and set the clock
       # to some time so we get a known state.
       self.session.run_script("g_timeMult = 0")
       self.session.run_script("g_clock = " + clock)
 
       # take a screenshot.
       screenshot = self.sample_path.replace("/", "_") + str(screenshot_id)
       self.assertTrue(TakeScreenShot(self.session, self.browser,
                                      client, screenshot))
       screenshot_id += 1