From 90680b0208bba24a970f8b7fd6f9092f5ca90407 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Mon, 2 May 2016 12:07:36 +0100
Subject: Read colspan for table cells

---
 .../zwobble/mammoth/tests/documents/DocumentElementMakers.java | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java')

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 3048acd..04d8aef 100644
--- a/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
+++ b/src/test/java/org/zwobble/mammoth/tests/documents/DocumentElementMakers.java
@@ -22,6 +22,7 @@ public class DocumentElementMakers {
     public static final Property<Run, Boolean> STRIKETHROUGH = newProperty();
     public static final Property<Run, VerticalAlignment> VERTICAL_ALIGNMENT = newProperty();
     public static final Property<HasChildren, List<DocumentElement>> CHILDREN = newProperty();
+    public static final Property<TableCell, Integer> COLSPAN = newProperty();
 
     public static final Instantiator<Paragraph> PARAGRAPH =
         propertyLookup -> new Paragraph(
@@ -39,6 +40,15 @@ public class DocumentElementMakers {
             propertyLookup.valueOf(STYLE, Optional.empty()),
             propertyLookup.valueOf(CHILDREN, list()));
 
+    public static final Instantiator<TableCell> TABLE_CELL =
+        propertyLookup -> new TableCell(
+            propertyLookup.valueOf(COLSPAN, 1),
+            propertyLookup.valueOf(CHILDREN, list()));
+
+    public static TableCell tableCell(List<DocumentElement> children) {
+        return make(a(TABLE_CELL, with(CHILDREN, children)));
+    }
+
     public static Paragraph paragraphWithText(String text) {
         return make(a(PARAGRAPH, with(CHILDREN, list(runWithText(text)))));
     }
-- 
cgit v1.2.3