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, }