Split tests into tests for models, views and API, very good recommendation from Two Scoops of Django by @pydanny and @audreyr.

mayan/apps/ocr/test_models.py

@@ -0,0 +1,54 @@
+from __future__ import unicode_literals
+
+from django.core.files.base import File
+from django.test import TransactionTestCase
+
+from documents.models import Document, DocumentType
+from documents.tests import TEST_SMALL_DOCUMENT_PATH, TEST_DOCUMENT_TYPE
+
+
+class DocumentOCRTestCase(TransactionTestCase):
+    def setUp(self):
+        self.document_type = DocumentType.objects.create(label=TEST_DOCUMENT_TYPE)
+
+        ocr_settings = self.document_type.ocr_settings
+        ocr_settings.auto_ocr = False
+        ocr_settings.save()
+
+        with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
+            self.document = self.document_type.new_document(file_object=File(file_object))
+
+    def _test_ocr_language_issue_16(self, language, result):
+        """
+        Reusable OCR test for a specific language
+        """
+
+        # Clear the document's first page content and switch language
+        self.document.language = language
+        self.document.save()
+        first_page = self.document.pages.first()
+        first_page.content = ''
+        first_page.save()
+
+        # Queue document for OCR
+        self.document.submit_for_ocr()
+
+        # Make sure content was extracted
+        self.assertTrue(result in self.document.pages.first().ocr_content.content)
+
+    def test_ocr_language_backends_end(self):
+        self._test_ocr_language_issue_16('eng', 'Mayan EDMS')
+
+    # TODO: Add documents with for the corresponding language
+    # def test_ocr_language_backends_deu(self):
+    #    self._test_ocr_language_issue_16('deu', 'Mayan EDMS')
+
+    # def test_ocr_language_backends_spa(self):
+    #    self._test_ocr_language_issue_16('spa', 'Mayan EDMS')
+
+    # def test_ocr_language_backends_rus(self):
+    #    self._test_ocr_language_issue_16('rus', '')
+
+    def tearDown(self):
+        self.document.delete()
+        self.document_type.delete()