matthias/mayan-edms
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Test simplification

Browse Source 
Unify usage of self.test_document and self.test_documents in
tests.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2019-05-07 00:54:59 -04:00
parent ebc68d3c36
commit a0d2000419
36 changed files with 306 additions and 276 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mayan/apps/documents/tests/mixins.py
+4 -5
View File
@@ -22,10 +22,10 @@ class DocumentTestMixin(object):
test_document_path = None
def _create_document_type(self):
self.document_type = DocumentType.objects.create(
self.test_document_type = DocumentType.objects.create(
label=TEST_DOCUMENT_TYPE_LABEL
)
self.test_document_type = self.document_type
self.test_document_type = self.test_document_type
def upload_document(self, label=None):
self._calculate_test_document_path()
@@ -34,13 +34,12 @@ class DocumentTestMixin(object):
label = self.test_document_filename
with open(self.test_document_path, mode='rb') as file_object:
document = self.document_type.new_document(
document = self.test_document_type.new_document(
file_object=file_object, label=label
)
self.test_document = document
self.test_documents.append(document)
return document
def _calculate_test_document_path(self):
if not self.test_document_path:
@@ -57,7 +56,7 @@ class DocumentTestMixin(object):
self._create_document_type()
if self.auto_upload_document:
self.document = self.upload_document()
self.upload_document()
def tearDown(self):
for document_type in DocumentType.objects.all():
mayan/apps/documents/tests/test_api.py
+23 -15
View File
@@ -77,7 +77,8 @@ class DocumentTypeAPITestCase(DocumentTestMixin, BaseAPITestCase):
label=TEST_DOCUMENT_TYPE_LABEL
)
self.grant_access(
permission=permission_document_type_edit, obj=self.test_document_type
obj=self.test_document_type,
permission=permission_document_type_edit
)
response = self._request_document_type_put()
@@ -106,7 +107,8 @@ class DocumentTypeAPITestCase(DocumentTestMixin, BaseAPITestCase):
label=TEST_DOCUMENT_TYPE_LABEL
)
self.grant_access(
permission=permission_document_type_edit, obj=self.test_document_type
obj=self.test_document_type,
permission=permission_document_type_edit
)
response = self._request_document_type_put()
@@ -138,7 +140,8 @@ class DocumentTypeAPITestCase(DocumentTestMixin, BaseAPITestCase):
label=TEST_DOCUMENT_TYPE_LABEL
)
self.grant_access(
permission=permission_document_type_delete, obj=self.test_document_type
obj=self.test_document_type,
permission=permission_document_type_delete
)
response = self._request_document_type_delete()
@@ -165,7 +168,7 @@ class DocumentAPITestCase(DocumentTestMixin, BaseAPITestCase):
def test_document_upload_with_access(self):
self.grant_access(
permission=permission_document_create, obj=self.test_document_type
obj=self.test_document_type, permission=permission_document_create
)
response = self._request_document_upload()
@@ -223,7 +226,7 @@ class DocumentAPITestCase(DocumentTestMixin, BaseAPITestCase):
def test_document_new_version_upload_with_access(self):
self.upload_document()
self.grant_access(
permission=permission_document_new_version, obj=self.test_document
obj=self.test_document, permission=permission_document_new_version
)
response = self._request_document_new_version_upload()
@@ -298,7 +301,7 @@ class DocumentAPITestCase(DocumentTestMixin, BaseAPITestCase):
self.upload_document()
self._create_new_version()
self.grant_access(
permission=permission_document_version_view, obj=self.test_document
obj=self.test_document, permission=permission_document_version_view
)
response = self._request_document_version_list()
self.assertEqual(response.status_code, status.HTTP_200_OK)
@@ -324,7 +327,7 @@ class DocumentAPITestCase(DocumentTestMixin, BaseAPITestCase):
def test_document_download_with_access(self):
self.upload_document()
self.grant_access(
permission=permission_document_download, obj=self.test_document
obj=self.test_document, permission=permission_document_download
)
response = self._request_document_download()
@@ -411,7 +414,7 @@ class DocumentAPITestCase(DocumentTestMixin, BaseAPITestCase):
def test_document_version_edit_via_patch_with_access(self):
self.upload_document()
self.grant_access(
permission=permission_document_edit, obj=self.test_document
obj=self.test_document, permission=permission_document_edit
)
response = self._request_document_version_edit_via_patch()
self.assertEqual(response.status_code, status.HTTP_200_OK)
@@ -439,7 +442,7 @@ class DocumentAPITestCase(DocumentTestMixin, BaseAPITestCase):
def test_document_version_edit_via_put_with_access(self):
self.upload_document()
self.grant_access(
permission=permission_document_edit, obj=self.test_document
obj=self.test_document, permission=permission_document_edit
)
response = self._request_document_version_edit_via_put()
@@ -497,7 +500,8 @@ class DocumentAPITestCase(DocumentTestMixin, BaseAPITestCase):
def test_document_description_edit_via_put_with_access(self):
self.upload_document()
self.grant_access(
permission=permission_document_properties_edit, obj=self.test_document
obj=self.test_document,
permission=permission_document_properties_edit
)
response = self._request_document_description_edit_via_put()
@@ -526,7 +530,7 @@ class DocumentPageAPITestCase(DocumentTestMixin, BaseAPITestCase):
def test_document_page_image_view_with_access(self):
self.grant_access(
permission=permission_document_view, obj=self.test_document
obj=self.test_document, permission=permission_document_view
)
response = self._request_document_page_image()
@@ -554,7 +558,7 @@ class TrashedDocumentAPITestCase(DocumentTestMixin, BaseAPITestCase):
self.upload_document()
self.grant_access(
permission=permission_document_trash, obj=self.test_document
obj=self.test_document, permission=permission_document_trash
)
response = self._request_document_move_to_trash()
@@ -613,11 +617,15 @@ class TrashedDocumentAPITestCase(DocumentTestMixin, BaseAPITestCase):
def test_trashed_document_detail_view_with_access(self):
self.upload_document()
self.test_document.delete()
self.grant_access(permission=permission_document_view, obj=self.test_document)
self.grant_access(
obj=self.test_document, permission=permission_document_view
)
response = self._request_trashed_document_detail_view()
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.data['uuid'], force_text(self.test_document.uuid))
self.assertEqual(
response.data['uuid'], force_text(self.test_document.uuid)
)
def _request_trashed_document_list_view(self):
return self.get(
@@ -636,7 +644,7 @@ class TrashedDocumentAPITestCase(DocumentTestMixin, BaseAPITestCase):
self.upload_document()
self.test_document.delete()
self.grant_access(
permission=permission_document_view, obj=self.test_document
obj=self.test_document, permission=permission_document_view
)
response = self._request_trashed_document_list_view()
mayan/apps/documents/tests/test_document_type_views.py
+9 -4
View File
@@ -62,7 +62,8 @@ class DocumentTypeViewsTestCase(GenericDocumentViewTestCase):
def test_document_type_delete_view_with_access(self):
self.grant_access(
obj=self.test_document_type, permission=permission_document_type_delete
obj=self.test_document_type,
permission=permission_document_type_delete
)
response = self._request_document_type_delete()
@@ -106,7 +107,9 @@ class DocumentTypeViewsTestCase(GenericDocumentViewTestCase):
)
class DocumentTypeQuickLabelViewsTestCase(DocumentTypeQuickLabelTestMixin, GenericDocumentViewTestCase):
class DocumentTypeQuickLabelViewsTestCase(
DocumentTypeQuickLabelTestMixin, GenericDocumentViewTestCase
):
def _request_quick_label_create(self):
return self.post(
viewname='documents:document_type_filename_create',
@@ -129,7 +132,8 @@ class DocumentTypeQuickLabelViewsTestCase(DocumentTypeQuickLabelTestMixin, Gener
def test_document_type_quick_label_create_with_access(self):
self.grant_access(
obj=self.test_document_type, permission=permission_document_type_edit
obj=self.test_document_type,
permission=permission_document_type_edit
)
response = self._request_quick_label_create()
@@ -154,7 +158,8 @@ class DocumentTypeQuickLabelViewsTestCase(DocumentTypeQuickLabelTestMixin, Gener
def test_document_type_quick_label_delete_with_access(self):
self.grant_access(
obj=self.test_document_type, permission=permission_document_type_edit
obj=self.test_document_type,
permission=permission_document_type_edit
)
self._create_test_quick_label()
mayan/apps/documents/tests/test_document_version_views.py
+7 -7
View File
@@ -13,7 +13,7 @@ class DocumentVersionTestCase(DocumentVersionTestMixin, GenericDocumentViewTestC
def _request_document_version_list_view(self):
return self.get(
viewname='documents:document_version_list',
kwargs={'pk': self.document.pk}
kwargs={'pk': self.test_document.pk}
)
def test_document_version_list_no_permission(self):
@@ -25,7 +25,7 @@ class DocumentVersionTestCase(DocumentVersionTestMixin, GenericDocumentViewTestC
def test_document_version_list_with_access(self):
self._upload_new_version()
self.grant_access(
obj=self.document, permission=permission_document_version_view
obj=self.test_document, permission=permission_document_version_view
)
response = self._request_document_version_list_view()
@@ -40,7 +40,7 @@ class DocumentVersionTestCase(DocumentVersionTestMixin, GenericDocumentViewTestC
)
def test_document_version_revert_no_permission(self):
first_version = self.document.latest_version
first_version = self.test_document.latest_version
self._upload_new_version()
response = self._request_document_version_revert_view(
@@ -48,14 +48,14 @@ class DocumentVersionTestCase(DocumentVersionTestMixin, GenericDocumentViewTestC
)
self.assertEqual(response.status_code, 404)
self.assertEqual(self.document.versions.count(), 2)
self.assertEqual(self.test_document.versions.count(), 2)
def test_document_version_revert_with_access(self):
first_version = self.document.latest_version
first_version = self.test_document.latest_version
self._upload_new_version()
self.grant_access(
obj=self.document, permission=permission_document_version_revert
obj=self.test_document, permission=permission_document_version_revert
)
response = self._request_document_version_revert_view(
@@ -63,4 +63,4 @@ class DocumentVersionTestCase(DocumentVersionTestMixin, GenericDocumentViewTestC
)
self.assertEqual(response.status_code, 302)
self.assertEqual(self.document.versions.count(), 1)
self.assertEqual(self.test_document.versions.count(), 1)
mayan/apps/documents/tests/test_duplicated_document_views.py
+14 -9
View File
@@ -7,7 +7,7 @@ from .base import GenericDocumentViewTestCase
class DuplicatedDocumentsViewsTestCase(GenericDocumentViewTestCase):
def _upload_duplicate_document(self):
self.test_duplicated_document = self.upload_document()
self.upload_document()
def _request_duplicated_document_list_view(self):
return self.get(viewname='documents:duplicated_document_list')
@@ -15,7 +15,7 @@ class DuplicatedDocumentsViewsTestCase(GenericDocumentViewTestCase):
def _request_document_duplicates_list_view(self):
return self.get(
viewname='documents:document_duplicates_list',
kwargs={'pk': self.test_document.pk}
kwargs={'pk': self.test_documents[0].pk}
)
def test_duplicated_document_list_no_permissions(self):
@@ -23,22 +23,25 @@ class DuplicatedDocumentsViewsTestCase(GenericDocumentViewTestCase):
response = self._request_duplicated_document_list_view()
self.assertNotContains(
response=response, text=self.test_document.label, status_code=200
response=response, text=self.test_documents[0].label,
status_code=200
)
def test_duplicated_document_list_with_access(self):
self._upload_duplicate_document()
self.grant_access(
obj=self.test_document, permission=permission_document_view
obj=self.test_documents[0],
permission=permission_document_view
)
self.grant_access(
obj=self.test_duplicated_document,
obj=self.test_documents[1],
permission=permission_document_view
)
response = self._request_duplicated_document_list_view()
self.assertContains(
response=response, text=self.test_document.label, status_code=200
response=response, text=self.test_documents[0].label,
status_code=200
)
def test_document_duplicates_list_no_permissions(self):
@@ -50,14 +53,16 @@ class DuplicatedDocumentsViewsTestCase(GenericDocumentViewTestCase):
def test_document_duplicates_list_with_access(self):
self._upload_duplicate_document()
self.grant_access(
obj=self.test_document, permission=permission_document_view
obj=self.test_documents[0],
permission=permission_document_view
)
self.grant_access(
obj=self.test_duplicated_document,
obj=self.test_documents[1],
permission=permission_document_view
)
response = self._request_document_duplicates_list_view()
self.assertContains(
response=response, text=self.test_document.label, status_code=200
response=response, text=self.test_documents[0].label,
status_code=200
)
mayan/apps/documents/tests/test_links.py
+13 -13
View File
@@ -21,11 +21,11 @@ from .literals import TEST_SMALL_DOCUMENT_PATH
class DocumentsLinksTestCase(GenericDocumentViewTestCase):
def test_document_version_revert_link_no_permission(self):
with open(TEST_SMALL_DOCUMENT_PATH, mode='rb') as file_object:
self.document.new_version(file_object=file_object)
self.test_document.new_version(file_object=file_object)
self.assertTrue(self.document.versions.count(), 2)
self.assertTrue(self.test_document.versions.count(), 2)
self.add_test_view(test_object=self.document.versions.first())
self.add_test_view(test_object=self.test_document.versions.first())
context = self.get_test_view()
resolved_link = link_document_version_revert.resolve(context=context)
@@ -37,16 +37,16 @@ class DocumentsLinksTestCase(GenericDocumentViewTestCase):
time.sleep(1.01)
with open(TEST_SMALL_DOCUMENT_PATH, mode='rb') as file_object:
self.document.new_version(file_object=file_object)
self.test_document.new_version(file_object=file_object)
self.assertTrue(self.document.versions.count(), 2)
self.assertTrue(self.test_document.versions.count(), 2)
self.grant_access(
obj=self.test_document,
permission=permission_document_version_revert
)
self.add_test_view(test_object=self.document.versions.first())
self.add_test_view(test_object=self.test_document.versions.first())
context = self.get_test_view()
resolved_link = link_document_version_revert.resolve(context=context)
@@ -55,12 +55,12 @@ class DocumentsLinksTestCase(GenericDocumentViewTestCase):
resolved_link.url,
reverse(
viewname=link_document_version_revert.view,
args=(self.document.versions.first().pk,)
args=(self.test_document.versions.first().pk,)
)
)
def test_document_version_download_link_no_permission(self):
self.add_test_view(test_object=self.document.latest_version)
self.add_test_view(test_object=self.test_document.latest_version)
context = self.get_test_view()
resolved_link = link_document_version_download.resolve(context=context)
@@ -72,7 +72,7 @@ class DocumentsLinksTestCase(GenericDocumentViewTestCase):
permission=permission_document_download
)
self.add_test_view(test_object=self.document.latest_version)
self.add_test_view(test_object=self.test_document.latest_version)
context = self.get_test_view()
resolved_link = link_document_version_download.resolve(context=context)
@@ -81,7 +81,7 @@ class DocumentsLinksTestCase(GenericDocumentViewTestCase):
resolved_link.url,
reverse(
viewname=link_document_version_download.view,
args=(self.document.latest_version.pk,)
args=(self.test_document.latest_version.pk,)
)
)
@@ -90,9 +90,9 @@ class DeletedDocumentsLinksTestCase(GenericDocumentViewTestCase):
def setUp(self):
super(DeletedDocumentsLinksTestCase, self).setUp()
self.login_user()
self.document.delete()
self.test_document.delete()
self.test_deleted_document = DeletedDocument.objects.get(
pk=self.document.pk
pk=self.test_document.pk
)
self.add_test_view(test_object=self.test_deleted_document)
self.context = self.get_test_view()
@@ -103,7 +103,7 @@ class DeletedDocumentsLinksTestCase(GenericDocumentViewTestCase):
def test_deleted_document_restore_link_with_permission(self):
self.grant_access(
obj=self.document, permission=permission_document_restore
obj=self.test_document, permission=permission_document_restore
)
resolved_link = link_document_restore.resolve(context=self.context)
self.assertNotEqual(resolved_link, None)
mayan/apps/documents/tests/test_models.py
+73 -61
View File
@@ -23,46 +23,46 @@ from .mixins import DocumentTestMixin
class DocumentTestCase(DocumentTestMixin, BaseTestCase):
def test_natural_keys(self):
self.document.pages.first().generate_image()
self.test_document.pages.first().generate_image()
self._test_database_conversion('documents')
def test_document_creation(self):
self.assertEqual(self.document_type.label, TEST_DOCUMENT_TYPE_LABEL)
self.assertEqual(self.test_document_type.label, TEST_DOCUMENT_TYPE_LABEL)
self.assertEqual(self.document.exists(), True)
self.assertEqual(self.document.size, TEST_SMALL_DOCUMENT_SIZE)
self.assertEqual(self.test_document.exists(), True)
self.assertEqual(self.test_document.size, TEST_SMALL_DOCUMENT_SIZE)
self.assertEqual(
self.document.file_mimetype, TEST_SMALL_DOCUMENT_MIMETYPE
self.test_document.file_mimetype, TEST_SMALL_DOCUMENT_MIMETYPE
)
self.assertEqual(self.document.file_mime_encoding, 'binary')
self.assertEqual(self.document.label, TEST_SMALL_DOCUMENT_FILENAME)
self.assertEqual(self.test_document.file_mime_encoding, 'binary')
self.assertEqual(self.test_document.label, TEST_SMALL_DOCUMENT_FILENAME)
self.assertEqual(
self.document.checksum, TEST_SMALL_DOCUMENT_CHECKSUM
self.test_document.checksum, TEST_SMALL_DOCUMENT_CHECKSUM
)
self.assertEqual(self.document.page_count, 1)
self.assertEqual(self.test_document.page_count, 1)
def test_version_creation(self):
with open(TEST_SMALL_DOCUMENT_PATH, mode='rb') as file_object:
self.document.new_version(file_object=file_object)
self.test_document.new_version(file_object=file_object)
with open(TEST_SMALL_DOCUMENT_PATH, mode='rb') as file_object:
self.document.new_version(
self.test_document.new_version(
file_object=file_object, comment='test comment 1'
)
self.assertEqual(self.document.versions.count(), 3)
self.assertEqual(self.test_document.versions.count(), 3)
def test_restoring_documents(self):
self.assertEqual(Document.objects.count(), 1)
# Trash the document
self.document.delete()
self.test_document.delete()
self.assertEqual(DeletedDocument.objects.count(), 1)
self.assertEqual(Document.objects.count(), 0)
# Restore the document
self.document.restore()
self.test_document.restore()
self.assertEqual(DeletedDocument.objects.count(), 0)
self.assertEqual(Document.objects.count(), 1)
@@ -70,12 +70,12 @@ class DocumentTestCase(DocumentTestMixin, BaseTestCase):
self.assertEqual(Document.objects.count(), 1)
# Trash the document
self.document.delete()
self.test_document.delete()
self.assertEqual(DeletedDocument.objects.count(), 1)
self.assertEqual(Document.objects.count(), 0)
# Delete the document
self.document.delete()
self.test_document.delete()
self.assertEqual(DeletedDocument.objects.count(), 0)
self.assertEqual(Document.objects.count(), 0)
@@ -84,10 +84,10 @@ class DocumentTestCase(DocumentTestMixin, BaseTestCase):
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
self.test_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()
self.test_document_type.trash_time_unit = 'seconds'
self.test_document_type.save()
# Needed by MySQL as milliseconds value is not store in timestamp
# field
@@ -106,15 +106,15 @@ class DocumentTestCase(DocumentTestMixin, BaseTestCase):
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
self.test_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.test_document_type.delete_time_unit = 'seconds'
self.test_document_type.save()
self.assertEqual(Document.objects.count(), 1)
self.assertEqual(DeletedDocument.objects.count(), 0)
self.document.delete()
self.test_document.delete()
self.assertEqual(Document.objects.count(), 0)
self.assertEqual(DeletedDocument.objects.count(), 1)
@@ -131,17 +131,17 @@ class DocumentTestCase(DocumentTestMixin, BaseTestCase):
class PDFCompatibilityTestCase(BaseTestCase):
def test_indirect_rotate(self):
self.document_type = DocumentType.objects.create(
self.test_document_type = DocumentType.objects.create(
label=TEST_DOCUMENT_TYPE_LABEL
)
with open(TEST_PDF_INDIRECT_ROTATE_PATH, mode='rb') as file_object:
self.document = self.document_type.new_document(
self.test_document = self.test_document_type.new_document(
file_object=file_object
)
self.assertQuerysetEqual(
qs=Document.objects.all(), values=(repr(self.document),)
qs=Document.objects.all(), values=(repr(self.test_document),)
)
@@ -149,106 +149,106 @@ class OfficeDocumentTestCase(BaseTestCase):
def setUp(self):
super(OfficeDocumentTestCase, self).setUp()
self.document_type = DocumentType.objects.create(
self.test_document_type = DocumentType.objects.create(
label=TEST_DOCUMENT_TYPE_LABEL
)
with open(TEST_OFFICE_DOCUMENT_PATH, mode='rb') as file_object:
self.document = self.document_type.new_document(
self.test_document = self.test_document_type.new_document(
file_object=file_object
)
def tearDown(self):
self.document_type.delete()
self.test_document_type.delete()
super(OfficeDocumentTestCase, self).tearDown()
def test_document_creation(self):
self.assertEqual(self.document.file_mimetype, 'application/msword')
self.assertEqual(self.test_document.file_mimetype, 'application/msword')
self.assertEqual(
self.document.file_mime_encoding, 'binary'
self.test_document.file_mime_encoding, 'binary'
)
self.assertEqual(
self.document.checksum,
self.test_document.checksum,
'03a7e9071d2c6ae05a6588acd7dff1d890fac2772cf61abd470c9ffa6ef71f03'
)
self.assertEqual(self.document.page_count, 2)
self.assertEqual(self.test_document.page_count, 2)
class MultiPageTiffTestCase(BaseTestCase):
def setUp(self):
super(MultiPageTiffTestCase, self).setUp()
self.document_type = DocumentType.objects.create(
self.test_document_type = DocumentType.objects.create(
label=TEST_DOCUMENT_TYPE_LABEL
)
with open(TEST_MULTI_PAGE_TIFF_PATH, mode='rb') as file_object:
self.document = self.document_type.new_document(
self.test_document = self.test_document_type.new_document(
file_object=file_object
)
def tearDown(self):
self.document_type.delete()
self.test_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.test_document.file_mimetype, 'image/tiff')
self.assertEqual(self.test_document.file_mime_encoding, 'binary')
self.assertEqual(
self.document.checksum,
self.test_document.checksum,
'40adaa9d658b65c70a7f002dfe084a8354bb77c0dfbf1993e31fb024a285fb1d'
)
self.assertEqual(self.document.page_count, 2)
self.assertEqual(self.test_document.page_count, 2)
class DocumentVersionTestCase(GenericDocumentTestCase):
def test_add_new_version(self):
self.assertEqual(self.document.versions.count(), 1)
self.assertEqual(self.test_document.versions.count(), 1)
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
self.document.new_version(
self.test_document.new_version(
file_object=file_object
)
self.assertEqual(self.document.versions.count(), 2)
self.assertEqual(self.test_document.versions.count(), 2)
self.assertEqual(
self.document.checksum,
self.test_document.checksum,
'c637ffab6b8bb026ed3784afdb07663fddc60099853fae2be93890852a69ecf3'
)
def test_revert_version(self):
self.assertEqual(self.document.versions.count(), 1)
self.assertEqual(self.test_document.versions.count(), 1)
# Needed by MySQL as milliseconds value is not store in timestamp
# field
time.sleep(1.01)
with open(TEST_DOCUMENT_PATH, mode='rb') as file_object:
self.document.new_version(
self.test_document.new_version(
file_object=file_object
)
self.assertEqual(self.document.versions.count(), 2)
self.assertEqual(self.test_document.versions.count(), 2)
self.document.versions.first().revert()
self.test_document.versions.first().revert()
self.assertEqual(self.document.versions.count(), 1)
self.assertEqual(self.test_document.versions.count(), 1)
class DocumentManagerTestCase(BaseTestCase):
def setUp(self):
super(DocumentManagerTestCase, self).setUp()
self.document_type = DocumentType.objects.create(
self.test_document_type = DocumentType.objects.create(
label=TEST_DOCUMENT_TYPE_LABEL
)
def tearDown(self):
self.document_type.delete()
self.test_document_type.delete()
super(DocumentManagerTestCase, self).tearDown()
def test_document_stubs_deletion(self):
document_stub = Document.objects.create(
document_type=self.document_type
document_type=self.test_document_type
)
Document.passthrough.delete_stubs()
@@ -267,19 +267,31 @@ class DocumentManagerTestCase(BaseTestCase):
class DuplicatedDocumentsTestCase(GenericDocumentTestCase):
def test_duplicates_after_delete(self):
document_2 = self.upload_document()
document_2.delete()
document_2.delete()
self.upload_document()
self.test_documents[1].delete()
self.test_documents[1].delete()
self.assertEqual(DuplicatedDocument.objects.filter(document=self.document).count(), 0)
self.assertEqual(
DuplicatedDocument.objects.filter(
document=self.test_documents[0]
).count(), 0
)
def test_duplicates_after_trash(self):
document_2 = self.upload_document()
document_2.delete()
self.upload_document()
self.test_documents[1].delete()
self.assertFalse(document_2 in DuplicatedDocument.objects.get(document=self.document).documents.all())
self.assertFalse(
self.test_documents[1] in DuplicatedDocument.objects.get(
document=self.test_documents[0]
).documents.all()
)
def test_duplicate_scan(self):
document_2 = self.upload_document()
self.upload_document()
self.assertTrue(document_2 in DuplicatedDocument.objects.get(document=self.document).documents.all())
self.assertTrue(
self.test_documents[1] in DuplicatedDocument.objects.get(
document=self.test_documents[0]
).documents.all()
)
mayan/apps/documents/tests/test_search.py
+6 -6
View File
@@ -9,32 +9,32 @@ from mayan.apps.documents.tests import DocumentTestMixin
class DocumentSearchTestCase(DocumentTestMixin, BaseTestCase):
def _perform_document_page_search(self):
return document_page_search.search(
query_string={'q': self.document.label}, user=self._test_case_user
query_string={'q': self.test_document.label}, user=self._test_case_user
)
def _perform_document_search(self):
return document_search.search(
query_string={'q': self.document.label}, user=self._test_case_user
query_string={'q': self.test_document.label}, user=self._test_case_user
)
def test_document_page_search_no_access(self):
queryset = self._perform_document_page_search()
self.assertFalse(self.document.pages.first() in queryset)
self.assertFalse(self.test_document.pages.first() in queryset)
def test_document_page_search_with_access(self):
self.grant_access(
obj=self.test_document, permission=permission_document_view
)
queryset = self._perform_document_page_search()
self.assertTrue(self.document.pages.first() in queryset)
self.assertTrue(self.test_document.pages.first() in queryset)
def test_document_search_no_access(self):
queryset = self._perform_document_search()
self.assertFalse(self.document in queryset)
self.assertFalse(self.test_document in queryset)
def test_document_search_with_access(self):
self.grant_access(
obj=self.test_document, permission=permission_document_view
)
queryset = self._perform_document_search()
self.assertTrue(self.document in queryset)
self.assertTrue(self.test_document in queryset)
mayan/apps/documents/tests/test_widgets.py
+2 -2
View File
@@ -9,6 +9,6 @@ class DocumentPageWidgetTestCase(GenericDocumentTestCase):
def test_document_list_view_document_with_no_pages(self):
document_thumbnail_widget = DocumentPageThumbnailWidget()
self.test_document.pages.all().delete()
result = document_thumbnail_widget.render(instance=self.document)
result = document_thumbnail_widget.render(instance=self.test_document)
self.assertTrue(self.document.get_absolute_url() in result)
self.assertTrue(self.test_document.get_absolute_url() in result)