From 1d5a60ef9eead835f1d120f652c9916c790c899c Mon Sep 17 00:00:00 2001
From: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
Date: Mon, 23 Oct 2017 14:23:20 -0400
Subject: [PATCH] Update DocumentTypeAPITestCase to with test with/without
 access.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
---
 mayan/apps/documents/tests/test_api.py | 55 ++++++++++++--------------
 1 file changed, 26 insertions(+), 29 deletions(-)

diff --git a/mayan/apps/documents/tests/test_api.py b/mayan/apps/documents/tests/test_api.py
index 5903597438..6e44cfc01d 100644
--- a/mayan/apps/documents/tests/test_api.py
+++ b/mayan/apps/documents/tests/test_api.py
@@ -59,6 +59,32 @@ class DocumentTypeAPITestCase(BaseAPITestCase):
             DocumentType.objects.all().first().label, TEST_DOCUMENT_TYPE_LABEL
         )
 
+    def _request_document_type_patch(self):
+        return self.patch(
+            viewname='rest_api:documenttype-detail', args=(
+                self.document_type.pk,
+            ), data={'label': TEST_DOCUMENT_TYPE_LABEL_EDITED}
+        )
+
+    def test_document_type_edit_via_patch_with_access(self):
+        self.document_type = DocumentType.objects.create(
+            label=TEST_DOCUMENT_TYPE_LABEL
+        )
+        self.grant_access(
+            permission=permission_document_type_edit, obj=self.document_type
+        )
+        self.document_type.refresh_from_db()
+        self.assertEqual(
+            self.document_type.label, TEST_DOCUMENT_TYPE_LABEL_EDITED
+        )
+
+    def test_document_type_edit_via_patch_no_permission(self):
+        self.document_type = DocumentType.objects.create(
+            label=TEST_DOCUMENT_TYPE_LABEL
+        )
+        response = self._request_document_type_patch()
+        self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
+
     def _request_document_type_put(self):
         return self.put(
             viewname='rest_api:documenttype-detail', args=(
@@ -82,35 +108,6 @@ class DocumentTypeAPITestCase(BaseAPITestCase):
         )
         response = self._request_document_type_put()
         self.assertEqual(response.status_code, status.HTTP_200_OK)
-        self.document_type.refresh_from_db()
-        self.assertEqual(
-            self.document_type.label, TEST_DOCUMENT_TYPE_LABEL_EDITED
-        )
-
-    def _request_document_type_patch(self):
-        return self.patch(
-            viewname='rest_api:documenttype-detail', args=(
-                self.document_type.pk,
-            ), data={'label': TEST_DOCUMENT_TYPE_LABEL_EDITED}
-        )
-
-    def test_document_type_edit_via_patch_no_permission(self):
-        self.document_type = DocumentType.objects.create(
-            label=TEST_DOCUMENT_TYPE_LABEL
-        )
-        response = self._request_document_type_patch()
-        self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
-
-    def test_document_type_edit_via_patch_with_access(self):
-        self.document_type = DocumentType.objects.create(
-            label=TEST_DOCUMENT_TYPE_LABEL
-        )
-        self.grant_access(
-            permission=permission_document_type_edit, obj=self.document_type
-        )
-
-        response = self._request_document_type_patch()
-        self.assertEqual(response.status_code, status.HTTP_200_OK)
 
         self.document_type.refresh_from_db()
         self.assertEqual(