from tests.support.classic.asserts import assert_error, assert_success def get_page_source(session): return session.transport.send( "GET", "session/{session_id}/source".format(**vars(session))) def test_no_top_browsing_context(session, closed_window): response = get_page_source(session) assert_error(response, "no such window") def test_no_browsing_context(session, closed_frame): response = get_page_source(session) assert_error(response, "no such window") def test_source_matches_outer_html(session, inline): session.url = inline("CheesePeas") expected = session.execute_script("return document.documentElement.outerHTML") response = get_page_source(session) assert_success(response, expected)