Commit 46541384 authored by Roque's avatar Roque Committed by Klaus Wölfel

testbrowser: cookie handling for logged-in user

parent d10fae80
...@@ -33,6 +33,7 @@ ...@@ -33,6 +33,7 @@
import logging import logging
import sys import sys
import urllib import urllib
import Cookie
from urlparse import urljoin from urlparse import urljoin
from z3c.etestbrowser.browser import ExtendedTestBrowser from z3c.etestbrowser.browser import ExtendedTestBrowser
...@@ -782,10 +783,25 @@ class MainForm(Form): ...@@ -782,10 +783,25 @@ class MainForm(Form):
(self.browser._username, self.browser._password)) (self.browser._username, self.browser._password))
def login(form): def login(form):
print "Login with user-pass: " + self.browser._username + " - " + self.browser._password
form.getControl(name='__ac_name').value = self.browser._username form.getControl(name='__ac_name').value = self.browser._username
form.getControl(name='__ac_password').value = self.browser._password form.getControl(name='__ac_password').value = self.browser._password
form.submit() form.submit()
def setCookies():
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:
for part in headers_cookie.split(","):
if '__ac=' in part:
for subpart in part.split(";"):
if '__ac=' in subpart:
ac_value = subpart.split("=")[1].replace('"','')
self.browser.cookies["__ac"] = ac_value
try: try:
login(self) login(self)
except LookupError: except LookupError:
...@@ -797,12 +813,7 @@ class MainForm(Form): ...@@ -797,12 +813,7 @@ class MainForm(Form):
(self.browser._erp5_base_url, (self.browser._erp5_base_url,
self.browser._username, self.browser._username,
self.browser._password)) self.browser._password))
setCookies()
import Cookie
cookie = Cookie.SimpleCookie()
cookie.load(self.browser.headers['set-cookie'])
ac_value = cookie['__ac'].value
self.browser.cookies["__ac"] = ac_value
def submitSelectFavourite(self, label=None, value=None, **kw): 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