async def test_exception(self):
async def handler(request):
raise ValueError("Oops")
async with http.serve(handler, HOST, 12345):
request = http.HTTPRequest()
request.headers["Host"] = "%s:12345" % HOST
response = await http.request(request)
assert response.status_code == 500
assert response.status_name == "Internal Server Error"
async def test_json(self):
async def handler(request):
response = http.HTTPResponse(200)
response.json = {"result": request.json["value"]}
return response
async with http.serve(handler, HOST, 12345):
request = http.HTTPRequest()
request.headers["Host"] = "%s:12345" % HOST
request.json = {"value": True}
response = await http.request(request)
assert response.json["result"] is True
async def test_form(self):
async def handler(request):
response = http.HTTPResponse(200)
response.form = {"$<result>": request.form["&value"]}
return response
async with http.serve(handler, HOST, 12345):
request = http.HTTPRequest()
request.headers["Host"] = "%s:12345" % HOST
request.form = {"&value": "&=?"}
response = await http.request(request)
assert response.form["$<result>"] == "&=?"
async def test_plainform(self):
async def handler(request):
assert request.text == "+value +=!=?"
response = http.HTTPResponse(200)
response.plainform = {"$<result>": request.plainform["+value +"]}
return response
async with http.serve(handler, HOST, 12345):
request = http.HTTPRequest()
request.headers["Host"] = "%s:12345" % HOST
request.plainform = {"+value +": "!=?"}
response = await http.request(request)
assert response.plainform["$<result>"] == "!=?"
async def test_continue(self):
async def handler(request):
response = http.HTTPResponse(200)
response.body = request.body
return response
async with http.serve(handler, HOST, 12345):
request = http.HTTPRequest()
request.headers["Host"] = "%s:12345" % HOST
request.continue_threshold = 64
request.body = b"a" * 80
response = await http.request(request)
assert response.body == b"a" * 80
async def test_text(self):
async def handler(request):
assert request.body == b"Hello"
response = http.HTTPResponse(200)
response.text = request.text.upper()
return response
async with http.serve(handler, HOST, 12345):
request = http.HTTPRequest()
request.headers["Host"] = "%s:12345" % HOST
request.text = "Hello"
response = await http.request(request)
assert response.text == "HELLO"
async def test_body(self):
async def handler(request):
assert request.body == b"abcdef"
response = http.HTTPResponse(200)
response.headers["X-Content-Size"] = len(request.body)
return response
async with http.serve(handler, HOST, 12345):
request = http.HTTPRequest()
request.headers["Host"] = "%s:12345" % HOST
request.body = b"abcdef"
response = await http.request(request)
assert response.headers["X-Content-Size"] == "6"
async def test_xml(self):
async def handler(request):
assert request.xml.name == "value"
tree = xml.XMLTree("result")
tree.text = request.xml.text
response = http.HTTPResponse(200)
response.xml = tree
return response
async with http.serve(handler, HOST, 12345):
tree = xml.XMLTree("value")
tree.text = "12345"
request = http.HTTPRequest()
request.headers["Host"] = "%s:12345" % HOST
request.xml = tree
response = await http.request(request)
assert response.xml.name == "result"
assert response.xml.text == "12345"
async def test_certificate(self):
# Create a self signed server certificate
serverkey = tls.TLSPrivateKey.generate()
servercert = tls.TLSCertificate.generate(serverkey)
servercert.subject["CN"] = NAME
servercert.issuer["CN"] = NAME
servercert.sign(serverkey)
# Create a certificate authority for the client certificate
authoritykey = tls.TLSPrivateKey.generate()
authoritycert = tls.TLSCertificate.generate(authoritykey)
authoritycert.subject["CN"] = "authority"
authoritycert.issuer["CN"] = "authority"
authoritycert.sign(authoritykey)
# Create a client certificate and sign it
clientkey = tls.TLSPrivateKey.generate()
clientcert = tls.TLSCertificate.generate(clientkey)
clientcert.subject["CN"] = "testclient"
clientcert.issuer["CN"] = "authority"
clientcert.sign(authoritykey)
# Create TLS context for the server
servercontext = tls.TLSContext()
servercontext.set_certificate(servercert, serverkey)
servercontext.set_authority(authoritycert)
clientcontext = tls.TLSContext()
clientcontext.set_certificate(clientcert, clientkey)
clientcontext.set_authority(servercert)
async def handler(request):
assert request.certificate.subject["CN"] == "testclient"
return http.HTTPResponse(200)
async with http.serve(handler, HOST, 12345, servercontext):
request = http.HTTPRequest()
request.headers["Host"] = "%s:12345" % NAME
response = await http.request(request, clientcontext)
assert response.success()