2016-12-22 01:45:43 -04:00
parent 71af09c1fc
commit 1e194e04fa
16 changed files with 181 additions and 99 deletions
							
							
								
							
							
						
@@ -6,9 +6,8 @@ import time
from common.tests import BaseTestCase
from django.test import override_settings
from ..exceptions import NewDocumentVersionNotAllowed
from ..literals import STUB_EXPIRATION_INTERVAL
from ..models import DeletedDocument, Document, DocumentType, NewVersionBlock
from ..models import DeletedDocument, Document, DocumentType
from .literals import (
    TEST_DOCUMENT_TYPE, TEST_DOCUMENT_PATH, TEST_MULTI_PAGE_TIFF_PATH,
							
								
							
							
								
							
							
						
@@ -275,58 +274,3 @@ class DocumentManagerTestCase(BaseTestCase):
        Document.objects.delete_stubs()
        self.assertEqual(Document.objects.count(), 0)
@override_settings(OCR_AUTO_OCR=False)
class NewVersionBlockTestCase(BaseTestCase):
    def setUp(self):
        super(NewVersionBlockTestCase, self).setUp()
        self.document_type = DocumentType.objects.create(
            label=TEST_DOCUMENT_TYPE
        )
        with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
            self.document = self.document_type.new_document(
                file_object=file_object
            )
    def tearDown(self):
        self.document.delete()
        self.document_type.delete()
        super(NewVersionBlockTestCase, self).tearDown()
    def test_blocking(self):
        NewVersionBlock.objects.block(document=self.document)
        self.assertEqual(NewVersionBlock.objects.count(), 1)
        self.assertEqual(
            NewVersionBlock.objects.first().document, self.document
        )
    def test_unblocking(self):
        NewVersionBlock.objects.create(document=self.document)
        NewVersionBlock.objects.unblock(document=self.document)
        self.assertEqual(NewVersionBlock.objects.count(), 0)
    def test_is_blocked(self):
        NewVersionBlock.objects.create(document=self.document)
        self.assertTrue(
            NewVersionBlock.objects.is_blocked(document=self.document)
        )
        NewVersionBlock.objects.all().delete()
        self.assertFalse(
            NewVersionBlock.objects.is_blocked(document=self.document)
        )
    def test_blocking_new_versions(self):
        NewVersionBlock.objects.block(document=self.document)
        with self.assertRaises(NewDocumentVersionNotAllowed):
            with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
                self.document.new_version(file_object=file_object)