Create as_icon templatetag and tests for it

1 files changed, 28 insertions, 0 deletions

diff --git a/pydis_site/apps/resources/tests/test_as_icon.py b/pydis_site/apps/resources/tests/test_as_icon.py
new file mode 100644
index 00000000..5b33910d
--- /dev/null
+++ b/pydis_site/apps/resources/tests/test_as_icon.py
@@ -0,0 +1,28 @@
+from django.test import TestCase
+
+from pydis_site.apps.resources.templatetags import as_icon
+
+
+class TestAsIcon(TestCase):
+    """Tests for `as_icon` templatetag."""
+
+    def test_as_icon(self):
+        """Should return proper icon type class and icon class based on input."""
+        test_cases = [
+            {
+                "input": "regular/icon",
+                "output": "fas fa-icon",
+            },
+            {
+                "input": "branding/brand",
+                "output": "fab fa-brand",
+            },
+            {
+                "input": "fake/my-icon",
+                "output": "fas fa-my-icon",
+            }
+        ]
+
+        for case in test_cases:
+            with self.subTest(input=case["input"], output=case["output"]):
+                self.assertEqual(case["output"], as_icon(case["input"]))