def test_delete_cookie(): response = Response() response.set_cookie('name', 'value') response.delete_cookie('name') cookies = [value for name, value in response.headerlist if name.title() == 'Set-Cookie'] assert 'name=;' in cookies[0] or 'name="";' in cookies[0]
def test_set_cookie_expires(): r = Response() r.set_cookie('name1', 'value', expires=42) r.set_cookie('name2', 'value', expires=datetime.datetime(1970,1,1,0,0,43)) cookies = sorted([value for name, value in r.headerlist if name.title() == 'Set-Cookie']) assert cookies[0] == 'name1=value; expires=Thu, 01 Jan 1970 00:00:42 GMT' assert cookies[1] == 'name2=value; expires=Thu, 01 Jan 1970 00:00:43 GMT'
def test_set_cookie_httponly(): r = Response() r.set_cookie('name1', 'value', httponly=True) r.set_cookie('name2', 'value', httponly=False) cookies = sorted([value for name, value in r.headerlist if name.title() == 'Set-Cookie']) assert cookies[0].lower() == 'name1=value; httponly' assert cookies[1] == 'name2=value'
def test_set_cookie_maxage(): import datetime r = Response() r.set_cookie('name1', 'value', max_age=5) r.set_cookie('name2', 'value', max_age=datetime.timedelta(days=1)) cookies = sorted([value for name, value in r.headerlist if name.title() == 'Set-Cookie']) assert cookies[0] == 'name1=value; Max-Age=5' assert cookies[1] == 'name2=value; Max-Age=86400'
def test_set_cookie(): r = Response() r.set_cookie('name1', 'value', max_age=5) r.set_cookie('name2', 'value 2', path='/foo') r.set_cookie('name4', 'value4', secret=True) with pytest.raises(TypeError): r.set_cookie('name3', 3) cookies = [value for name, value in r.headerlist if name.title() == 'Set-Cookie'] cookies.sort() assert cookies[0] == 'name1=value; Max-Age=5' assert cookies[1] == 'name2="value 2"; Path=/foo'