Process incomplete rows in header and footer. Add a test.

ars18wrw · ars18wrw · commit 8e00c5ff051d · 2017-11-28T12:28:21.000+03:00
DEVSIX-1597
diff --git a/src/main/java/com/itextpdf/html2pdf/attach/wrapelement/TableWrapper.java b/src/main/java/com/itextpdf/html2pdf/attach/wrapelement/TableWrapper.java
@@ -195,16 +195,22 @@ public Table toTable(WaitingColgroupsHelper colgroupsHelper) {
             table = new Table(1);
         }
         if (headerRows != null) {
-            for (List<CellWrapper> headerRow : headerRows) {
-                for (CellWrapper headerCell : headerRow) {
-                    table.addHeaderCell(headerCell.cell);
+            for (int i = 0; i < headerRows.size(); i++) {
+                for (int j = 0; j < headerRows.get(i).size(); j++) {
+                    table.addHeaderCell(headerRows.get(i).get(j).cell);
+                }
+                if (i != headerRows.size() - 1) {
+                    table.getHeader().startNewRow();
                 }
             }
         }
         if (footerRows != null) {
-            for (List<CellWrapper> footerRow : footerRows) {
-                for (CellWrapper footerCell : footerRow) {
-                    table.addFooterCell(footerCell.cell);
+            for (int i = 0; i < footerRows.size(); i++) {
+                for (int j = 0; j < footerRows.get(i).size(); j++) {
+                    table.addFooterCell(footerRows.get(i).get(j).cell);
+                }
+                if (i != footerRows.size() - 1) {
+                    table.getFooter().startNewRow();
                 }
             }
         }
diff --git a/src/test/java/com/itextpdf/html2pdf/element/TableTest.java b/src/test/java/com/itextpdf/html2pdf/element/TableTest.java
@@ -321,6 +321,11 @@ public void tableMaxHeightTest02() throws IOException, InterruptedException {
         runTest("tableMaxHeight02");
     }
 
+    @Test
+    public void multipleRowsInHeade01() throws IOException, InterruptedException {
+        runTest("multipleRowsInHeader01");
+    }
+
     @Test
     @Ignore("DEVSIX-994")
     public void tableCollapseColCellBoxSizingWidthDifference() throws IOException, InterruptedException {
diff --git a/src/test/resources/com/itextpdf/html2pdf/element/TableTest/cmp_multipleRowsInHeader01.pdf b/src/test/resources/com/itextpdf/html2pdf/element/TableTest/cmp_multipleRowsInHeader01.pdf
diff --git a/src/test/resources/com/itextpdf/html2pdf/element/TableTest/multipleRowsInHeader01.html b/src/test/resources/com/itextpdf/html2pdf/element/TableTest/multipleRowsInHeader01.html
@@ -0,0 +1,36 @@
+<html>
+<body>
+<table border="1">
+    <thead>
+    <tr>
+        <td>Header Row 1 Col 1</td>
+    </tr>
+    <tr>
+        <td>Header Row 2 Col 1</td>
+        <td>Header Row 2 Col 2</td>
+        <td>Header Row 2 Col 3</td>
+    </tr>
+    </thead>
+    <tbody>
+    <tr>
+        <td>Body Row 1 Col 1</td>
+    </tr>
+    <tr>
+        <td>Body Row 2 Col 1</td>
+        <td>Body Row 2 Col 2</td>
+        <td>Body Row 2 Col 3</td>
+    </tr>
+    </tbody>
+    <tfoot>
+    <tr>
+        <td>Footer Row 1 Col 1</td>
+    </tr>
+    <tr>
+        <td>Footer Row 2 Col 1</td>
+        <td>Footer Row 2 Col 2</td>
+        <td>Footer Row 2 Col 3</td>
+    </tr>
+    </tfoot>
+</table>
+</body>
+</html>

Original file line number	Diff line number	Diff line change
`@@ -195,16 +195,22 @@ public Table toTable(WaitingColgroupsHelper colgroupsHelper) {`
`195`	`195`	`table = new Table(1);`
`196`	`196`	`}`
`197`	`197`	`if (headerRows != null) {`
`198`		`- for (List<CellWrapper> headerRow : headerRows) {`
`199`		`- for (CellWrapper headerCell : headerRow) {`
`200`		`- table.addHeaderCell(headerCell.cell);`
	`198`	`+ for (int i = 0; i < headerRows.size(); i++) {`
	`199`	`+ for (int j = 0; j < headerRows.get(i).size(); j++) {`
	`200`	`+ table.addHeaderCell(headerRows.get(i).get(j).cell);`
	`201`	`+ }`
	`202`	`+ if (i != headerRows.size() - 1) {`
	`203`	`+ table.getHeader().startNewRow();`
`201`	`204`	`}`
`202`	`205`	`}`
`203`	`206`	`}`
`204`	`207`	`if (footerRows != null) {`
`205`		`- for (List<CellWrapper> footerRow : footerRows) {`
`206`		`- for (CellWrapper footerCell : footerRow) {`
`207`		`- table.addFooterCell(footerCell.cell);`
	`208`	`+ for (int i = 0; i < footerRows.size(); i++) {`
	`209`	`+ for (int j = 0; j < footerRows.get(i).size(); j++) {`
	`210`	`+ table.addFooterCell(footerRows.get(i).get(j).cell);`
	`211`	`+ }`
	`212`	`+ if (i != footerRows.size() - 1) {`
	`213`	`+ table.getFooter().startNewRow();`
`208`	`214`	`}`
`209`	`215`	`}`
`210`	`216`	`}`