Commit cba90e22 authored by Roque's avatar Roque

testbrowser: cookie handling for logged-in user

Small recent change to handle cookies in connections to frontend urls.
* The refactoring of old testbrowser version was already reviewed and merge to master: 5a0b3763 and 388a508b.


/reviewed-on !587
parents 0ffa37d7 eed1e4f1
......@@ -33,6 +33,7 @@
import logging
import sys
import urllib
import Cookie
from urlparse import urljoin
from z3c.etestbrowser.browser import ExtendedTestBrowser
......@@ -786,6 +787,23 @@ class MainForm(Form):
form.getControl(name='__ac_password').value = self.browser._password
form.submit()
def setCookies():
"""
Gets the __ac value from the browser headers and sets it as cookie in order to keep
the user logged during the tests. If SimpleCookie can't parse the values,
regular expressions are used.
"""
headers_cookie = self.browser.headers['set-cookie']
cookie = Cookie.SimpleCookie()
cookie.load(headers_cookie)
if '__ac' in cookie.keys():
ac_value = cookie['__ac'].value
else:
reg = '__ac=\"([A-Za-z0-9%]+)*\"'
match = re.search(reg, headers_cookie)
ac_value = match.group(1) if match else None
self.browser.cookies["__ac"] = ac_value
try:
login(self)
except LookupError:
......@@ -797,12 +815,7 @@ class MainForm(Form):
(self.browser._erp5_base_url,
self.browser._username,
self.browser._password))
import Cookie
cookie = Cookie.SimpleCookie()
cookie.load(self.browser.headers['set-cookie'])
ac_value = cookie['__ac'].value
self.browser.cookies["__ac"] = ac_value
setCookies()
def submitSelectFavourite(self, label=None, value=None, **kw):
"""
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment