Write thead and th for headers in table

1 files changed, 8 insertions, 2 deletions

diff --git a/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java b/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
index 560eb54..a3595ae 100644
--- a/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
+++ b/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
@@ -21,6 +21,7 @@ public class DocumentElementMakers {
     private static final ArgumentKey<Boolean> STRIKETHROUGH = new ArgumentKey<>("strikethrough");
     private static final ArgumentKey<VerticalAlignment> VERTICAL_ALIGNMENT = new ArgumentKey<>("verticalAlignment");
     private static final ArgumentKey<List<DocumentElement>> CHILDREN = new ArgumentKey<>("children");
+    private static final ArgumentKey<Boolean> IS_HEADER = new ArgumentKey<>("isHeader");
     private static final ArgumentKey<Integer> COLSPAN = new ArgumentKey<>("colspan");
     private static final ArgumentKey<Integer> ROWSPAN = new ArgumentKey<>("rowspan");
     private static final ArgumentKey<List<Comment>> COMMENTS = new ArgumentKey<>("comments");
@@ -57,6 +58,10 @@ public class DocumentElementMakers {
         return arg(CHILDREN, asList(children));
     }
 
+    public static Argument<Boolean> withIsHeader(boolean isHeader) {
+        return arg(IS_HEADER, isHeader);
+    }
+
     public static Argument<Integer> withRowspan(Integer rowspan) {
         return arg(ROWSPAN, rowspan);
     }
@@ -108,8 +113,9 @@ public class DocumentElementMakers {
         return run(withChildren(new Text(text)));
     }
 
-    public static TableRow tableRow(List<DocumentElement> cells) {
-        return new TableRow(cells, false);
+    public static TableRow tableRow(List<DocumentElement> cells, Object... args) {
+        Arguments arguments = new Arguments(args);
+        return new TableRow(cells, arguments.get(IS_HEADER, false));
     }
 
     public static TableCell tableCell(Object... args) {