Add tests for the checkouts app.

mayan/apps/checkouts/tests/test_models.py

@@ -0,0 +1,117 @@
+from __future__ import unicode_literals
+
+import datetime
+import time
+
+from django.contrib.auth import get_user_model
+from django.core.files import File
+from django.test import TestCase, override_settings
+from django.utils.timezone import now
+
+from authentication.tests.literals import (
+    TEST_ADMIN_USERNAME, TEST_ADMIN_EMAIL, TEST_ADMIN_PASSWORD
+)
+from documents.models import Document, DocumentType
+from documents.tests.literals import (
+    TEST_DOCUMENT_TYPE, TEST_DOCUMENT_PATH, TEST_MULTI_PAGE_TIFF_PATH,
+    TEST_OFFICE_DOCUMENT_PATH, TEST_SMALL_DOCUMENT_PATH
+)
+
+from ..exceptions import (
+    DocumentAlreadyCheckedOut, DocumentNotCheckedOut,
+    NewDocumentVersionNotAllowed
+)
+from ..models import DocumentCheckout
+
+
+@override_settings(OCR_AUTO_OCR=False)
+class DocumentCheckoutTestCase(TestCase):
+    def setUp(self):
+        self.admin_user = get_user_model().objects.create_superuser(
+            username=TEST_ADMIN_USERNAME, email=TEST_ADMIN_EMAIL,
+            password=TEST_ADMIN_PASSWORD
+        )
+
+        self.document_type = DocumentType.objects.create(
+            label=TEST_DOCUMENT_TYPE
+        )
+
+        with open(TEST_DOCUMENT_PATH) as file_object:
+            self.document = self.document_type.new_document(
+                file_object=File(file_object)
+            )
+
+    def tearDown(self):
+        self.document_type.delete()
+
+    def test_document_checkout(self):
+        expiration_datetime = now() + datetime.timedelta(days=1)
+
+        DocumentCheckout.objects.checkout_document(
+            document=self.document, expiration_datetime=expiration_datetime,
+            user=self.admin_user, block_new_version=True
+        )
+
+        self.assertTrue(self.document.is_checked_out())
+        self.assertTrue(
+            DocumentCheckout.objects.is_document_checked_out(document=self.document)
+        )
+
+    def test_version_creation_blocking(self):
+        expiration_datetime = now() + datetime.timedelta(days=1)
+
+        DocumentCheckout.objects.checkout_document(
+            document=self.document, expiration_datetime=expiration_datetime,
+            user=self.admin_user, block_new_version=True
+        )
+
+        with self.assertRaises(NewDocumentVersionNotAllowed):
+            with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
+                self.document.new_version(file_object=File(file_object))
+
+    def test_checkin_in(self):
+        expiration_datetime = now() + datetime.timedelta(days=1)
+
+        DocumentCheckout.objects.checkout_document(
+            document=self.document, expiration_datetime=expiration_datetime,
+            user=self.admin_user, block_new_version=True
+        )
+
+        self.document.check_in()
+
+        self.assertFalse(self.document.is_checked_out())
+        self.assertFalse(
+            DocumentCheckout.objects.is_document_checked_out(document=self.document)
+        )
+
+    def test_double_checkout(self):
+        expiration_datetime = now() + datetime.timedelta(days=1)
+
+        DocumentCheckout.objects.checkout_document(
+            document=self.document, expiration_datetime=expiration_datetime,
+            user=self.admin_user, block_new_version=True
+        )
+
+        with self.assertRaises(DocumentAlreadyCheckedOut):
+            DocumentCheckout.objects.checkout_document(
+                document=self.document, expiration_datetime=expiration_datetime,
+                user=self.admin_user, block_new_version=True
+            )
+
+    def test_checkin_without_checkout(self):
+        with self.assertRaises(DocumentNotCheckedOut):
+            self.document.check_in()
+
+    def test_auto_checkin(self):
+        expiration_datetime = now() + datetime.timedelta(seconds=1)
+
+        DocumentCheckout.objects.checkout_document(
+            document=self.document, expiration_datetime=expiration_datetime,
+            user=self.admin_user, block_new_version=True
+        )
+
+        time.sleep(2)
+
+        DocumentCheckout.objects.check_in_expired_check_outs()
+
+        self.assertFalse(self.document.is_checked_out())