Calculate the path of the test document before it is uploaded.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2018-09-21 00:47:35 -04:00
parent a96e7574b2
commit 7c4de79c8f

View File

@@ -25,19 +25,23 @@ class DocumentTestMixin(object):
)
def upload_document(self):
self._calculate_test_document_path()
with open(self.test_document_path, 'rb') as file_object:
document = self.document_type.new_document(
file_object=file_object, label=self.test_document_filename
)
return document
def setUp(self):
super(DocumentTestMixin, self).setUp()
def _calculate_test_document_path(self):
self.test_document_path = os.path.join(
settings.BASE_DIR, 'apps', 'documents', 'tests', 'contrib',
'sample_documents', self.test_document_filename
)
def setUp(self):
super(DocumentTestMixin, self).setUp()
if self.auto_create_document_type:
self._create_document_type()