Re-estructure tests to cleanup the apps' namespace.

mayan/apps/document_indexing/tests/test_models.py

@@ -0,0 +1,116 @@
+from __future__ import unicode_literals
+
+from django.core.files.base import File
+from django.test import TestCase
+
+from documents.models import DocumentType
+from documents.tests import TEST_SMALL_DOCUMENT_PATH, TEST_DOCUMENT_TYPE
+from metadata.models import MetadataType, DocumentTypeMetadataType
+
+from ..models import Index, IndexInstanceNode, IndexTemplateNode
+
+
+class IndexTestCase(TestCase):
+    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 tearDown(self):
+        for document_type in DocumentType.objects.all():
+            document_type.delete()
+
+    def test_indexing(self):
+        metadata_type = MetadataType.objects.create(name='test', label='test')
+        DocumentTypeMetadataType.objects.create(
+            document_type=self.document_type, metadata_type=metadata_type
+        )
+
+        # Create empty index
+        index = Index.objects.create(label='test')
+        self.assertQuerysetEqual(Index.objects.all(), [repr(index)])
+
+        # Add our document type to the new index
+        index.document_types.add(self.document_type)
+        self.assertQuerysetEqual(index.document_types.all(), [repr(self.document_type)])
+
+        # Create simple index template
+        root = index.template_root
+        index.node_templates.create(parent=root, expression='{{ document.metadata_value_of.test }}', link_documents=True)
+        self.assertEqual(list(IndexTemplateNode.objects.values_list('expression', flat=True)), ['', '{{ document.metadata_value_of.test }}'])
+
+        # Add document metadata value to trigger index node instance creation
+        self.document.metadata.create(metadata_type=metadata_type, value='0001')
+        self.assertEqual(list(IndexInstanceNode.objects.values_list('value', flat=True)), ['', '0001'])
+
+        # Check that document is in instance node
+        instance_node = IndexInstanceNode.objects.get(value='0001')
+        self.assertQuerysetEqual(instance_node.documents.all(), [repr(self.document)])
+
+        # Change document metadata value to trigger index node instance update
+        document_metadata = self.document.metadata.get(metadata_type=metadata_type)
+        document_metadata.value = '0002'
+        document_metadata.save()
+        self.assertEqual(list(IndexInstanceNode.objects.values_list('value', flat=True)), ['', '0002'])
+
+        # Check that document is in new instance node
+        instance_node = IndexInstanceNode.objects.get(value='0002')
+        self.assertQuerysetEqual(instance_node.documents.all(), [repr(self.document)])
+
+        # Check node instance is destoyed when no metadata is available
+        self.document.metadata.get(metadata_type=metadata_type).delete()
+        self.assertEqual(list(IndexInstanceNode.objects.values_list('value', flat=True)), [''])
+
+        # Add document metadata value again to trigger index node instance creation
+        self.document.metadata.create(metadata_type=metadata_type, value='0003')
+        self.assertEqual(list(IndexInstanceNode.objects.values_list('value', flat=True)), ['', '0003'])
+
+        # Check node instance is destroyed when no documents are contained
+        self.document.delete()
+
+        # Document is in trash, index structure should remain unchanged
+        self.assertEqual(list(IndexInstanceNode.objects.values_list('value', flat=True)), ['', '0003'])
+
+        # Document deleted from, index structure should update
+        self.document.delete()
+        self.assertEqual(list(IndexInstanceNode.objects.values_list('value', flat=True)), [''])
+
+    def test_rebuild_all_indexes(self):
+        # Add metadata type and connect to document type
+        metadata_type = MetadataType.objects.create(name='test', label='test')
+        DocumentTypeMetadataType.objects.create(document_type=self.document_type, metadata_type=metadata_type)
+
+        # Add document metadata value
+        self.document.metadata.create(metadata_type=metadata_type, value='0001')
+
+        # Create empty index
+        index = Index.objects.create(label='test')
+        self.assertEqual(list(Index.objects.all()), [index])
+
+        # Add our document type to the new index
+        index.document_types.add(self.document_type)
+        self.assertQuerysetEqual(index.document_types.all(), [repr(self.document_type)])
+
+        # Create simple index template
+        root = index.template_root
+        index.node_templates.create(parent=root, expression='{{ document.metadata_value_of.test }}', link_documents=True)
+        self.assertEqual(list(IndexTemplateNode.objects.values_list('expression', flat=True)), ['', '{{ document.metadata_value_of.test }}'])
+
+        # There should be no index instances
+        self.assertEqual(list(IndexInstanceNode.objects.all()), [])
+
+        # Rebuild all indexes
+        IndexInstanceNode.objects.rebuild_all_indexes()
+
+        # Check that document is in instance node
+        instance_node = IndexInstanceNode.objects.get(value='0001')
+        self.assertQuerysetEqual(instance_node.documents.all(), [repr(self.document)])