Fix: Support cells.pop()

Author: BeyondEvil

src/pytest_html/basereport.py

@@ -246,8 +246,8 @@ def pytest_sessionstart(self, session):
 
         header_cells = Header()
         session.config.hook.pytest_html_results_table_header(cells=header_cells)
-
         self._report.set_data("resultsTableHeader", header_cells.html)
+        self._report.set_data("headerPops", header_cells.get_pops())
 
         self._report.set_data("runningState", "Started")
         self._generate_report()

src/pytest_html/scripts/main.js

@@ -44,6 +44,17 @@ const renderContent = (tests) => {
     table.querySelectorAll('.extra').forEach((item) => {
         item.colSpan = document.querySelectorAll('th').length
     })
+
+    const { headerPops } = manager.renderData
+    // remove 'headerPops' number of header columns
+    findAll('#results-table-head th').splice(-headerPops).forEach(column => column.remove())
+
+    // remove 'headerPops' number of row columns
+    const resultRows = findAll('.results-table-row')
+    resultRows.forEach((elem) => {
+        findAll('td:not(.extra)', elem).splice(-headerPops).forEach(column => column.remove())
+    })
+
     findAll('.sortable').forEach((elem) => {
         elem.addEventListener('click', (evt) => {
             const { target: element } = evt

src/pytest_html/scripts/sort.js

@@ -29,7 +29,6 @@ const doInitSort = () => {
     const ascending = storageModule.getSortDirection()
     const list = manager.testSubset
     const initialOrder = ['Error', 'Failed', 'Rerun', 'XFailed', 'XPassed', 'Skipped', 'Passed']
-    console.log(list)
     if (type?.toLowerCase() === 'original') {
         manager.setRender(list)
     } else {

src/pytest_html/table.py

@@ -33,6 +33,7 @@ class Cell(Table):
     def __init__(self):
         super().__init__()
         self._append_counter = 0
+        self._pop_counter = 0
         self._sortables = dict()
 
     def __setitem__(self, key, value):
@@ -69,10 +70,10 @@ def insert(self, index, html):
         self._html[index] = html
 
     def pop(self, *args):
-        warnings.warn(
-            "'pop' is deprecated and no longer supported.",
-            DeprecationWarning,
-        )
+        self._pop_counter += 1
+
+    def get_pops(self):
+        return self._pop_counter
 
     def _extract_sortable(self, html):
         match = re.search(r'<td class="col-(\w+)">(.*?)</', html)
@@ -90,3 +91,7 @@ class Row(Cell):
     def __delitem__(self, key):
         # This means the item should be removed
         self._html = None
+
+    def pop(self, *args):
+        # Calling pop on header is sufficient
+        pass

testing/test_integration.py

@@ -690,6 +690,27 @@ def test_pass(): pass
         page = run(pytester)
         assert_results(page, passed=1)
 
+    def test_results_table_hook_pop(self, pytester):
+        pytester.makeconftest(
+            """
+            def pytest_html_results_table_header(cells):
+                cells.pop()
+
+            def pytest_html_results_table_row(report, cells):
+                cells.pop()
+        """
+        )
+        pytester.makepyfile("def test_pass(): pass")
+        page = run(pytester)
+
+        header_columns = page.select(".summary #results-table-head th")
+        assert_that(header_columns).is_length(3)
+
+        row_columns = page.select_one(".summary .results-table-row").select(
+            "td:not(.extra)"
+        )
+        assert_that(row_columns).is_length(3)
+
     @pytest.mark.parametrize("no_capture", ["", "-s"])
     def test_standard_streams(self, pytester, no_capture):
         pytester.makepyfile(