Commit 9dbcf0ba by Will Daly

Updated login session handling to fix issue with ChromeDriver on Ubuntu

parent 36fb8f91
...@@ -42,6 +42,8 @@ def log_in(username, password): ...@@ -42,6 +42,8 @@ def log_in(username, password):
Log the user in programatically Log the user in programatically
''' '''
world.browser.visit(django_url('/'))
# Authenticate the user # Authenticate the user
user = authenticate(username=username, password=password) user = authenticate(username=username, password=password)
assert(user is not None and user.is_active) assert(user is not None and user.is_active)
...@@ -60,15 +62,8 @@ def log_in(username, password): ...@@ -60,15 +62,8 @@ def log_in(username, password):
# Retrieve the sessionid and add it to the browser's cookies # Retrieve the sessionid and add it to the browser's cookies
cookie_dict = {settings.SESSION_COOKIE_NAME: request.session.session_key} cookie_dict = {settings.SESSION_COOKIE_NAME: request.session.session_key}
try: world.browser.cookies.delete()
world.browser.cookies.add(cookie_dict) world.browser.cookies.add(cookie_dict)
# WebDriver has an issue where we cannot set cookies
# before we make a GET request, so if we get an error,
# we load the '/' page and try again
except:
world.browser.visit(django_url('/'))
world.browser.cookies.add(cookie_dict)
@world.absorb @world.absorb
......
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