def test_view_object_with_custom_link(self):
customlink = CustomLink(
content_type=ContentType.objects.get_for_model(Site),
name="Test",
text="FOO {{ obj.name }} BAR",
target_url="http://example.com/?site={{ obj.slug }}",
new_window=False,
)
customlink.save()
site = Site(name="Test Site", slug="test-site")
site.save()
response = self.client.get(site.get_absolute_url(), follow=True)
self.assertEqual(response.status_code, 200)
self.assertIn(f"FOO {site.name} BAR", str(response.content))
def setUpTestData(cls):
obj_type = ContentType.objects.get_for_model(Site)
customlinks = (
CustomLink(
content_type=obj_type,
name="customlink-1",
text="customlink text 1",
target_url="http://customlink1.com",
weight=100,
button_class="default",
new_window=False,
),
CustomLink(
content_type=obj_type,
name="customlink-2",
text="customlink text 2",
target_url="http://customlink2.com",
weight=100,
button_class="default",
new_window=False,
),
CustomLink(
content_type=obj_type,
name="customlink-3",
text="customlink text 3",
target_url="http://customlink3.com",
weight=100,
button_class="default",
new_window=False,
),
)
for link in customlinks:
link.save()
cls.form_data = {
"content_type": obj_type.pk,
"name": "customlink-4",
"text": "customlink text 4",
"target_url": "http://customlink4.com",
"weight": 100,
"button_class": "default",
"new_window": False,
}