From 86f88650f8428b32944a28407d5386da696cdad5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Agust=C3=ADn=20M=C3=A9ndez?= <matagus@gmail.com>
Date: Mon, 5 Feb 2024 19:47:28 -0600
Subject: [PATCH] Added test case for template tag get_links_for

---
 tests/test_templatetags.py | 63 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 tests/test_templatetags.py

diff --git a/tests/test_templatetags.py b/tests/test_templatetags.py
new file mode 100644
index 0000000..03cb81f
--- /dev/null
+++ b/tests/test_templatetags.py
@@ -0,0 +1,63 @@
+from django.contrib.auth.models import User
+from django.contrib.contenttypes.models import ContentType
+from django.template import Context
+from django.template import Template
+from django.test import TestCase
+
+from generic_links.models import GenericLink
+
+
+class TemplateTagTestCase(TestCase):
+    """
+    Test case for the template tag  `get_links_for <obj>` in the `generic_links_tags.py` file.
+    """
+
+    def setUp(self):
+        content_type = ContentType.objects.get_for_model(User)
+
+        self.user1 = User.objects.create(username="test_user")
+
+        self.link1 = GenericLink.objects.create(
+            url="https://example.com/path/to/this-thing/",
+            content_type=content_type,
+            object_id=self.user1.id,
+        )
+
+        self.link2 = GenericLink.objects.create(
+            url="https://example.com/path/to/other-thing/",
+            content_type=content_type,
+            object_id=self.user1.id,
+        )
+
+        self.user2 = User.objects.create(username="test_user2")
+
+        self.link3 = GenericLink.objects.create(
+            url="https://example.com/foobar/",
+            content_type=content_type,
+            object_id=self.user2.id,
+        )
+
+    def test_get_links_for(self):
+        """
+        Test the `get_links_for` template tag.
+        """
+
+        template_string = """
+            {% load generic_links_tags %}
+            {% get_links_for my_obj as links_qs %}
+            {% for link in links_qs %}
+                {{ link.url }}
+            {% endfor %}
+        """
+
+        template = Template(template_string)
+
+        rendered_string = template.render(Context({"my_obj": self.user1}))
+
+        self.assertIn(self.link1.url, rendered_string)
+        self.assertIn(self.link2.url, rendered_string)
+        self.assertNotIn(self.link3.url, rendered_string)
+
+        rendered_string2 = template.render(Context({"my_obj": self.user2}))
+
+        self.assertEqual(self.link3.url, rendered_string2.strip())