Tests: Fix failing tests

Fix failing tests in the OCR and parsing apps.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2018-11-27 17:17:07 -04:00
parent 67e79d0e19
commit 2ca38c20b0
2 changed files with 7 additions and 16 deletions

View File

@@ -15,6 +15,7 @@ class DocumentAutoParsingTestCase(GenericDocumentTestCase):
def test_disable_auto_parsing(self): def test_disable_auto_parsing(self):
self._create_document_type() self._create_document_type()
self.document = self.upload_document() self.document = self.upload_document()
with self.assertRaises(StopIteration): with self.assertRaises(StopIteration):
next(self.document.latest_version.content()) next(self.document.latest_version.content())
@@ -23,5 +24,5 @@ class DocumentAutoParsingTestCase(GenericDocumentTestCase):
self._create_document_type() self._create_document_type()
self.document = self.upload_document() self.document = self.upload_document()
self.assertTrue( self.assertTrue(
TEST_DOCUMENT_CONTENT in next(self.document.content()) TEST_DOCUMENT_CONTENT in self.document.content
) )

View File

@@ -75,9 +75,7 @@ class OCRViewsTestCase(GenericDocumentViewTestCase):
def test_document_submit_view_no_permission(self): def test_document_submit_view_no_permission(self):
self._request_document_submit_view() self._request_document_submit_view()
self.assertEqual( self.assertEqual(self.document.ocr_content, '')
''.join(self.document.latest_version.ocr_content()), ''
)
def test_document_submit_view_with_access(self): def test_document_submit_view_with_access(self):
self.grant_access( self.grant_access(
@@ -85,9 +83,7 @@ class OCRViewsTestCase(GenericDocumentViewTestCase):
) )
self._request_document_submit_view() self._request_document_submit_view()
self.assertTrue( self.assertTrue(
TEST_DOCUMENT_CONTENT in ''.join( TEST_DOCUMENT_CONTENT in self.document.ocr_content
self.document.latest_version.ocr_content()
)
) )
def _request_multiple_document_submit_view(self): def _request_multiple_document_submit_view(self):
@@ -100,9 +96,7 @@ class OCRViewsTestCase(GenericDocumentViewTestCase):
def test_multiple_document_submit_view_no_permission(self): def test_multiple_document_submit_view_no_permission(self):
self._request_multiple_document_submit_view() self._request_multiple_document_submit_view()
self.assertEqual( self.assertEqual(self.document.ocr_content, '')
''.join(self.document.latest_version.ocr_content()), ''
)
def test_multiple_document_submit_view_with_access(self): def test_multiple_document_submit_view_with_access(self):
self.grant_access( self.grant_access(
@@ -110,9 +104,7 @@ class OCRViewsTestCase(GenericDocumentViewTestCase):
) )
self._request_multiple_document_submit_view() self._request_multiple_document_submit_view()
self.assertTrue( self.assertTrue(
TEST_DOCUMENT_CONTENT in ''.join( TEST_DOCUMENT_CONTENT in self.document.ocr_content
self.document.latest_version.ocr_content()
)
) )
def _request_document_ocr_download_view(self): def _request_document_ocr_download_view(self):
@@ -135,9 +127,7 @@ class OCRViewsTestCase(GenericDocumentViewTestCase):
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assert_download_response( self.assert_download_response(
response=response, content=( response=response, content=self.document.ocr_content
''.join(get_document_ocr_content(document=self.document))
),
) )
def _request_document_type_ocr_settings_view(self): def _request_document_type_ocr_settings_view(self):