From 5af821e1551ff17667f30218898067197a9f03d8 Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Mon, 11 Nov 2019 02:12:04 -0400 Subject: [PATCH] Add document checkout view get test Signed-off-by: Roberto Rosario --- mayan/apps/checkouts/tests/mixins.py | 9 ++++++++- mayan/apps/checkouts/tests/test_views.py | 24 ++++++++++++++++++++---- 2 files changed, 28 insertions(+), 5 deletions(-) diff --git a/mayan/apps/checkouts/tests/mixins.py b/mayan/apps/checkouts/tests/mixins.py index 5fca8e56a0..379c388d82 100644 --- a/mayan/apps/checkouts/tests/mixins.py +++ b/mayan/apps/checkouts/tests/mixins.py @@ -72,7 +72,14 @@ class DocumentCheckoutViewTestMixin(object): } ) - def _request_test_document_check_out_view(self): + def _request_test_document_check_out_get_view(self): + return self.get( + viewname='checkouts:check_out_document', kwargs={ + 'pk': self.test_document.pk + }, + ) + + def _request_test_document_check_out_post_view(self): return self.post( viewname='checkouts:check_out_document', kwargs={ 'pk': self.test_document.pk diff --git a/mayan/apps/checkouts/tests/test_views.py b/mayan/apps/checkouts/tests/test_views.py index c701618511..d563d33936 100644 --- a/mayan/apps/checkouts/tests/test_views.py +++ b/mayan/apps/checkouts/tests/test_views.py @@ -147,13 +147,29 @@ class DocumentCheckoutViewTestCase( ) ) - def test_document_check_out_view_no_permission(self): - response = self._request_test_document_check_out_view() + def test_document_check_out_get_view_no_permission(self): + response = self._request_test_document_check_out_get_view() self.assertEqual(response.status_code, 404) self.assertFalse(self.test_document.is_checked_out()) - def test_document_check_out_view_with_access(self): + def test_document_check_out_get_view_with_access(self): + self.grant_access( + obj=self.test_document, permission=permission_document_check_out + ) + + response = self._request_test_document_check_out_get_view() + self.assertEqual(response.status_code, 200) + + self.assertFalse(self.test_document.is_checked_out()) + + def test_document_check_out_post_view_no_permission(self): + response = self._request_test_document_check_out_post_view() + self.assertEqual(response.status_code, 404) + + self.assertFalse(self.test_document.is_checked_out()) + + def test_document_check_out_post_view_with_access(self): self.grant_access( obj=self.test_document, permission=permission_document_check_out ) @@ -162,7 +178,7 @@ class DocumentCheckoutViewTestCase( permission=permission_document_check_out_detail_view ) - response = self._request_test_document_check_out_view() + response = self._request_test_document_check_out_post_view() self.assertEqual(response.status_code, 302) self.assertTrue(self.test_document.is_checked_out())