From 26467b831a3e3a7dafc577932cdd93baee3804ae Mon Sep 17 00:00:00 2001
From: Elijah <elijahshellsanchez@outlook.com>
Date: Wed, 3 Jan 2024 14:39:42 +0100
Subject: [PATCH] Extended test coverage to admin Template tables

---
 idhub/tests/test_tables.py | 65 ++++++++++++++++++++++++++++++++++++--
 1 file changed, 63 insertions(+), 2 deletions(-)

diff --git a/idhub/tests/test_tables.py b/idhub/tests/test_tables.py
index 3aee017..81d2886 100644
--- a/idhub/tests/test_tables.py
+++ b/idhub/tests/test_tables.py
@@ -1,11 +1,13 @@
 from datetime import datetime
+from unittest.mock import MagicMock
 
 from django.test import TestCase
 from django.urls import reverse
+from django.core.exceptions import FieldError
 
 from idhub_auth.models import User
-from idhub.admin.tables import DashboardTable, UserTable
-from idhub.models import Event, Membership, Rol, UserRol, Service
+from idhub.admin.tables import DashboardTable, UserTable, TemplateTable
+from idhub.models import Event, Membership, Rol, UserRol, Service, Schemas
 
 
 class AdminDashboardTableTest(TestCase):
@@ -94,3 +96,62 @@ class UserTableTest(TestCase):
         # Use the render_role method of UserTable
         rendered_column = self.table.columns['role'].render(user)
         self.assertIn("Role 1", str(rendered_column))
+
+
+class TemplateTableTest(TestCase):
+
+    def setUp(self):
+        self.table = TemplateTable(Schemas.objects.all())
+        self.create_schemas(amount=3)
+
+    def create_schemas(self, amount):
+        for i in range(amount):
+            self.create_schemas_object("testname" + str(i), "testdesc" + str(i))
+
+    def create_schemas_object(self, name, description):
+        data = self.format_data_for_json_reader(name, description)
+        Schemas.objects.create(
+                type="testy",
+                file_schema="filey",
+                data=data,
+                created_at=datetime.now()
+        )
+
+    def format_data_for_json_reader(self, name, description):
+        return '{"name": "'+name+'", "description": "'+description+'"}'
+
+    def test_order_table_by_name_throws_no_exception(self):
+        try:
+            # Apply sorting
+            self.table.order_by = 'name'
+        except FieldError:
+            self.fail("Ordering template table by name raised FieldError")
+
+    def test_order_table_by_name_correctly_orders(self):
+        self.table.order_by = 'name'
+        # Fetch the sorted records
+        sorted_records = list(self.table.rows)
+
+        # Verify the order is as expected
+        self.assertTrue(sorted_records[0].record.name
+                        < sorted_records[1].record.name)
+        self.assertTrue(sorted_records[1].record.name
+                        < sorted_records[2].record.name)
+
+    def test_order_table_by_name_ascending_correctly_orders(self):
+        self.table.order_by = '-name'
+        # Fetch the sorted records
+        sorted_records = list(self.table.rows)
+
+        # Verify the order is as expected
+        self.assertTrue(sorted_records[0].record.name
+                        > sorted_records[1].record.name)
+        self.assertTrue(sorted_records[1].record.name
+                        > sorted_records[2].record.name)
+
+    def test_order_table_by_description_works(self):
+        try:
+            # Apply sorting
+            self.table.order_by = 'description'
+        except FieldError:
+            self.fail("Ordering template table by descripition raised FieldError")