Bug 818287 - Support screenshots without a data URL prefix. r=dhunt

mozilla-b2g · Nov 1, 2013 · d654c10 · d654c10
1 parent 44ed9e1
commit d654c10
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/python/gaia-ui-tests/gaiatest/runtests.py b/tests/python/gaia-ui-tests/gaiatest/runtests.py
@@ -96,7 +96,11 @@ def gather_debug(self):
         try:
             # TODO make screenshot consistant size by using full viewport
             # Bug 883294 - Add ability to take full viewport screenshots
-            debug['screenshot'] = self.marionette.screenshot()[22:]
+            # TODO Bug 838287 - remove condition after marionette release.
+            screenshot = self.marionette.screenshot()
+            if screenshot.startswith('data:'):
+                screenshot = screenshot[22:]
+            debug['screenshot'] = screenshot
             debug['source'] = self.marionette.page_source
             self.marionette.switch_to_frame()
             debug['settings'] = json.dumps(self.marionette.execute_async_script("""