def test_cookies_with_domain(client, app):
response = Response(HTTPStatus.OK, body="body")
response.cookies.set('name', 'value', domain='www.example.com')
assert (
b'\r\nSet-Cookie: name=value; Domain=www.example.com; '
b'Path=/\r\n'
) in bytes(response)
def test_cookies_with_multiple_attributes(client, app):
response = Response(HTTPStatus.OK, body="body")
response.cookies.set('name', 'value', secure=True, max_age=300)
assert (
b'\r\nSet-Cookie: name=value; Max-Age=300; Path=/; '
b'Secure\r\n'
) in bytes(response)
async def full(request, one, two):
response = Response.json({
'parameters': f'{one} and {two}',
'query': request.query.get('query'),
'cookie': request.cookies['test'],
})
response.cookies.set(name='bench', value='value')
return response
def test_cookies_with_expires():
response = Response(HTTPStatus.OK, body="body")
response.cookies.set(
'name', 'value',
expires=datetime(2027, 9, 21, 11, 22)
)
assert (
b'\r\nSet-Cookie: name=value; '
b'Expires=Tue, 21 Sep 2027 11:22:00 GMT; Path=/\r\n'
) in bytes(response)
def test_multiple_cookies():
response = Response(HTTPStatus.OK, body="body")
response.cookies.set('name', 'value')
response.cookies.set('other', 'value2')
assert b'\r\nSet-Cookie: name=value; Path=/\r\n' in bytes(response)
assert b'\r\nSet-Cookie: other=value2; Path=/\r\n' in bytes(response)
async def post(request):
return Response.raw(b'POST')
async def get(request):
return Response.raw(b'GET')
async def feed(request):
await request.parse_body()
files = list(request.files.keys())
return Response.raw("You got here and it's all read: {}".format(files))
def test_cookies_with_http_only(client, app):
response = Response(HTTPStatus.OK, body="body")
response.cookies.set('name', 'value', httponly=True)
assert (
b'\r\nSet-Cookie: name=value; Path=/; HttpOnly\r\n'
) in bytes(response)
async def query(request):
response = Response.json({'query': request.query.get('query')})
return response
async def cookie(request):
response = Response.json({'cookie': request.cookies['test']})
response.cookies.set(name='bench', value='value')
return response
async def parameter(request, parameter):
response.json = Response.json({'parameter': parameter})
async def serve_websocket(request):
return Response.streamer(
file_iterator('websocket.html'),
content_type="text/html")
async def ignore(request):
# A post where we ignore the body
return Response.raw("You got here and all was ignored.")
async def minimal(request):
response = Response.json({'message': 'Hello, World!'})
return response
def test_cookies_with_max_age(client, app):
response = Response(HTTPStatus.OK, body="body")
response.cookies.set('name', 'value', max_age=600)
assert (
b'\r\nSet-Cookie: name=value; Max-Age=600; Path=/\r\n'
) in bytes(response)
async def bodyless(request):
return Response(status=HTTPStatus.ACCEPTED)
def test_cookies():
response = Response(HTTPStatus.OK, body="body")
response.cookies.set(name='name', value='value')
assert b'\r\nSet-Cookie: name=value; Path=/\r\n' in bytes(response)
async def hello(request):
return Response.raw(b'Hello World !')
def test_cookies_with_secure(client, app):
response = Response(HTTPStatus.OK, body="body")
response.cookies.set('name', 'value', secure=True)
assert (
b'\r\nSet-Cookie: name=value; Path=/; Secure\r\n'
) in bytes(response)
async def post(request):
body = await request.raw_body
return Response.raw(body)
def test_delete_cookies(client, app):
response = Response(HTTPStatus.OK, body="body")
response.cookies.set('name', 'value')
del response.cookies['name']
assert b'Set-Cookie:' not in bytes(response)
async def raw(request):
return Response.raw(b'You got here')