From 34f5fccd84455c8c8c43a3f207b86c71fadc2a3f Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Fri, 21 Jul 2017 01:35:33 -0400 Subject: [PATCH] Add tests for the document download form view. Signed-off-by: Roberto Rosario --- mayan/apps/documents/tests/test_views.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/mayan/apps/documents/tests/test_views.py b/mayan/apps/documents/tests/test_views.py index 5210c6a88c..d84056ab99 100644 --- a/mayan/apps/documents/tests/test_views.py +++ b/mayan/apps/documents/tests/test_views.py @@ -200,6 +200,29 @@ class DocumentsViewsTestCase(GenericDocumentViewTestCase): Document.objects.first().document_type, document_type_2 ) + def _request_document_download_form_view(self): + return self.get( + 'documents:document_download_form', args=(self.document.pk,), + follow=True, + ) + + def test_document_download_form_view_no_permission(self): + response = self._request_document_download_form_view() + + self.assertNotContains( + response, text=self.document.label, status_code=200 + ) + + def test_document_download_form_view_with_access(self): + self.grant_access( + obj=self.document, permission=permission_document_download + ) + response = self._request_document_download_form_view() + + self.assertContains( + response, text=self.document.label, status_code=200 + ) + def test_document_download_view_no_permission(self): response = self.get( 'documents:document_download', args=(self.document.pk,)