mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-08-04 10:59:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Refs #34342 -- Added tests for handling sync streaming responses by test client.

Browse source 
Co-authored-by: Carlton Gibson <carlton.gibson@noumenal.es>
This commit is contained in:
Alexandre Spaeth 2023-02-17 10:35:19 +01:00 • committed by Mariusz Felisiak
parent 8eef22dfed
commit bfb8fda3e6
3 changed files with 22 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/handlers/tests.py
View file

@ -171,7 +171,7 @@ class SignalsTests(SimpleTestCase):
def test_request_signals_streaming_response(self):
response = self.client.get("/streaming/")
self.assertEqual(self.signals, ["started"])
self.assertEqual(b"".join(response.streaming_content), b"streaming content")
self.assertEqual(b"".join(list(response)), b"streaming content")
self.assertEqual(self.signals, ["started", "finished"])
@ -248,6 +248,11 @@ class HandlerRequestTests(SimpleTestCase):
):
self.client.get(url)
def test_streaming(self):
response = self.client.get("/streaming/")
self.assertEqual(response.status_code, 200)
self.assertEqual(b"".join(list(response)), b"streaming content")
class ScriptNameTests(SimpleTestCase):
def test_get_script_name(self):
@ -312,3 +317,15 @@ class AsyncHandlerRequestTests(SimpleTestCase):
)
with self.assertRaisesMessage(ValueError, msg):
await self.async_client.get("/unawaited/")
async def test_sync_streaming(self):
response = await self.async_client.get("/streaming/")
self.assertEqual(response.status_code, 200)
msg = (
"StreamingHttpResponse must consume synchronous iterators in order to "
"serve them asynchronously. Use an asynchronous iterator instead."
)
with self.assertWarnsMessage(Warning, msg):
self.assertEqual(
b"".join([chunk async for chunk in response]), b"streaming content"
)