Fix tests
Signed-off-by: Roberto Rosario <roberto.rosario@mayan-edms.com>
This commit is contained in:
@@ -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(
|
||||
|
||||
@@ -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)
|
||||
|
||||
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user