jbaskin/morethantext-web
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Altered page to retain cookies.

Browse Source
This commit is contained in:
Jeff Baskin 2022-06-23 14:09:03 -04:00
parent c8680b33d4
commit 4f29abc818
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/step_defs/page.py
View File

@ -1,5 +1,6 @@
"""Fisture class for page functions.""" """Fisture class for page functions."""
from http.cookiejar import CookieJar
from urllib import request from urllib import request
from urllib.error import HTTPError from urllib.error import HTTPError
@ -10,11 +11,14 @@ class Page:
def __init__(self): def __init__(self):
"""Init function.""" """Init function."""
self.res = None self.res = None
self.cjar = CookieJar()
self.opener = request.build_opener(request.HTTPCookieProcessor(self.cjar))
def request_url(self, url): def request_url(self, url):
"""Make a page request.""" """Make a page request."""
try: try:
page = request.urlopen(url) # page = request.urlopen(url)
page = self.opener.open(url)
except HTTPError as err: except HTTPError as err:
self.res = err self.res = err
else: else:
@ -25,6 +29,6 @@ class Page:
"""Return the status code.""" """Return the status code."""
return self.res.status return self.res.status
def get_header(self, header): def get_cookies(self):
"""Did a session id get set.""" """Did a session id get set."""
return self.res.headers[header] return self.cjar

2
tests/step_defs/test_session.py
View File

@ -22,4 +22,4 @@ def access_home_page(server, page):
@then("there is a session id") @then("there is a session id")
def confirm_session(page): def confirm_session(page):
"""Confirm session id exists.""" """Confirm session id exists."""
assert page.get_header("set-cookie") assert len(page.get_cookies()), "No cookies received."