diff options
| author | Boulder Sprinters <boulder-sprinters@djangoproject.com> | 2007-06-18 16:43:17 +0000 |
|---|---|---|
| committer | Boulder Sprinters <boulder-sprinters@djangoproject.com> | 2007-06-18 16:43:17 +0000 |
| commit | 750549569ea8d911db0397bd034406d5ce35d923 (patch) | |
| tree | e516db51e7fb41b457c2caa64f54d7210128840f /django/test | |
| parent | 1f09aa1e7b4079e22d1dfcde0f7d4e571389d8b5 (diff) | |
boulder-oracle-sprint: Merged to [5490]
git-svn-id: http://code.djangoproject.com/svn/django/branches/boulder-oracle-sprint@5491 bcc190cf-cafb-0310-a4f2-bffc1f526a37
Diffstat (limited to 'django/test')
| -rw-r--r-- | django/test/utils.py | 20 |
1 files changed, 16 insertions, 4 deletions
diff --git a/django/test/utils.py b/django/test/utils.py index 46de1a896c..66fb2e66e2 100644 --- a/django/test/utils.py +++ b/django/test/utils.py @@ -11,12 +11,21 @@ from django.template import Template TEST_DATABASE_PREFIX = 'test_' def instrumented_test_render(self, context): - """An instrumented Template render method, providing a signal - that can be intercepted by the test system Client - + """ + An instrumented Template render method, providing a signal that can be + intercepted by the test system Client. """ dispatcher.send(signal=signals.template_rendered, sender=self, template=self, context=context) return self.nodelist.render(context) + +def instrumented_test_iter_render(self, context): + """ + An instrumented Template iter_render method, providing a signal that can be + intercepted by the test system Client. + """ + for chunk in self.nodelist.iter_render(context): + yield chunk + dispatcher.send(signal=signals.template_rendered, sender=self, template=self, context=context) class TestSMTPConnection(object): """A substitute SMTP connection for use during test sessions. @@ -44,7 +53,9 @@ def setup_test_environment(): """ Template.original_render = Template.render + Template.original_iter_render = Template.iter_render Template.render = instrumented_test_render + Template.iter_render = instrumented_test_render mail.original_SMTPConnection = mail.SMTPConnection mail.SMTPConnection = TestSMTPConnection @@ -59,7 +70,8 @@ def teardown_test_environment(): """ Template.render = Template.original_render - del Template.original_render + Template.iter_render = Template.original_iter_render + del Template.original_render, Template.original_iter_render mail.SMTPConnection = mail.original_SMTPConnection del mail.original_SMTPConnection |