mirror of https://github.com/django/django.git
[4.2.x] Refs #34342 -- Added tests for handling sync streaming responses by test client.
Co-authored-by: Carlton Gibson <carlton.gibson@noumenal.es>
Backport of bfb8fda3e6
from main
This commit is contained in:
parent
e1c74bf458
commit
1ecbc04624
|
@ -253,7 +253,7 @@ class AdminActionsTest(TestCase):
|
|||
response = self.client.post(
|
||||
reverse("admin:admin_views_externalsubscriber_changelist"), action_data
|
||||
)
|
||||
content = b"".join(response.streaming_content)
|
||||
content = b"".join(list(response))
|
||||
self.assertEqual(content, b"This is the content of the file")
|
||||
self.assertEqual(response.status_code, 200)
|
||||
|
||||
|
|
|
@ -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"
|
||||
)
|
||||
|
|
|
@ -40,9 +40,10 @@ class StaticTests(SimpleTestCase):
|
|||
def test_chunked(self):
|
||||
"The static view should stream files in chunks to avoid large memory usage"
|
||||
response = self.client.get("/%s/%s" % (self.prefix, "long-line.txt"))
|
||||
first_chunk = next(response.streaming_content)
|
||||
response_iterator = iter(response)
|
||||
first_chunk = next(response_iterator)
|
||||
self.assertEqual(len(first_chunk), FileResponse.block_size)
|
||||
second_chunk = next(response.streaming_content)
|
||||
second_chunk = next(response_iterator)
|
||||
response.close()
|
||||
# strip() to prevent OS line endings from causing differences
|
||||
self.assertEqual(len(second_chunk.strip()), 1449)
|
||||
|
|
Loading…
Reference in New Issue