def testImageRecordsOnlySolrIndexing(self):
image_record = Record.objects.create()
image_record2 = Record.objects.create()
image_record.fieldvalue_set.create(
field=self.dcrelation, refinement='IsPartOf', value='SET1')
image_record2.fieldvalue_set.create(
field=self.dcrelation, refinement='IsPartOf', value='SET1')
identifiers = [image_record.id, image_record2.id]
index = SolrIndex()
work_to_images = index._preload_work_to_images(identifiers)
self.assertEqual(2, len(work_to_images))
self.assertEqual([image_record.id], work_to_images[image_record2.id])
self.assertEqual([image_record2.id], work_to_images[image_record.id])
image_to_works = index._preload_image_to_works(identifiers)
self.assertEqual(0, len(image_to_works))
def testSolrIndexing(self):
work_record = Record.objects.create()
image_record = Record.objects.create()
image_record2 = Record.objects.create()
work_record.fieldvalue_set.create(field=self.dcid, value='SOLR')
image_record.fieldvalue_set.create(
field=self.dcrelation, refinement='IsPartOf', value='SOLR')
image_record2.fieldvalue_set.create(
field=self.dcrelation, refinement='IsPartOf', value='SOLR')
identifiers = [work_record.id, image_record.id, image_record2.id]
index = SolrIndex()
work_to_images = index._preload_work_to_images(identifiers)
self.assertEqual(3, len(work_to_images))
w2i = work_to_images[work_record.id]
self.assertTrue(image_record.id in w2i)
self.assertTrue(image_record2.id in w2i)
image_to_works = index._preload_image_to_works(identifiers)
self.assertEqual(2, len(image_to_works))
self.assertEqual([work_record.id], image_to_works[image_record.id])
self.assertEqual([work_record.id], image_to_works[image_record2.id])
