oci/zeronet
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make current url getting compatible with chromedriver

Browse source
This commit is contained in:
shortcutme 2018-11-26 00:11:58 +01:00
parent d588327c1c
commit 131926cad8
No known key found for this signature in database
GPG key ID: 5B63BAE6CB9613AE
1 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/Test/TestWeb.py
View file

@ -21,6 +21,10 @@ class WaitForPageLoad(object):
WebDriverWait(self.browser, 10).until(staleness_of(self.old_page)) WebDriverWait(self.browser, 10).until(staleness_of(self.old_page))
def getContextUrl(browser):
return browser.execute_script("return window.location.toString()")
def getUrl(url): def getUrl(url):
content = urllib.urlopen(url).read() content = urllib.urlopen(url).read()
assert "server error" not in content.lower(), "Got a server error! " + repr(url) assert "server error" not in content.lower(), "Got a server error! " + repr(url)
@ -56,18 +60,18 @@ class TestWeb:
def testLinkSecurity(self, browser, site_url): def testLinkSecurity(self, browser, site_url):
browser.get("%s/1EU1tbG9oC1A8jz2ouVwGZyQ5asrNsE4Vr/test/security.html" % site_url) browser.get("%s/1EU1tbG9oC1A8jz2ouVwGZyQ5asrNsE4Vr/test/security.html" % site_url)
WebDriverWait(browser, 10).until(title_is("ZeroHello - ZeroNet")) WebDriverWait(browser, 10).until(title_is("ZeroHello - ZeroNet"))
assert browser.current_url == "%s/1EU1tbG9oC1A8jz2ouVwGZyQ5asrNsE4Vr/test/security.html" % site_url assert getContextUrl(browser) == "%s/1EU1tbG9oC1A8jz2ouVwGZyQ5asrNsE4Vr/test/security.html" % site_url
# Switch to inner frame # Switch to inner frame
browser.switch_to.frame(browser.find_element_by_id("inner-iframe")) browser.switch_to.frame(browser.find_element_by_id("inner-iframe"))
assert "wrapper_nonce" in browser.current_url assert "wrapper_nonce" in getContextUrl(browser)
browser.switch_to.default_content() browser.switch_to.default_content()
# Clicking on links without target # Clicking on links without target
browser.switch_to.frame(browser.find_element_by_id("inner-iframe")) browser.switch_to.frame(browser.find_element_by_id("inner-iframe"))
with WaitForPageLoad(browser): with WaitForPageLoad(browser):
browser.find_element_by_id("link_to_current").click() browser.find_element_by_id("link_to_current").click()
assert "wrapper_nonce" not in browser.current_url # The browser object back to default content assert "wrapper_nonce" not in getContextUrl(browser) # The browser object back to default content
assert "Forbidden" not in browser.page_source assert "Forbidden" not in browser.page_source
# Check if we have frame inside frame # Check if we have frame inside frame
browser.switch_to.frame(browser.find_element_by_id("inner-iframe")) browser.switch_to.frame(browser.find_element_by_id("inner-iframe"))
@ -79,15 +83,15 @@ class TestWeb:
browser.switch_to.frame(browser.find_element_by_id("inner-iframe")) browser.switch_to.frame(browser.find_element_by_id("inner-iframe"))
with WaitForPageLoad(browser): with WaitForPageLoad(browser):
browser.find_element_by_id("link_to_top").click() browser.find_element_by_id("link_to_top").click()
assert "wrapper_nonce" not in browser.current_url # The browser object back to default content assert "wrapper_nonce" not in getContextUrl(browser) # The browser object back to default content
assert "Forbidden" not in browser.page_source assert "Forbidden" not in browser.page_source
browser.switch_to.default_content() browser.switch_to.default_content()
# Try to escape from inner_frame # Try to escape from inner_frame
browser.switch_to.frame(browser.find_element_by_id("inner-iframe")) browser.switch_to.frame(browser.find_element_by_id("inner-iframe"))
assert "wrapper_nonce" in browser.current_url # Make sure we are inside of the inner-iframe assert "wrapper_nonce" in getContextUrl(browser) # Make sure we are inside of the inner-iframe
with WaitForPageLoad(browser): with WaitForPageLoad(browser):
browser.execute_script("window.top.location = window.location") browser.execute_script("window.top.location = window.location")
assert "wrapper_nonce" in browser.current_url # We try to use nonce-ed html without iframe assert "wrapper_nonce" in getContextUrl(browser) # We try to use nonce-ed html without iframe
assert "<iframe" in browser.page_source # Only allow to use nonce once-time assert "<iframe" in browser.page_source # Only allow to use nonce once-time
browser.switch_to.default_content() browser.switch_to.default_content()