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_append_header(): response = Response() response.set_header('x-test', 'foo') headers = [value for name, value in response.headerlist if name.title() == 'X-Test'] assert ['foo'] == headers assert 'foo' == response['x-test'] response.add_header('X-Test', 'bar') headers = [value for name, value in response.headerlist if name.title() == 'X-Test'] assert ['foo', 'bar'] == headers assert 'foo' == response['x-test']
def test_expires_header(): import datetime response = Response() now = datetime.datetime.now() response.expires = now def seconds(a, b): td = max(a,b) - min(a,b) return td.days*360*24 + td.seconds assert 0 == seconds(response.expires, now) now2 = datetime.datetime.utcfromtimestamp( parse_date(response.headers['Expires'])) assert 0 == seconds(now, now2)
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'
def test_expires_header(): import datetime response = Response() now = datetime.datetime.now() response.expires = now def seconds(a, b): td = max(a, b) - min(a, b) return td.days * 360 * 24 + td.seconds assert 0 == seconds(response.expires, now) now2 = datetime.datetime.utcfromtimestamp( parse_date(response.headers['Expires'])) assert 0 == seconds(now, now2)
def test_delete_header(): response = Response() response['x-test'] = 'foo' assert 'foo', response['x-test'] del response['X-tESt'] with pytest.raises(KeyError): response['x-test']
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'
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_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_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_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_header(): response = Response() response['x-test'] = 'foo' headers = [ value for name, value in response.headerlist if name.title() == 'X-Test' ] assert ['foo'] == headers assert 'foo' == response['x-test'] response['X-Test'] = 'bar' headers = [ value for name, value in response.headerlist if name.title() == 'X-Test' ] assert ['bar'] == headers assert 'bar' == response['x-test']
def test_append_header(): response = Response() response.set_header('x-test', 'foo') headers = [ value for name, value in response.headerlist if name.title() == 'X-Test' ] assert ['foo'] == headers assert 'foo' == response['x-test'] response.add_header('X-Test', 'bar') headers = [ value for name, value in response.headerlist if name.title() == 'X-Test' ] assert ['foo', 'bar'] == headers assert 'foo' == response['x-test']
def test_non_string_header(): response = Response() response['x-test'] = 5 assert '5' == response['x-test'] response['x-test'] = None assert 'None' == response['x-test']