def test_create_cookie_header_dict_cookie():
""" Test header creation for cookies when cookie contents in dictionary """
cookie_dict = {"JSESSIONID": "ABC123DEF456",
"SPECIALTOKEN": "TOPS.E.KRATZ"}
msg_dict_cookie = HttpRequestMessage("GET", "/", 'localhost', {}, cookies=cookie_dict)
cookie_header = msg_dict_cookie.create_cookie_header()
assert cookie_header == "SPECIALTOKEN=TOPS.E.KRATZ; JSESSIONID=ABC123DEF456;"
def test_create_cookie_header_mixed_cookies():
""" Test header createion for cookies when cookie contents
are both dict and obj
"""
cookie_one = Cookie("CSRF_TOKEN", "1234zyxw", "mydomain.com", "/")
cookie_two = None
cookie_dict = {"JSESSIONID": "ABC123DEF456",
"cookie_obj123": cookie_one,
"cookie_obj2": cookie_two,
"Tracker": "trackingnum_1"}
msg_mixed_cookie = HttpRequestMessage("GET", "/", 'localhost', {}, cookies=cookie_dict)
cookie_str = ("Tracker=trackingnum_1; "
"CSRF_TOKEN=1234zyxw; "
"JSESSIONID=ABC123DEF456;")
assert msg_mixed_cookie.create_cookie_header() == cookie_str
def test_create_cookie_header_obj_cookie():
""" Test header creation for cookies when cookie contents in obj form """
cookie_one = Cookie("CSRF_TOKEN", "1234zyxw", "mydomain.com", "/")
msg_obj_cookie = HttpRequestMessage("GET", "/", 'localhost', {}, cookies=cookie_one)
assert msg_obj_cookie.create_cookie_header() == "CSRF_TOKEN=1234zyxw;"
def test_create_cookie_header_none_cookie():
""" Test header creation for cookies when cookie is declared to be none """
msg_none_cookie = HttpRequestMessage("GET", "/", 'localhost', {}, cookies=None)
assert msg_none_cookie.cookies == {}
assert msg_none_cookie.create_cookie_header() == ""
def test_create_cookie_header_no_cookie():
""" Test header creation for cookies from no supplied cookie """
msg_no_cookie = HttpRequestMessage("GET", "/", 'localhost', {})
assert msg_no_cookie.create_cookie_header() == ""
