From 6442b4ff15cbf615400c0a093af0393b2272a22c Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Wed, 24 May 2017 21:45:25 +0100
Subject: Write thead and th for headers in table

---
 tests/conversion_tests.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

(limited to 'tests')

diff --git a/tests/conversion_tests.py b/tests/conversion_tests.py
index 3dfb9f3..3fd9e34 100644
--- a/tests/conversion_tests.py
+++ b/tests/conversion_tests.py
@@ -316,6 +316,35 @@ def docx_table_is_converted_to_table_in_html():
     assert_equal(expected_html, result.value)
 
 
+@istest
+def header_rows_are_wrapped_in_thead():
+    table = documents.table([
+        documents.table_row([documents.table_cell([])], is_header=True),
+        documents.table_row([documents.table_cell([])], is_header=True),
+        documents.table_row([documents.table_cell([])], is_header=False),
+    ])
+    result = convert_document_element_to_html(table)
+    expected_html = (
+        "<table>" +
+        "<thead><tr><th></th></tr><tr><th></th></tr></thead>" +
+        "<tbody><tr><td></td></tr></tbody>" +
+        "</table>")
+    assert_equal(expected_html, result.value)
+
+
+@istest
+def tbody_is_omitted_if_all_rows_are_headers():
+    table = documents.table([
+        documents.table_row([documents.table_cell([])], is_header=True),
+    ])
+    result = convert_document_element_to_html(table)
+    expected_html = (
+        "<table>" +
+        "<thead><tr><th></th></tr></thead>" +
+        "</table>")
+    assert_equal(expected_html, result.value)
+
+
 @istest
 def empty_cells_are_preserved_in_table():
     table = documents.table([
-- 
cgit v1.2.3