def test_cannot_merge_one(self):
"""
Should raise a RuntimeError if less than two :class:`.ConceptEntity`
instances are passed in the QuerySet.
"""
ConceptEntity.objects.create(name='entity only')
entities = ConceptEntity.objects.all()
with self.assertRaises(RuntimeError):
operations.merge_conceptentities(entities)
def test_cannot_merge_with_two_concepts(self):
"""
"""
for i in xrange(5):
c = ConceptEntity.objects.create(name='entity %i' % i)
if i == 3 or i == 1:
c.concept = Concept.objects.create(uri='http://f%i.ake' % i)
c.save()
entities = ConceptEntity.objects.all()
with self.assertRaises(RuntimeError):
operations.merge_conceptentities(entities)
def entity_merge(request):
"""
User can merge selected entities.
"""
entity_ids = request.GET.getlist('entity', [])
if len(entity_ids) <= 1:
raise ValueError('')
qs = ConceptEntity.objects.filter(pk__in=entity_ids)
qs = auth.apply_filter(ResourceAuthorization.EDIT, request.user, qs)
# q = auth.get_auth_filter('merge_conceptentities', request.user)
if qs.count() < 2:
# TODO: make this pretty and informative.
return HttpResponseForbidden('Only the owner can do that')
if request.GET.get('confirm', False) == 'true':
master_id = request.GET.get('master', None)
if master_id is not None:
master = operations.merge_conceptentities(qs,
master_id,
user=request.user)
return HttpResponseRedirect(
reverse('entity-details', args=(master.id, )))
context = {
'entities': qs,
}
template = 'entity_merge.html'
return render(request, template, context)
def test_merge_N(self):
"""
Only one :class:`.ConceptEntity` should remain from the original
QuerySet.
"""
N = 5 # This should pass for aribitrary N > 1.
for i in xrange(N):
# c = Concept.
ConceptEntity.objects.create(name='entity %i' % i)
entities = ConceptEntity.objects.all()
count_before = ConceptEntity.objects.all().count()
operations.merge_conceptentities(entities)
count_after = ConceptEntity.objects.all().count()
self.assertEqual(count_before - (N - 1), count_after,
"merge_conceptentities() should delete all but one"
" of the original ConceptEntity instances.")
def test_merge_N(self):
"""
Only one :class:`.ConceptEntity` should remain from the original
QuerySet.
"""
N = 5 # This should pass for aribitrary N > 1.
for i in xrange(N):
# c = Concept.
ConceptEntity.objects.create(name='entity %i' % i)
entities = ConceptEntity.objects.all()
count_before = ConceptEntity.objects.all().count()
operations.merge_conceptentities(
entities, user=User.objects.create(username='******'))
count_after = ConceptEntity.objects.all().count()
self.assertEqual(count_before, count_after,
"No ConceptEntity instances should be deleted.")
def test_merge_with_concept(self):
"""
The remaining :class:`.ConceptEntity` should be associated with
the one :class:`.Concept` in the queryset.
"""
uri = 'http://f.ake'
for i in xrange(5):
c = ConceptEntity.objects.create(name='entity %i' % i)
if i == 3:
c.concept.add(Concept.objects.create(uri=uri))
c.save()
entities = ConceptEntity.objects.all()
master = operations.merge_conceptentities(
entities, user=User.objects.create(username='******'))
self.assertIn(uri, master.concept.values_list('uri', flat=True))
def test_merge_two(self):
"""
Only one :class:`.ConceptEntity` should remain.
"""
for i in xrange(2):
ConceptEntity.objects.create(name='entity %i' % i)
entities = ConceptEntity.objects.all()
count_before = ConceptEntity.objects.all().count()
master = operations.merge_conceptentities(entities)
count_after = ConceptEntity.objects.all().count()
self.assertEqual(count_before - 1, count_after,
"merge_conceptentities() should delete all but one"
" of the original ConceptEntity instances.")
self.assertIsInstance(master, ConceptEntity,
"merge_conceptentities should return a"
" ConceptEntity instance")
def test_merge_updates_relations(self):
"""
:class:`.Relation` instances that point to merged
:class:`.ConceptEntity` instances should accrue to the "master"
:class:`.ConceptEntity`\.
"""
N = 5
ce1 = ConceptEntity.objects.create(name='the first one')
ce2 = ConceptEntity.objects.create(name='the second one')
predicate = Field.objects.get_or_create(name='predicate')[0]
for i in xrange(N):
Relation.objects.create(source=ce1, predicate=predicate, target=ce2)
entities = ConceptEntity.objects.all()
master = operations.merge_conceptentities(entities)
self.assertEqual(master.relations_from.count(), N)
self.assertEqual(master.relations_to.count(), N)
def test_merge_with_concept(self):
"""
The remaining :class:`.ConceptEntity` should be associated with
the one :class:`.Concept` in the queryset.
"""
uri = 'http://f.ake'
for i in xrange(5):
c = ConceptEntity.objects.create(name='entity %i' % i)
if i == 3:
c.concept = Concept.objects.create(uri=uri)
c.save()
entities = ConceptEntity.objects.all()
master = operations.merge_conceptentities(entities)
self.assertEqual(master.concept.uri, uri,
"The concept for the master ConceptEntity was not"
" set correctly.")
def test_merge_two(self):
"""
Only one :class:`.ConceptEntity` should remain.
"""
for i in xrange(2):
ConceptEntity.objects.create(name='entity %i' % i)
entities = ConceptEntity.objects.all()
count_before = ConceptEntity.objects.all().count()
master = operations.merge_conceptentities(
entities, user=User.objects.create(username='******'))
count_after = ConceptEntity.objects.all().count()
self.assertEqual(count_before, count_after,
"No ConceptEntity instances should be deleted.")
self.assertIsInstance(
master, ConceptEntity, "merge_conceptentities should return a"
" ConceptEntity instance")
self.assertEqual(Identity.objects.count(), 1,
"Should create a new Identity")
def entity_merge(request):
entity_ids = request.GET.getlist('entity', [])
if len(entity_ids) <= 1:
raise ValueError('')
qs = ConceptEntity.objects.filter(pk__in=entity_ids)
qs = authorization.apply_filter(request.user, 'change_conceptentity', qs)
# q = authorization.get_auth_filter('merge_conceptentities', request.user)
if qs.count() < 2:
# TODO: make this pretty and informative.
return HttpResponseForbidden('Only the owner can do that')
if request.GET.get('confirm', False) == 'true':
master_id = request.GET.get('master', None)
master = operations.merge_conceptentities(qs, master_id)
return HttpResponseRedirect(
reverse('entity-details', args=(master.id, )))
context = RequestContext(request, {
'entities': qs,
})
template = loader.get_template('entity_merge.html')
return HttpResponse(template.render(context))
