diff --git a/mayan/apps/metadata/tests/mixins.py b/mayan/apps/metadata/tests/mixins.py new file mode 100644 index 0000000000..053b1cb0fb --- /dev/null +++ b/mayan/apps/metadata/tests/mixins.py @@ -0,0 +1,20 @@ +from __future__ import unicode_literals + +from common.tests import BaseTestCase + +from ..models import MetadataType, DocumentMetadata + +from .literals import ( + TEST_DEFAULT_VALUE, TEST_LOOKUP_TEMPLATE, TEST_INCORRECT_LOOKUP_VALUE, + TEST_CORRECT_LOOKUP_VALUE, TEST_DATE_VALIDATOR, TEST_DATE_PARSER, + TEST_METADATA_TYPE_NAME, TEST_METADATA_TYPE_LABEL, TEST_INVALID_DATE, + TEST_VALID_DATE, TEST_PARSED_VALID_DATE +) + + +class MetadataTypeMixin(object): + def setUp(self): + super(MetadataTypeMixin, self).setUp() + self.metadata_type = MetadataType.objects.create( + name=TEST_METADATA_TYPE_NAME, label=TEST_METADATA_TYPE_LABEL + ) diff --git a/mayan/apps/metadata/tests/test_models.py b/mayan/apps/metadata/tests/test_models.py index 7f538442ff..de722a006d 100644 --- a/mayan/apps/metadata/tests/test_models.py +++ b/mayan/apps/metadata/tests/test_models.py @@ -19,20 +19,17 @@ from .literals import ( TEST_METADATA_TYPE_NAME, TEST_METADATA_TYPE_LABEL, TEST_INVALID_DATE, TEST_VALID_DATE, TEST_PARSED_VALID_DATE ) +from .mixins import MetadataTypeMixin @override_settings(OCR_AUTO_OCR=False) -class MetadataTestCase(BaseTestCase): +class MetadataTestCase(MetadataTypeMixin, BaseTestCase): def setUp(self): super(MetadataTestCase, self).setUp() self.document_type = DocumentType.objects.create( label=TEST_DOCUMENT_TYPE_LABEL ) - self.metadata_type = MetadataType.objects.create( - name=TEST_METADATA_TYPE_NAME, label=TEST_METADATA_TYPE_LABEL - ) - self.document_type.metadata.create(metadata_type=self.metadata_type) with open(TEST_SMALL_DOCUMENT_PATH) as file_object: