Use table style mappings during document conversion

2 files changed, 20 insertions, 2 deletions

diff --git a/src/test/java/org/zwobble/mammoth/tests/conversion/DocumentToHtmlTests.java b/src/test/java/org/zwobble/mammoth/tests/conversion/DocumentToHtmlTests.java
index 7b12085..123bd8e 100644
--- a/src/test/java/org/zwobble/mammoth/tests/conversion/DocumentToHtmlTests.java
+++ b/src/test/java/org/zwobble/mammoth/tests/conversion/DocumentToHtmlTests.java
@@ -258,6 +258,23 @@ public class DocumentToHtmlTests {
     }
 
     @Test
+    public void tableStyleMappingsCanBeUsedToMapTables() {
+        assertThat(
+            convertToHtml(
+                table(list(), withStyle(new Style("TableNormal", Optional.of("Normal Table")))),
+                StyleMap.builder()
+                    .mapTable(
+                        TableMatcher.styleName("Normal Table"),
+                        HtmlPath.element("table", map("class", "normal-table"))
+                    )
+                    .build()
+            ),
+
+            deepEquals(list(Html.element("table", map("class", "normal-table"))))
+        );
+    }
+
+    @Test
     public void headerRowsAreWrappedInThead() {
         assertThat(
             convertToHtml(table(list(
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 0289bdb..f206640 100644
--- a/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
+++ b/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
@@ -122,8 +122,9 @@ public class DocumentElementMakers {
         return run(withChildren(new Text(text)));
     }
 
-    public static Table table(List<DocumentElement> rows) {
-        return new Table(Optional.empty(), rows);
+    public static Table table(List<DocumentElement> rows, Object... args) {
+        Arguments arguments = new Arguments(args);
+        return new Table(arguments.get(STYLE, Optional.empty()), rows);
     }
 
     public static TableRow tableRow(List<DocumentElement> cells, Object... args) {