From 9ac722cd96111afa9ec0522af4f6dbf95a62b7b4 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 20 Feb 2016 14:39:16 +0000
Subject: Add style mappings for paragraphs by style ID

---
 .../java/org/zwobble/mammoth/tests/DocumentConverterTests.java | 10 ++++++++++
 .../zwobble/mammoth/tests/documents/DocumentElementMakers.java |  3 ++-
 2 files changed, 12 insertions(+), 1 deletion(-)

(limited to 'src/test/java/org')

diff --git a/src/test/java/org/zwobble/mammoth/tests/DocumentConverterTests.java b/src/test/java/org/zwobble/mammoth/tests/DocumentConverterTests.java
index f888f6a..8e76619 100644
--- a/src/test/java/org/zwobble/mammoth/tests/DocumentConverterTests.java
+++ b/src/test/java/org/zwobble/mammoth/tests/DocumentConverterTests.java
@@ -51,6 +51,16 @@ public class DocumentConverterTests {
                 Html.element("p", list(Html.text("there"))))));
     }
 
+    @Test
+    public void styleMappingsUsingStyleIdsCanBeUsedToMapParagraphs() {
+        assertThat(
+            convertToHtml(
+                make(a(PARAGRAPH, with(STYLE, Optional.of(new Style("TipsParagraph", Optional.empty()))))),
+                StyleMap.builder().mapParagraph("TipsParagraph", new HtmlPath(list(HtmlPath.element("p", map("class", "tip"))))).build()),
+
+            deepEquals(list(Html.element("p", map("class", "tip")))));
+    }
+
     @Test
     public void boldRunsAreWrappedInStrongTags() {
         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 2fa5fea..1091ab8 100644
--- a/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
+++ b/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
@@ -14,6 +14,7 @@ import static com.natpryce.makeiteasy.Property.newProperty;
 import static org.zwobble.mammoth.util.MammothLists.list;
 
 public class DocumentElementMakers {
+    public static final Property<HasChildren, Optional<Style>> STYLE = newProperty();
     public static final Property<HasChildren, Boolean> BOLD = newProperty();
     public static final Property<HasChildren, Boolean> ITALIC = newProperty();
     public static final Property<HasChildren, Boolean> UNDERLINE = newProperty();
@@ -23,7 +24,7 @@ public class DocumentElementMakers {
 
     public static final Instantiator<Paragraph> PARAGRAPH =
         propertyLookup -> new Paragraph(
-            Optional.empty(),
+            propertyLookup.valueOf(STYLE, Optional.empty()),
             Optional.empty(),
             propertyLookup.valueOf(CHILDREN, list()));
 
-- 
cgit v1.2.3