Commit 64a4ec49 authored by Roque's avatar Roque

testbrowser: cookie handling for logged-in user

parent b7efa170
...@@ -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):
""" """
......
  • You might use regular expression and add docstring to better explain what you do

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