Fix failing document parsing tests.

Signed-off-by: Michael Price <loneviking72@gmail.com>
This commit is contained in:
Michael Price
2018-03-14 03:37:58 -04:00
committed by Roberto Rosario
parent ed4f97cea7
commit ed8cdb9da4
2 changed files with 5 additions and 6 deletions

View File

@@ -1,21 +1,20 @@
from __future__ import unicode_literals
from django.test import override_settings
from documents.tests import (
GenericDocumentViewTestCase, TEST_DOCUMENT_FILENAME
GenericDocumentViewTestCase, TEST_DOCUMENT_FILENAME,
TEST_DOCUMENT_PATH
)
from ..permissions import permission_content_view
from ..utils import get_document_content
@override_settings(OCR_AUTO_OCR=True)
class DocumentContentViewsTestCase(GenericDocumentViewTestCase):
_skip_file_descriptor_test = True
# Ensure we use a PDF file
test_document_filename = TEST_DOCUMENT_FILENAME
test_document_path = TEST_DOCUMENT_PATH
def setUp(self):
super(DocumentContentViewsTestCase, self).setUp()
@@ -33,7 +32,6 @@ class DocumentContentViewsTestCase(GenericDocumentViewTestCase):
def test_document_content_view_with_permission(self):
self.grant_permission(permission=permission_content_view)
response = self._document_content_view()
self.assertContains(

View File

@@ -17,6 +17,7 @@ from .literals import (
@override_settings(OCR_AUTO_OCR=False)
class GenericDocumentViewTestCase(GenericViewTestCase):
test_document_filename = TEST_SMALL_DOCUMENT_FILENAME
test_document_path = TEST_SMALL_DOCUMENT_PATH
def setUp(self):
super(GenericDocumentViewTestCase, self).setUp()
@@ -24,7 +25,7 @@ class GenericDocumentViewTestCase(GenericViewTestCase):
label=TEST_DOCUMENT_TYPE_LABEL
)
with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
with open(self.test_document_path) as file_object:
self.document = self.document_type.new_document(
file_object=file_object, label=self.test_document_filename
)