TST: test_output/{test_html[_headerless]}: check that htmlescape escapes

westurner · westurner · commit 65a34648b914 · 2019-12-10T17:22:55.000-05:00
diff --git a/test/test_output.py b/test/test_output.py
@@ -897,22 +897,27 @@ def test_moinmoin_headerless():
     assert_equal(expected, result)
 
 
+_test_table_html_headers = ["<strings>", "<&numbers&>"]
+_test_table_html = [["spam >", 41.9999], ["eggs &", 451.0]]
+assert_equal.__self__.maxDiff = None
+
 def test_html():
     "Output: html with headers"
     expected = "\n".join(
         [
             "<table>",
             "<thead>",
-            '<tr><th>strings  </th><th style="text-align: right;">  numbers</th></tr>',
+            '<tr><th>&lt;strings&gt;  </th><th style="text-align: right;">  &lt;&amp;numbers&amp;&gt;</th></tr>',
             "</thead>",
             "<tbody>",
-            '<tr><td>spam     </td><td style="text-align: right;">  41.9999</td></tr>',
-            '<tr><td>eggs     </td><td style="text-align: right;"> 451     </td></tr>',
+            '<tr><td>spam &gt;     </td><td style="text-align: right;">      41.9999</td></tr>',
+            '<tr><td>eggs &amp;     </td><td style="text-align: right;">     451     </td></tr>',
             "</tbody>",
             "</table>",
         ]
     )
-    result = tabulate(_test_table, _test_table_headers, tablefmt="html")
+    result = tabulate(_test_table_html, _test_table_html_headers,
+            tablefmt="html")
     assert_equal(expected, result)
 
 
@@ -922,13 +927,13 @@ def test_html_headerless():
         [
             "<table>",
             "<tbody>",
-            '<tr><td>spam</td><td style="text-align: right;"> 41.9999</td></tr>',
-            '<tr><td>eggs</td><td style="text-align: right;">451     </td></tr>',
+            '<tr><td>spam &gt;</td><td style="text-align: right;"> 41.9999</td></tr>',
+            '<tr><td>eggs &amp;</td><td style="text-align: right;">451     </td></tr>',
             "</tbody>",
             "</table>",
         ]
     )
-    result = tabulate(_test_table, tablefmt="html")
+    result = tabulate(_test_table_html, tablefmt="html")
     assert_equal(expected, result)
 
 
