[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:
Alexandre Spaeth 2023-02-17 10:35:19 +01:00 committed by Mariusz Felisiak
parent e1c74bf458
commit 1ecbc04624
3 changed files with 22 additions and 4 deletions

View File

@ -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)

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"
)

View File

@ -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)