Test: added test for using cells-sort table class and the default whi…

…ch sort table rows.
LeeWannacott · May 22, 2023 · 7e44cfe · 7e44cfe
1 parent c56953d
commit 7e44cfe
Show file tree

Hide file tree

Showing 2 changed files with 66 additions and 5 deletions.
diff --git a/test/table.js b/test/table.js
@@ -7,7 +7,7 @@ const tableSortJs = require("../public/table-sort");
 function createTestTable(
   testTableData,
   thAttributes = { classTags: "", colspan: "" },
-  props = { colsToClick: [], invisibleIndex: [], tableTags: "" }
+  props = { colsToClick: [], invisibleIndex: [], tableTags: "", trClasses: "" }
 ) {
   const numberOfTableColumns = Object.keys(testTableData).length;
   let testTableHeaders = "";
@@ -44,7 +44,15 @@ function createTestTable(
     ) {
       testTableTdRows.push(`<tr style="display: none;">${testTableTdRow}</tr>`);
     } else {
-      testTableTdRows.push(`<tr> ${testTableTdRow}</tr>`);
+      if (props.tableTags === "cells-sort" || props.tableTags === "tr-sort") {
+        testTableTdRows.push(
+          `<tr class="${props.trClasses}-${i}"> ${testTableTdRow}</tr>`
+        );
+      } else {
+        testTableTdRows.push(
+          `<tr class="${props.trClasses}"> ${testTableTdRow}</tr>`
+        );
+      }
     }
   }
 
@@ -97,9 +105,13 @@ function createTestTable(
   for (let [i, tr] of tableRows.entries()) {
     if (tr.style.display !== "none") {
       for (let i = 0; i < numberOfTableColumns; i++)
-        testIfSortedList[`col${i}`].push(
-          tr.querySelectorAll("td").item(i).innerHTML
-        );
+        if (props.tableTags === "cells-sort" || props.tableTags ==="tr-sort") {
+          testIfSortedList[`col${i}`].push(tr.outerHTML);
+        } else {
+          testIfSortedList[`col${i}`].push(
+            tr.querySelectorAll("td").item(i).innerHTML
+          );
+        }
     }
   }
   return testIfSortedList;

diff --git a/test/table.test.js b/test/table.test.js
@@ -552,3 +552,52 @@ test("Sort all combination of negative and positive integers and decimal numbers
     ],
   });
 });
+
+test("default behavior without cells-sort (tr's move when sorted)", () => {
+  expect(
+    createTestTable(
+      {
+        col0: ["8", "2", "3", "4", "5", "6", "0", "1", "7"],
+      },
+      { classTags: "" },
+      // note this isn't an actual class just used for testing purposes
+      { tableTags: "tr-sort", trClasses: "test" }
+    )
+  ).toStrictEqual({
+    col0: [
+      '<tr class="test-6"> <td>0</td></tr>',
+      '<tr class="test-7"> <td>1</td></tr>',
+      '<tr class="test-1"> <td>2</td></tr>',
+      '<tr class="test-2"> <td>3</td></tr>',
+      '<tr class="test-3"> <td>4</td></tr>',
+      '<tr class="test-4"> <td>5</td></tr>',
+      '<tr class="test-5"> <td>6</td></tr>',
+      '<tr class="test-8"> <td>7</td></tr>',
+      '<tr class="test-0"> <td>8</td></tr>',
+    ],
+  });
+});
+
+test("cells-sort table class (tr's stay in place, but td's are sorted)", () => {
+  expect(
+    createTestTable(
+      {
+        col0: ["8", "2", "3", "4", "5", "6", "0", "1", "7"],
+      },
+      { classTags: "" },
+      { tableTags: "cells-sort", trClasses: "test" }
+    )
+  ).toStrictEqual({
+    col0: [
+      '<tr class="test-0"> <td>0</td></tr>',
+      '<tr class="test-1"> <td>1</td></tr>',
+      '<tr class="test-2"> <td>2</td></tr>',
+      '<tr class="test-3"> <td>3</td></tr>',
+      '<tr class="test-4"> <td>4</td></tr>',
+      '<tr class="test-5"> <td>5</td></tr>',
+      '<tr class="test-6"> <td>6</td></tr>',
+      '<tr class="test-7"> <td>7</td></tr>',
+      '<tr class="test-8"> <td>8</td></tr>',
+    ],
+  });
+});