Re-estructure tests to cleanup the apps' namespace.

mayan/apps/folders/tests/test_api.py

@@ -0,0 +1,115 @@
+from __future__ import unicode_literals
+
+from django.contrib.auth.models import User
+from django.core.files import File
+from django.core.urlresolvers import reverse
+
+from rest_framework.test import APITestCase
+
+from documents.models import DocumentType
+from documents.tests import (
+    TEST_ADMIN_EMAIL, TEST_ADMIN_PASSWORD, TEST_ADMIN_USERNAME,
+    TEST_DOCUMENT_TYPE, TEST_SMALL_DOCUMENT_PATH,
+)
+
+from ..models import Folder
+
+TEST_FOLDER_LABEL = 'test folder'
+
+
+class FolderAPITestCase(APITestCase):
+    """
+    Test the folder API endpoints
+    """
+
+    def setUp(self):
+        self.admin_user = User.objects.create_superuser(
+            username=TEST_ADMIN_USERNAME, email=TEST_ADMIN_EMAIL,
+            password=TEST_ADMIN_PASSWORD
+        )
+
+        self.client.force_authenticate(user=self.admin_user)
+
+    def tearDown(self):
+        self.admin_user.delete()
+
+    def test_folder_create(self):
+        self.client.post(
+            reverse('rest_api:folder-list'), {'label': TEST_FOLDER_LABEL}
+        )
+
+        folder = Folder.objects.first()
+
+        self.assertEqual(Folder.objects.count(), 1)
+        self.assertEqual(folder.label, TEST_FOLDER_LABEL)
+        self.assertEqual(folder.user, self.admin_user)
+
+    def test_folder_delete(self):
+        folder = Folder.objects.create(
+            label=TEST_FOLDER_LABEL, user=self.admin_user
+        )
+
+        self.client.delete(reverse('rest_api:folder-detail', args=(folder.pk,)))
+
+        self.assertEqual(Folder.objects.count(), 0)
+
+    def test_folder_edit(self):
+        folder = Folder.objects.create(
+            label=TEST_FOLDER_LABEL, user=self.admin_user
+        )
+
+        self.client.put(
+            reverse('rest_api:folder-detail', args=(folder.pk,)),
+            {'label': TEST_FOLDER_LABEL + ' edited'}
+        )
+
+        folder = Folder.objects.first()
+
+        self.assertEqual(folder.label, TEST_FOLDER_LABEL + ' edited')
+
+    def test_folder_add_document(self):
+        folder = Folder.objects.create(label=TEST_FOLDER_LABEL, user=self.admin_user)
+
+        document_type = DocumentType.objects.create(
+            label=TEST_DOCUMENT_TYPE
+        )
+
+        ocr_settings = document_type.ocr_settings
+        ocr_settings.auto_ocr = False
+        ocr_settings.save()
+
+        with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
+            document = document_type.new_document(
+                file_object=File(file_object),
+            )
+
+        self.client.post(
+            reverse('rest_api:folder-document-list', args=(folder.pk,)),
+            {'document': document.pk}
+        )
+
+        self.assertEqual(folder.documents.count(), 1)
+
+    def test_folder_remove_document(self):
+        folder = Folder.objects.create(label=TEST_FOLDER_LABEL, user=self.admin_user)
+
+        document_type = DocumentType.objects.create(
+            label=TEST_DOCUMENT_TYPE
+        )
+
+        ocr_settings = document_type.ocr_settings
+        ocr_settings.auto_ocr = False
+        ocr_settings.save()
+
+        with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
+            document = document_type.new_document(
+                file_object=File(file_object),
+            )
+
+        folder.documents.add(document)
+
+        self.client.delete(
+            reverse('rest_api:folder-document', args=(folder.pk, document.pk)),
+        )
+
+        self.assertEqual(folder.documents.count(), 0)