# This Source Code Form is subject to the terms of the Mozilla Public # License, v. 2.0. If a copy of the MPL was not distributed with this # file, You can obtain one at http://mozilla.org/MPL/2.0/. from urllib.parse import quote from marionette_harness import MarionetteTestCase def inline(doc, mime=None, charset=None): mime = "html" if mime is None else mime charset = "utf-8" if (charset is None) else charset return "data:text/{};charset={},{}".format(mime, charset, quote(doc)) class TestPageSource(MarionetteTestCase): def testShouldReturnTheSourceOfAPage(self): test_html = inline("

Check the PageSource") self.marionette.navigate(test_html) source = self.marionette.page_source from_web_api = self.marionette.execute_script( "return document.documentElement.outerHTML" ) self.assertTrue("' ) self.marionette.navigate(test_html) # if we don't throw on the next line we are good! source = self.marionette.page_source from_web_api = self.marionette.execute_script( "return document.documentElement.outerHTML" ) self.assertEqual(source, from_web_api) def testShouldReturnAXMLDocumentSource(self): test_xml = inline("baz", "xml") self.marionette.navigate(test_xml) source = self.marionette.page_source from_web_api = self.marionette.execute_script( "return document.documentElement.outerHTML" ) import re self.assertEqual( re.sub(r"\s", "", source), "baz" ) self.assertEqual(source, from_web_api)