matthias/mayan-edms
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update OCR app tests to work with Python 3.

Browse Source 
Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2018-09-06 16:59:38 -04:00
parent e41f2f73e7
commit 4ae7a32443
2 changed files with 5 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
mayan/apps/ocr/tests/test_api.py
View File

@@ -4,8 +4,7 @@ from django.test import override_settings
from rest_framework import status
from documents.models import DocumentType
from documents.tests import TEST_DOCUMENT_TYPE_LABEL, TEST_SMALL_DOCUMENT_PATH
from documents.tests import DocumentTestMixin
from rest_api.tests import BaseAPITestCase
from ..permissions import (
@@ -15,25 +14,13 @@ from ..permissions import (
@override_settings(OCR_AUTO_OCR=False)
@override_settings(DOCUMENT_PARSING_PDFTOTEXT_PATH='')
class OCRAPITestCase(BaseAPITestCase):
class OCRAPITestCase(DocumentTestMixin, BaseAPITestCase):
"""
Test the OCR app API endpoints
"""
def setUp(self):
super(OCRAPITestCase, self).setUp()
self.login_user()
self.document_type = DocumentType.objects.create(
label=TEST_DOCUMENT_TYPE_LABEL
)
with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
self.document = self.document_type.new_document(
file_object=file_object,
)
def tearDown(self):
self.document_type.delete()
super(OCRAPITestCase, self).tearDown()
def _request_document_ocr_submit_view(self):
return self.post(