PEP8 Cleanups.
This commit is contained in:
@@ -45,52 +45,93 @@ class IndexTestCase(TestCase):
|
||||
|
||||
# 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 }}'])
|
||||
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'])
|
||||
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)])
|
||||
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'])
|
||||
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)])
|
||||
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)), [''])
|
||||
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'])
|
||||
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'])
|
||||
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)), [''])
|
||||
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)
|
||||
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')
|
||||
self.document.metadata.create(
|
||||
metadata_type=metadata_type, value='0001'
|
||||
)
|
||||
|
||||
# Create empty index
|
||||
index = Index.objects.create(label='test')
|
||||
@@ -98,12 +139,21 @@ class IndexTestCase(TestCase):
|
||||
|
||||
# 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)])
|
||||
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 }}'])
|
||||
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()), [])
|
||||
@@ -113,4 +163,6 @@ class IndexTestCase(TestCase):
|
||||
|
||||
# Check that document is in instance node
|
||||
instance_node = IndexInstanceNode.objects.get(value='0001')
|
||||
self.assertQuerysetEqual(instance_node.documents.all(), [repr(self.document)])
|
||||
self.assertQuerysetEqual(
|
||||
instance_node.documents.all(), [repr(self.document)]
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user