334 lines
11 KiB
Python
334 lines
11 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from datetime import timedelta
|
|
import time
|
|
|
|
from django.test import TestCase, override_settings
|
|
|
|
from organizations.tests.base import OrganizationTestCase
|
|
|
|
from ..exceptions import NewDocumentVersionNotAllowed
|
|
from ..literals import STUB_EXPIRATION_INTERVAL
|
|
from ..models import Document, DocumentType, NewVersionBlock, TrashedDocument
|
|
|
|
from .literals import (
|
|
TEST_DOCUMENT_TYPE, TEST_DOCUMENT_PATH, TEST_MULTI_PAGE_TIFF_PATH,
|
|
TEST_OFFICE_DOCUMENT_PATH, TEST_SMALL_DOCUMENT_PATH
|
|
)
|
|
|
|
|
|
@override_settings(OCR_AUTO_OCR=False)
|
|
class DocumentTestCase(OrganizationTestCase):
|
|
def setUp(self):
|
|
super(DocumentTestCase, self).setUp()
|
|
self.document_type = DocumentType.on_organization.create(
|
|
label=TEST_DOCUMENT_TYPE
|
|
)
|
|
|
|
with open(TEST_DOCUMENT_PATH) as file_object:
|
|
self.document = self.document_type.new_document(
|
|
file_object=file_object, label='mayan_11_1.pdf'
|
|
)
|
|
|
|
def tearDown(self):
|
|
self.document_type.delete()
|
|
super(DocumentTestCase, self).tearDown()
|
|
|
|
def test_document_creation(self):
|
|
self.assertEqual(self.document_type.label, TEST_DOCUMENT_TYPE)
|
|
|
|
self.assertEqual(self.document.exists(), True)
|
|
self.assertEqual(self.document.size, 272213)
|
|
|
|
self.assertEqual(self.document.file_mimetype, 'application/pdf')
|
|
self.assertEqual(self.document.file_mime_encoding, 'binary')
|
|
self.assertEqual(self.document.label, 'mayan_11_1.pdf')
|
|
self.assertEqual(
|
|
self.document.checksum,
|
|
'c637ffab6b8bb026ed3784afdb07663fddc60099853fae2be93890852a69ecf3'
|
|
)
|
|
self.assertEqual(self.document.page_count, 47)
|
|
|
|
def test_version_creation(self):
|
|
with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
|
|
self.document.new_version(file_object=file_object)
|
|
|
|
with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
|
|
self.document.new_version(
|
|
file_object=file_object, comment='test comment 1'
|
|
)
|
|
|
|
self.assertEqual(self.document.versions.count(), 3)
|
|
|
|
def test_restoring_documents(self):
|
|
self.assertEqual(Document.on_organization.count(), 1)
|
|
|
|
# Trash the document
|
|
self.document.delete()
|
|
self.assertEqual(TrashedDocument.on_organization.count(), 1)
|
|
self.assertEqual(Document.on_organization.count(), 0)
|
|
|
|
# Restore the document
|
|
self.document.restore()
|
|
self.assertEqual(TrashedDocument.on_organization.count(), 0)
|
|
self.assertEqual(Document.on_organization.count(), 1)
|
|
|
|
def test_trashing_documents(self):
|
|
self.assertEqual(Document.on_organization.count(), 1)
|
|
|
|
# Trash the document
|
|
self.document.delete()
|
|
self.assertEqual(TrashedDocument.on_organization.count(), 1)
|
|
self.assertEqual(Document.on_organization.count(), 0)
|
|
|
|
# Delete the document
|
|
self.document.delete()
|
|
self.assertEqual(TrashedDocument.on_organization.count(), 0)
|
|
self.assertEqual(Document.on_organization.count(), 0)
|
|
|
|
def test_auto_trashing(self):
|
|
"""
|
|
Test document type trashing policies. Documents are moved to the
|
|
trash, x amount of time after being uploaded
|
|
"""
|
|
|
|
self.document_type.trash_time_period = 1
|
|
# 'seconds' is not a choice via the model, used here for convenience
|
|
self.document_type.trash_time_unit = 'seconds'
|
|
self.document_type.save()
|
|
|
|
# Needed by MySQL as milliseconds value is not store in timestamp
|
|
# field
|
|
time.sleep(2)
|
|
|
|
self.assertEqual(Document.on_organization.count(), 1)
|
|
self.assertEqual(TrashedDocument.on_organization.count(), 0)
|
|
|
|
DocumentType.objects.check_trash_periods()
|
|
|
|
self.assertEqual(Document.on_organization.count(), 0)
|
|
self.assertEqual(TrashedDocument.on_organization.count(), 1)
|
|
|
|
def test_auto_delete(self):
|
|
"""
|
|
Test document type deletion policies. Documents are deleted from the
|
|
trash, x amount of time after being trashed
|
|
"""
|
|
|
|
self.document_type.delete_time_period = 1
|
|
# 'seconds' is not a choice via the model, used here for convenience
|
|
self.document_type.delete_time_unit = 'seconds'
|
|
self.document_type.save()
|
|
|
|
self.assertEqual(Document.on_organization.count(), 1)
|
|
self.assertEqual(TrashedDocument.on_organization.count(), 0)
|
|
|
|
self.document.delete()
|
|
|
|
self.assertEqual(Document.on_organization.count(), 0)
|
|
self.assertEqual(TrashedDocument.on_organization.count(), 1)
|
|
|
|
# Needed by MySQL as milliseconds value is not store in timestamp
|
|
# field
|
|
time.sleep(2)
|
|
|
|
DocumentType.objects.check_delete_periods()
|
|
|
|
self.assertEqual(Document.on_organization.count(), 0)
|
|
self.assertEqual(TrashedDocument.on_organization.count(), 0)
|
|
|
|
|
|
@override_settings(OCR_AUTO_OCR=False)
|
|
class OfficeDocumentTestCase(OrganizationTestCase):
|
|
def setUp(self):
|
|
super(OfficeDocumentTestCase, self).setUp()
|
|
|
|
self.document_type = DocumentType.on_organization.create(
|
|
label=TEST_DOCUMENT_TYPE
|
|
)
|
|
|
|
with open(TEST_OFFICE_DOCUMENT_PATH) as file_object:
|
|
self.document = self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
def tearDown(self):
|
|
self.document_type.delete()
|
|
super(OfficeDocumentTestCase, self).tearDown()
|
|
|
|
def test_document_creation(self):
|
|
self.assertEqual(self.document.file_mimetype, 'application/msword')
|
|
self.assertEqual(
|
|
self.document.file_mime_encoding, 'application/mswordbinary'
|
|
)
|
|
self.assertEqual(
|
|
self.document.checksum,
|
|
'03a7e9071d2c6ae05a6588acd7dff1d890fac2772cf61abd470c9ffa6ef71f03'
|
|
)
|
|
self.assertEqual(self.document.page_count, 2)
|
|
|
|
|
|
@override_settings(OCR_AUTO_OCR=False)
|
|
class MultiPageTiffTestCase(OrganizationTestCase):
|
|
def setUp(self):
|
|
super(MultiPageTiffTestCase, self).setUp()
|
|
|
|
self.document_type = DocumentType.on_organization.create(
|
|
label=TEST_DOCUMENT_TYPE
|
|
)
|
|
|
|
with open(TEST_MULTI_PAGE_TIFF_PATH) as file_object:
|
|
self.document = self.document_type.new_document(
|
|
file_object=file_object
|
|
)
|
|
|
|
def tearDown(self):
|
|
self.document_type.delete()
|
|
super(MultiPageTiffTestCase, self).tearDown()
|
|
|
|
def test_document_creation(self):
|
|
self.assertEqual(self.document.file_mimetype, 'image/tiff')
|
|
self.assertEqual(self.document.file_mime_encoding, 'binary')
|
|
self.assertEqual(
|
|
self.document.checksum,
|
|
'40adaa9d658b65c70a7f002dfe084a8354bb77c0dfbf1993e31fb024a285fb1d'
|
|
)
|
|
self.assertEqual(self.document.page_count, 2)
|
|
|
|
|
|
@override_settings(OCR_AUTO_OCR=False)
|
|
class DocumentVersionTestCase(OrganizationTestCase):
|
|
def setUp(self):
|
|
super(DocumentVersionTestCase, self).setUp()
|
|
|
|
self.document_type = DocumentType.on_organization.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_type.delete()
|
|
super(DocumentVersionTestCase, self).tearDown()
|
|
|
|
def test_add_new_version(self):
|
|
self.assertEqual(self.document.versions.count(), 1)
|
|
|
|
with open(TEST_DOCUMENT_PATH) as file_object:
|
|
self.document.new_version(
|
|
file_object=file_object
|
|
)
|
|
|
|
self.assertEqual(self.document.versions.count(), 2)
|
|
|
|
self.assertEqual(
|
|
self.document.checksum,
|
|
'c637ffab6b8bb026ed3784afdb07663fddc60099853fae2be93890852a69ecf3'
|
|
)
|
|
|
|
def test_revert_version(self):
|
|
self.assertEqual(self.document.versions.count(), 1)
|
|
|
|
# Needed by MySQL as milliseconds value is not store in timestamp
|
|
# field
|
|
time.sleep(2)
|
|
|
|
with open(TEST_DOCUMENT_PATH) as file_object:
|
|
self.document.new_version(
|
|
file_object=file_object
|
|
)
|
|
|
|
self.assertEqual(self.document.versions.count(), 2)
|
|
|
|
self.document.versions.first().revert()
|
|
|
|
self.assertEqual(self.document.versions.count(), 1)
|
|
|
|
|
|
@override_settings(OCR_AUTO_OCR=False)
|
|
class DocumentManagerTestCase(OrganizationTestCase):
|
|
def setUp(self):
|
|
super(DocumentManagerTestCase, self).setUp()
|
|
self.document_type = DocumentType.on_organization.create(
|
|
label=TEST_DOCUMENT_TYPE
|
|
)
|
|
|
|
def tearDown(self):
|
|
self.document_type.delete()
|
|
super(DocumentManagerTestCase, self).tearDown()
|
|
|
|
def test_document_stubs_deletion(self):
|
|
document_stub = Document.on_organization.create(
|
|
document_type=self.document_type
|
|
)
|
|
|
|
Document.objects.delete_stubs()
|
|
|
|
self.assertEqual(Document.on_organization.count(), 1)
|
|
|
|
document_stub.date_added = document_stub.date_added - timedelta(
|
|
seconds=STUB_EXPIRATION_INTERVAL + 1
|
|
)
|
|
document_stub.save()
|
|
|
|
Document.objects.delete_stubs()
|
|
|
|
self.assertEqual(Document.on_organization.count(), 0)
|
|
|
|
|
|
@override_settings(OCR_AUTO_OCR=False)
|
|
class NewVersionBlockTestCase(OrganizationTestCase):
|
|
def setUp(self):
|
|
super(NewVersionBlockTestCase, self).setUp()
|
|
self.document_type = DocumentType.on_organization.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)
|