51 lines
2.2 KiB
Python
51 lines
2.2 KiB
Python
from __future__ import absolute_import
|
|
|
|
from json import loads
|
|
import os
|
|
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import User
|
|
from django.core.files.base import File
|
|
from django.core.urlresolvers import reverse
|
|
from django.test.client import Client
|
|
from django.test import TestCase
|
|
|
|
from rest_framework import status
|
|
from rest_framework.test import APIClient
|
|
|
|
from documents.models import Document, DocumentType
|
|
from documents.tests import (TEST_SMALL_DOCUMENT_PATH, TEST_DOCUMENT_DESCRIPTION,
|
|
TEST_DOCUMENT_TYPE)
|
|
from metadata.models import (MetadataType, DocumentMetadata,
|
|
DocumentTypeMetadataType)
|
|
from .models import Index, IndexInstanceNode, IndexTemplateNode
|
|
|
|
|
|
class IndexTestCase(TestCase):
|
|
def setUp(self):
|
|
self.document_type = DocumentType.objects.create(name=TEST_DOCUMENT_DESCRIPTION)
|
|
|
|
with open(TEST_SMALL_DOCUMENT_PATH) as file_object:
|
|
self.document = Document.objects.new_document(file_object=File(file_object), document_type=self.document_type)[0].document
|
|
|
|
def test_indexing(self):
|
|
metadata_type = MetadataType.objects.create(name='test', title='test')
|
|
DocumentTypeMetadataType.objects.create(document_type=self.document_type, metadata_type=metadata_type)
|
|
|
|
# Create empty index
|
|
index = Index.objects.create(name='test', title='test')
|
|
self.failUnlessEqual(list(Index.objects.all()), [index])
|
|
|
|
# Add our document type to the new index
|
|
index.document_types.add(self.document_type)
|
|
self.failUnlessEqual(list(index.document_types.all()), [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.failUnlessEqual(list(IndexTemplateNode.objects.values_list('expression', flat=True)), [u'', u'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.failUnlessEqual(list(IndexInstanceNode.objects.values_list('value', flat=True)), [u'', u'0001'])
|