innov
/
django1
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

Refs #29069 -- Added test for calling request_finished signal by static file responses. 

Fixed in 41a3b3d186.
This commit is contained in:
Tom Forbes 2020-04-18 21:59:06 +01:00 committed by Mariusz Felisiak
parent 537d422942
commit 141ab6bc6d
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/builtin_server/tests.py
View File

@ -1,10 +1,11 @@
import sys
import traceback
from io import BytesIO
from unittest import TestCase
from unittest import TestCase, mock
from wsgiref import simple_server
from django.core.servers.basehttp import get_internal_wsgi_application
from django.core.signals import request_finished
from django.test import RequestFactory, override_settings
from .views import FILE_RESPONSE_HOLDER
@ -115,6 +116,15 @@ class WSGIFileWrapperTests(TestCase):
self.assertIs(buf2.closed, True)
FILE_RESPONSE_HOLDER.clear()
@override_settings(ROOT_URLCONF='builtin_server.urls')
def test_file_response_call_request_finished(self):
env = RequestFactory().get('/fileresponse/').environ
handler = FileWrapperHandler(None, BytesIO(), BytesIO(), env)
with mock.MagicMock() as signal_handler:
request_finished.connect(signal_handler)
handler.run(get_internal_wsgi_application())
self.assertEqual(signal_handler.call_count, 1)
class WriteChunkCounterHandler(ServerHandler):
"""