Fix tests
Signed-off-by: Roberto Rosario <roberto.rosario@mayan-edms.com>
This commit is contained in:
@@ -12,13 +12,34 @@ from ..permissions import (
|
||||
from .literals import TEST_DOCUMENT_CONTENT
|
||||
|
||||
|
||||
class OCRAPITestCase(DocumentTestMixin, BaseAPITestCase):
|
||||
class OCRAPIViewTestMixin(object):
|
||||
def _request_document_ocr_submit_view(self):
|
||||
return self.post(
|
||||
viewname='rest_api:document-ocr-submit-view',
|
||||
kwargs={'pk': self.test_document.pk}
|
||||
)
|
||||
|
||||
def _request_document_version_ocr_submit_view(self):
|
||||
return self.post(
|
||||
viewname='rest_api:document-version-ocr-submit-view', kwargs={
|
||||
'document_pk': self.test_document.pk,
|
||||
'version_pk': self.test_document.latest_version.pk
|
||||
}
|
||||
)
|
||||
|
||||
def _request_document_version_page_content_view(self):
|
||||
return self.get(
|
||||
viewname='rest_api:document-page-ocr-content-view', kwargs={
|
||||
'document_pk': self.test_document.pk,
|
||||
'version_pk': self.test_document.latest_version.pk,
|
||||
'page_pk': self.test_document.latest_version.pages.first().pk,
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
class OCRAPIViewTestCase(
|
||||
OCRAPIViewTestMixin, DocumentTestMixin, BaseAPITestCase
|
||||
):
|
||||
def test_submit_document_no_access(self):
|
||||
response = self._request_document_ocr_submit_view()
|
||||
self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
|
||||
@@ -35,15 +56,9 @@ class OCRAPITestCase(DocumentTestMixin, BaseAPITestCase):
|
||||
self.assertEqual(response.status_code, status.HTTP_202_ACCEPTED)
|
||||
|
||||
self.assertTrue(
|
||||
hasattr(self.test_document.pages.first(), 'ocr_content')
|
||||
)
|
||||
|
||||
def _request_document_version_ocr_submit_view(self):
|
||||
return self.post(
|
||||
viewname='rest_api:document-version-ocr-submit-view', kwargs={
|
||||
'document_pk': self.test_document.pk,
|
||||
'version_pk': self.test_document.latest_version.pk
|
||||
}
|
||||
hasattr(
|
||||
self.test_document.pages.first().content_type, 'ocr_content'
|
||||
)
|
||||
)
|
||||
|
||||
def test_submit_document_version_no_access(self):
|
||||
@@ -62,16 +77,7 @@ class OCRAPITestCase(DocumentTestMixin, BaseAPITestCase):
|
||||
self.assertEqual(response.status_code, status.HTTP_202_ACCEPTED)
|
||||
|
||||
self.assertTrue(
|
||||
hasattr(self.test_document.pages.first(), 'ocr_content')
|
||||
)
|
||||
|
||||
def _request_document_version_page_content_view(self):
|
||||
return self.get(
|
||||
viewname='rest_api:document-page-ocr-content-view', kwargs={
|
||||
'document_pk': self.test_document.pk,
|
||||
'version_pk': self.test_document.latest_version.pk,
|
||||
'page_pk': self.test_document.latest_version.pages.first().pk,
|
||||
}
|
||||
hasattr(self.test_document_version.pages.first(), 'ocr_content')
|
||||
)
|
||||
|
||||
def test_get_document_version_page_content_no_access(self):
|
||||
|
||||
Reference in New Issue
Block a user