From d6f00b6a57a19349de058ee17e1f963c0a972cf6 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 4 Nov 2017 11:45:06 +0000
Subject: Use table style mappings during document conversion

---
 .../mammoth/tests/conversion/DocumentToHtmlTests.java   | 17 +++++++++++++++++
 .../mammoth/tests/documents/DocumentElementMakers.java  |  5 +++--
 2 files changed, 20 insertions(+), 2 deletions(-)

(limited to 'src/test')

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
@@ -257,6 +257,23 @@ public class DocumentToHtmlTests {
                     Html.element("td", list(Html.FORCE_WRITE, Html.element("p", list(Html.text("Bottom right"))))))))))));
     }
 
+    @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(
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) {
-- 
cgit v1.2.3