def test_id_gets_added_to_cookie(self):
response = self.app.get(self.product.get_absolute_url())
request = HttpRequest()
request.COOKIES[COOKIE_NAME] = _unquote(
response.test_app.cookies[COOKIE_NAME])
self.assertTrue(
self.product.id in CustomerHistoryManager.extract(request))
def cookies(self) -> Dict[str, str]:
"""
Returns cookies in as a `dict`.
NOTE: Modifications to this dictionary will not affect the
response value. In fact, this value should not be modified.
"""
cookies: Dict[str, str] = {}
cookie_header = self.headers.get("cookie", "")
# This function has been adapted from Django 3.1.0.
# Note: we are explicitly _NOT_ using `SimpleCookie.load` because it is based
# on an outdated spec and will fail on lots of input we want to support
for chunk in cookie_header.split(";"):
if not chunk:
continue
if "=" in chunk:
key, val = chunk.split("=", 1)
else:
# Assume an empty name per
# https://bugzilla.mozilla.org/show_bug.cgi?id=169091
key, val = "", chunk
key, val = key.strip(), val.strip()
if key or val:
# unquote using Python's algorithm.
cookies[key] = http_cookies._unquote(val) # type: ignore
return cookies
def test_price_is_recorded(self):
oscar_open_basket_cookie = _unquote(self.response.test_app.cookies['oscar_open_basket'])
basket_id = oscar_open_basket_cookie.split(':')[0]
basket = Basket.objects.get(id=basket_id)
line = basket.lines.get(product=self.product)
stockrecord = self.product.stockrecords.all()[0]
self.assertEqual(stockrecord.price, line.price_excl_tax)
def cookie_parser(cookie_string: str) -> typing.Dict[str, str]:
"""
This function parses a ``Cookie`` HTTP header into a dict of key/value pairs.
It attempts to mimic browser cookie parsing behavior: browsers and web servers
frequently disregard the spec (RFC 6265) when setting and reading cookies,
so we attempt to suit the common scenarios here.
This function has been adapted from Django 3.1.0.
Note: we are explicitly _NOT_ using `SimpleCookie.load` because it is based
on an outdated spec and will fail on lots of input we want to support
"""
cookie_dict: typing.Dict[str, str] = {}
for chunk in cookie_string.split(";"):
if "=" in chunk:
key, val = chunk.split("=", 1)
else:
# Assume an empty name per
# https://bugzilla.mozilla.org/show_bug.cgi?id=169091
key, val = "", chunk
key, val = key.strip(), val.strip()
if key or val:
# unquote using Python's algorithm.
cookie_dict[key] = http_cookies._unquote(val) # type: ignore
return cookie_dict
def test_price_is_recorded(self):
oscar_open_basket_cookie = _unquote(self.response.test_app.cookies['oscar_open_basket'])
basket_id = oscar_open_basket_cookie.split(':')[0]
basket = Basket.objects.get(id=basket_id)
line = basket.lines.get(product=self.product)
stockrecord = self.product.stockrecords.all()[0]
self.assertEqual(stockrecord.price_excl_tax, line.price_excl_tax)
def parse_cookie_header(header_value):
"""Parse a Cookie header value into a dict of named values.
(See also: RFC 6265, Section 5.4)
Args:
header_value (str): Value of a Cookie header
Returns:
dict: Map of cookie names to a list of all cookie values found in the
header for that name. If a cookie is specified more than once in the
header, the order of the values will be preserved.
"""
# See also:
#
# https://tools.ietf.org/html/rfc6265#section-5.4
# https://tools.ietf.org/html/rfc6265#section-4.1.1
#
cookies = {}
for token in header_value.split(';'):
name, __, value = token.partition('=')
# NOTE(kgriffs): RFC6265 is more strict about whitespace, but we
# are more lenient here to better handle old user agents and to
# mirror Python's standard library cookie parsing behavior
name = name.strip()
value = value.strip()
# NOTE(kgriffs): Skip malformed cookie-pair
if not name:
continue
# NOTE(kgriffs): Skip cookies with invalid names
if _COOKIE_NAME_RESERVED_CHARS.search(name):
continue
# NOTE(kgriffs): To maximize compatibility, we mimic the support in the
# standard library for escaped characters within a double-quoted
# cookie value according to the obsolete RFC 2109. However, we do not
# expect to see this encoding used much in practice, since Base64 is
# the current de-facto standard, as recommended by RFC 6265.
#
# PERF(kgriffs): These checks have been hoisted from within _unquote()
# to avoid the extra function call in the majority of the cases when it
# is not needed.
if len(value) > 2 and value[0] == '"' and value[-1] == '"':
value = http_cookies._unquote(value)
# PERF(kgriffs): This is slightly more performant as
# compared to using dict.setdefault()
if name in cookies:
cookies[name].append(value)
else:
cookies[name] = [value]
return cookies
def parse_cookie_header(header_value):
"""Parse a Cookie header value into a dict of named values.
(See also: RFC 6265, Section 5.4)
Args:
header_value (str): Value of a Cookie header
Returns:
dict: Map of cookie names to a list of all cookie values found in the
header for that name. If a cookie is specified more than once in the
header, the order of the values will be preserved.
"""
# See also:
#
# https://tools.ietf.org/html/rfc6265#section-5.4
# https://tools.ietf.org/html/rfc6265#section-4.1.1
#
cookies = {}
for token in header_value.split(';'):
name, __, value = token.partition('=')
# NOTE(kgriffs): RFC6265 is more strict about whitespace, but we
# are more lenient here to better handle old user agents and to
# mirror Python's standard library cookie parsing behavior
name = name.strip()
value = value.strip()
# NOTE(kgriffs): Skip malformed cookie-pair
if not name:
continue
# NOTE(kgriffs): Skip cookies with invalid names
if _COOKIE_NAME_RESERVED_CHARS.search(name):
continue
# NOTE(kgriffs): To maximize compatibility, we mimic the support in the
# standard library for escaped characters within a double-quoted
# cookie value according to the obsolete RFC 2109. However, we do not
# expect to see this encoding used much in practice, since Base64 is
# the current de-facto standard, as recommended by RFC 6265.
#
# PERF(kgriffs): These checks have been hoisted from within _unquote()
# to avoid the extra function call in the majority of the cases when it
# is not needed.
if len(value) > 2 and value[0] == '"' and value[-1] == '"':
value = http_cookies._unquote(value)
# PERF(kgriffs): This is slightly more performant as
# compared to using dict.setdefault()
if name in cookies:
cookies[name].append(value)
else:
cookies[name] = [value]
return cookies
def parse_cookie(cookie):
"""
Return a dictionary parsed from a `Cookie:` header string.
"""
cookiedict = {}
for chunk in cookie.split(';'):
if '=' in chunk:
key, val = chunk.split('=', 1)
else:
# Assume an empty name per
# https://bugzilla.mozilla.org/show_bug.cgi?id=169091
key, val = '', chunk
key, val = key.strip(), val.strip()
if key or val:
# unquote using Python's algorithm.
cookiedict[key] = cookies._unquote(val)
return cookiedict
def parse_cookie(cookie):
"""
Return a dictionary parsed from a `Cookie:` header string.
"""
cookiedict = {}
for chunk in cookie.split(';'):
if '=' in chunk:
key, val = chunk.split('=', 1)
else:
# Assume an empty name per
# https://bugzilla.mozilla.org/show_bug.cgi?id=169091
key, val = '', chunk
key, val = key.strip(), val.strip()
if key or val:
# unquote using Python's algorithm.
cookiedict[key] = cookies._unquote(val)
return cookiedict
def unquote_cookie(cookie_value):
"""
Make sure a cookie value is unescaped from double quotes
"""
return _unquote(cookie_value)
def test_id_gets_added_to_cookie(self):
response = self.app.get(self.product.get_absolute_url())
request = HttpRequest()
request.COOKIES[COOKIE_NAME] = _unquote(response.test_app.cookies[COOKIE_NAME])
self.assertTrue(self.product.id in history.extract(request))
def unquote_cookie(cookie_value):
"""
Make sure a cookie value is unescaped from double quotes
"""
return _unquote(cookie_value)
