Fix tests

Signed-off-by: Roberto Rosario <roberto.rosario@mayan-edms.com>
This commit is contained in:
Roberto Rosario
2019-10-09 16:55:02 -04:00
parent 7fbb94a8ae
commit 64abf66f22
3 changed files with 10 additions and 5 deletions

View File

@@ -68,8 +68,9 @@ class Parser(object):
self.process_document_page(document_page=document_page)
def process_document_page(self, document_page):
DocumentPageContent = apps.get_model(
app_label='document_parsing', model_name='DocumentPageContent'
DocumentVersionPageContent = apps.get_model(
app_label='document_parsing',
model_name='DocumentVersionPageContent'
)
logger.info(
@@ -80,7 +81,7 @@ class Parser(object):
file_object = document_page.document_version.get_intermediate_file()
try:
document_page_content, created = DocumentPageContent.objects.get_or_create(
document_page_content, created = DocumentVersionPageContent.objects.get_or_create(
document_page=document_page
)
document_page_content.content = self.execute(

View File

@@ -56,7 +56,9 @@ class TrashedDocumentTestMixin(object):
return self.get(viewname='documents:document_list_deleted')
class TrashedDocumentTestCase(GenericDocumentViewTestCase):
class TrashedDocumentTestCase(
TrashedDocumentTestMixin, GenericDocumentViewTestCase
):
def test_document_restore_get_view_no_permission(self):
self.test_document.delete()
self.assertEqual(Document.objects.count(), 0)

View File

@@ -46,7 +46,9 @@ class DocumentPageOCRContentManager(models.Manager):
)
)
cache_filename = task.get(timeout=DOCUMENT_IMAGE_TASK_TIMEOUT, disable_sync_subtasks=False)
cache_filename = task.get(
timeout=DOCUMENT_IMAGE_TASK_TIMEOUT, disable_sync_subtasks=False
)
with document_page.cache_partition.get_file(filename=cache_filename).open() as file_object:
document_page_content, created = DocumentPageOCRContent.objects.get_or_create(