def test_unlock_link_only_present_for_draft_versions(self):
draft_version = factories.PollVersionFactory(
created_by=self.user_author)
published_version = Version.objects.create(
content=factories.PollContentFactory(
poll=draft_version.content.poll),
created_by=factories.UserFactory(),
state=constants.PUBLISHED)
draft_unlock_url = self.get_admin_url(
self.versionable.version_model_proxy, 'unlock', draft_version.pk)
draft_unlock_control = render_to_string(
'djangocms_version_locking/admin/unlock_icon.html',
{'unlock_url': draft_unlock_url})
published_unlock_url = self.get_admin_url(
self.versionable.version_model_proxy, 'unlock',
published_version.pk)
published_unlock_control = render_to_string(
'djangocms_version_locking/admin/unlock_icon.html',
{'unlock_url': published_unlock_url})
changelist_url = version_list_url(draft_version.content)
with self.login_user_context(self.superuser):
response = self.client.post(changelist_url)
# The draft version unlock control exists
self.assertContains(response, draft_unlock_control, html=True)
# The published version exists
self.assertNotContains(response, published_unlock_control, html=True)
def test_archive_link_not_present_for_published_version(self):
draft_version = factories.PollVersionFactory(
created_by=self.user_author)
published_version = Version.objects.create(
content=factories.PollContentFactory(
poll=draft_version.content.poll),
created_by=factories.UserFactory(),
state=constants.PUBLISHED)
with self.login_user_context(self.user_author):
request = RequestFactory()
request.user = self.user_author
discard_url = self.version_admin._get_discard_link(
published_version, request)
self.assertEqual("", discard_url)
with self.login_user_context(self.user_author):
request = RequestFactory()
request.user = self.user_author
discard_url = self.version_admin._get_discard_link(
published_version, request)
self.assertEqual("", discard_url)