def get_object(self, request, object_id):
obj = super(ContentRepositoryAdmin, self).get_object(
request, object_id)
if obj is None: # pragma: no cover
return
if not obj.targets.exists():
obj.targets.add(PublishingTarget.get_default_target())
return obj
def test_get_object_side_effects(self):
"""
NOTE: What we're testing here is that for any content repository
we're always adding the default target if the Admin
receives a `content repository` object to display
"""
request = RequestFactory().get('/')
repo_admin = ContentRepositoryAdmin(ContentRepository, admin)
content_repo = ContentRepository.objects.create(name='the repo')
self.assertFalse(content_repo.targets.exists())
repo_admin.get_object(request, content_repo.pk)
[target] = content_repo.targets.all()
self.assertEqual(target, PublishingTarget.get_default_target())
def test_get_default_target(self):
self.assertFalse(PublishingTarget.objects.exists())
default_target = PublishingTarget.get_default_target()
self.assertEqual(default_target.name, settings.DEFAULT_TARGET_NAME)
self.assertTrue(PublishingTarget.objects.exists())
def test_has_add_permission(self):
target_admin = PublishingTargetAdmin(PublishingTarget, admin)
self.assertFalse(PublishingTarget.objects.exists())
self.assertFalse(target_admin.has_add_permission())
[target] = PublishingTarget.objects.all()
self.assertEqual(target, PublishingTarget.get_default_target())
def has_add_permission(self, *args, **kwargs):
PublishingTarget.get_default_target()
return False
