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() == ""