mirror of
https://github.com/django/daphne.git
synced 2024-11-11 02:26:35 +03:00
99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
# coding: utf8
|
|
|
|
from hypothesis import given, settings
|
|
|
|
import http_strategies
|
|
from http_base import DaphneTestCase
|
|
|
|
|
|
class TestHTTPResponse(DaphneTestCase):
|
|
"""
|
|
Tests HTTP response handling.
|
|
"""
|
|
|
|
def normalize_headers(self, headers):
|
|
"""
|
|
Lowercases and sorts headers, and strips transfer-encoding ones.
|
|
"""
|
|
return sorted([
|
|
(name.lower(), value)
|
|
for name, value in headers
|
|
if name.lower() != "transfer-encoding"
|
|
])
|
|
|
|
def test_minimal_response(self):
|
|
"""
|
|
Smallest viable example. Mostly verifies that our response building works.
|
|
"""
|
|
response = self.run_daphne_response([
|
|
{
|
|
"type": "http.response",
|
|
"status": 200,
|
|
"content": b"hello world",
|
|
},
|
|
])
|
|
self.assertEqual(response.status, 200)
|
|
self.assertEqual(response.read(), b"hello world")
|
|
|
|
def test_status_code_required(self):
|
|
"""
|
|
Asserts that passing in the 'status' key is required.
|
|
|
|
Previous versions of Daphne did not enforce this, so this test is here
|
|
to make sure it stays required.
|
|
"""
|
|
with self.assertRaises(ValueError):
|
|
self.run_daphne_response([
|
|
{
|
|
"type": "http.response",
|
|
"content": b"hello world",
|
|
},
|
|
])
|
|
|
|
def test_custom_status_code(self):
|
|
"""
|
|
Tries a non-default status code.
|
|
"""
|
|
response = self.run_daphne_response([
|
|
{
|
|
"type": "http.response",
|
|
"status": 201,
|
|
"content": b"i made a thing!",
|
|
},
|
|
])
|
|
self.assertEqual(response.status, 201)
|
|
self.assertEqual(response.read(), b"i made a thing!")
|
|
|
|
@given(body=http_strategies.http_body())
|
|
@settings(max_examples=5, deadline=2000)
|
|
def test_body(self, body):
|
|
"""
|
|
Tries body variants.
|
|
"""
|
|
response = self.run_daphne_response([
|
|
{
|
|
"type": "http.response",
|
|
"status": 200,
|
|
"content": body,
|
|
},
|
|
])
|
|
self.assertEqual(response.status, 200)
|
|
self.assertEqual(response.read(), body)
|
|
|
|
@given(headers=http_strategies.headers())
|
|
@settings(max_examples=5, deadline=2000)
|
|
def test_headers(self, headers):
|
|
# The ASGI spec requires us to lowercase our header names
|
|
response = self.run_daphne_response([
|
|
{
|
|
"type": "http.response",
|
|
"status": 200,
|
|
"headers": self.normalize_headers(headers),
|
|
},
|
|
])
|
|
# Check headers in a sensible way. Ignore transfer-encoding.
|
|
self.assertEqual(
|
|
self.normalize_headers(response.getheaders()),
|
|
self.normalize_headers(headers),
|
|
)
|