diff --git a/mayan/apps/documents/tests/literals.py b/mayan/apps/documents/tests/literals.py index 5b4f3edcf6..b4ed159c7f 100644 --- a/mayan/apps/documents/tests/literals.py +++ b/mayan/apps/documents/tests/literals.py @@ -34,6 +34,7 @@ TEST_DOCUMENT_TYPE_2_LABEL = 'test document type 2' TEST_DOCUMENT_TYPE_LABEL_EDITED = 'test document type edited label' TEST_DOCUMENT_TYPE_QUICK_LABEL = 'test quick label' TEST_DOCUMENT_TYPE_QUICK_LABEL_EDITED = 'test quick label edited' +TEST_DOCUMENT_TYPE_RANDOM_LABEL_LENGTH = 16 TEST_DOCUMENT_VERSION_COMMENT_EDITED = 'test document version comment edited' TEST_HYBRID_DOCUMENT = 'hybrid_text_and_image.pdf' TEST_MULTI_PAGE_TIFF = 'multi_page.tiff' diff --git a/mayan/apps/documents/tests/mixins.py b/mayan/apps/documents/tests/mixins.py index 52d423578d..f85d5b9c96 100644 --- a/mayan/apps/documents/tests/mixins.py +++ b/mayan/apps/documents/tests/mixins.py @@ -1,6 +1,8 @@ from __future__ import unicode_literals import os +import random +import string import time from django.conf import settings @@ -9,7 +11,7 @@ from ..models import DocumentType from .literals import ( TEST_DOCUMENT_TYPE_LABEL, TEST_DOCUMENT_TYPE_QUICK_LABEL, - TEST_SMALL_DOCUMENT_FILENAME + TEST_DOCUMENT_TYPE_RANDOM_LABEL_LENGTH, TEST_SMALL_DOCUMENT_FILENAME ) __all__ = ('DocumentTestMixin',) @@ -22,10 +24,25 @@ class DocumentTestMixin(object): test_document_path = None use_document_stub = False + @staticmethod + def _get_random_alphanumeric_string(length): + return ''.join( + [random.choice(string.ascii_letters + string.digits) for _ in range(length)] + ) + def _create_document_type(self): self.document_type = DocumentType.objects.create( label=TEST_DOCUMENT_TYPE_LABEL ) + self.test_document_type = self.document_type + + def _create_document_type_random(self): + self.document_type = DocumentType.objects.create( + label=DocumentTestMixin._get_random_alphanumeric_string( + length=TEST_DOCUMENT_TYPE_RANDOM_LABEL_LENGTH + ) + ) + self.test_document_type = self.document_type def _create_document(self, *args, **kwargs): """