Add a metadata test mixin.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2017-08-31 02:17:22 -04:00
parent bcae72bb34
commit 104e779087
2 changed files with 22 additions and 5 deletions

View File

@@ -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
)

View File

@@ -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: