From cd53ef495c6863d8fe616eec1488dbacd3c2fcd7 Mon Sep 17 00:00:00 2001
From: Philip Niedertscheider <dev@philpri.me>
Date: Wed, 23 Sep 2020 10:54:00 +0200
Subject: [PATCH] WIP: Changed table all row assertion

---
 Package.resolved                              | 25 ++++++++++++++++
 Shared/Examples/ExperimentFactory.swift       | 29 +++++-------------
 Source/API/Table/Sections/PDFTableRow.swift   |  4 +--
 Source/API/Table/Sections/PDFTableRows.swift  |  6 ++--
 .../API/Table/Sections/PDFTableRowSpec.swift  | 30 +++++++++++++++++++
 .../Sections/PDFTable_RowSubscriptsSpec.swift | 24 +++++++++++++++
 6 files changed, 92 insertions(+), 26 deletions(-)
 create mode 100644 Package.resolved
 create mode 100644 Tests/TPPDFTests/API/Table/Sections/PDFTableRowSpec.swift
 create mode 100644 Tests/TPPDFTests/API/Table/Sections/PDFTable_RowSubscriptsSpec.swift

diff --git a/Package.resolved b/Package.resolved
new file mode 100644
index 00000000..4a3a2b86
--- /dev/null
+++ b/Package.resolved
@@ -0,0 +1,25 @@
+{
+  "object": {
+    "pins": [
+      {
+        "package": "Nimble",
+        "repositoryURL": "https://github.com/Quick/Nimble",
+        "state": {
+          "branch": null,
+          "revision": "7a46a5fc86cb917f69e3daf79fcb045283d8f008",
+          "version": "8.1.2"
+        }
+      },
+      {
+        "package": "Quick",
+        "repositoryURL": "https://github.com/Quick/Quick",
+        "state": {
+          "branch": null,
+          "revision": "09b3becb37cb2163919a3842a4c5fa6ec7130792",
+          "version": "2.2.1"
+        }
+      }
+    ]
+  },
+  "version": 1
+}
diff --git a/Shared/Examples/ExperimentFactory.swift b/Shared/Examples/ExperimentFactory.swift
index 58234932..516af57d 100644
--- a/Shared/Examples/ExperimentFactory.swift
+++ b/Shared/Examples/ExperimentFactory.swift
@@ -13,30 +13,15 @@ class ExperimentFactory: ExampleFactory {
     func generateDocument() -> [PDFDocument] {
         let document = PDFDocument(format: .a4)
 
-        document.add(space: 100)
-        let table = PDFTable(rows: 50, columns: 4)
-        table.widths = [0.1, 0.3, 0.3, 0.3]
-        table.margin = 10
-        table.padding = 10
-        table.showHeadersOnEveryPage = false
-        table.shouldSplitCellsOnPageBreak = false
-        table.style.columnHeaderCount = 3
-
-        for row in 0..<table.size.rows {
-            table[row, 0].content = "\(row)".asTableContent
-            for column in 1..<table.size.columns {
-                table[row, column].content = "\(row),\(column)".asTableContent
-            }
-        }
-
+        let table = PDFTable(rows: 3, columns: 4)
+        table.content = [
+            ["0,0", "0,1", "0,2", "0,3"],
+            ["1,0", "1,1", "1,2", "1,3"],
+            ["2,0", "2,1", "2,2", "2,3"],
+        ]
+        table.rows.allRowsAlignment = [.left, .left, .right, .right]
         document.add(table: table)
 
-//        let singleCellTable = PDFTable(rows: 1, columns: 1)
-//        singleCellTable[0,0].content = (0...100).map(String.init)
-//            .joined(separator: "\n")
-//            .asTableContent
-//        document.add(table: singleCellTable)
-
         return [document]
     }
 }
diff --git a/Source/API/Table/Sections/PDFTableRow.swift b/Source/API/Table/Sections/PDFTableRow.swift
index 67ed9105..0f7e67da 100644
--- a/Source/API/Table/Sections/PDFTableRow.swift
+++ b/Source/API/Table/Sections/PDFTableRow.swift
@@ -67,7 +67,7 @@ public class PDFTableRow {
         }
         set {
             assert(newValue.count <= cells.count, "Can not access more cells than available")
-            for (idx, cell) in cells.enumerated() {
+            for (idx, cell) in cells.enumerated() where idx <= newValue.count - 1 {
                 cell.style = newValue[idx]
             }
         }
@@ -97,7 +97,7 @@ public class PDFTableRow {
         }
         set {
             assert(newValue.count <= cells.count, "Can not access more cells than available")
-            for (idx, cell) in cells.enumerated() {
+            for (idx, cell) in cells.enumerated() where idx <= newValue.count - 1 {
                 cell.alignment = newValue[idx]
             }
         }
diff --git a/Source/API/Table/Sections/PDFTableRows.swift b/Source/API/Table/Sections/PDFTableRows.swift
index 1f899483..de9c9009 100644
--- a/Source/API/Table/Sections/PDFTableRows.swift
+++ b/Source/API/Table/Sections/PDFTableRows.swift
@@ -140,8 +140,10 @@ public class PDFTableRows {
             fatalError("You cannot read from this object.")
         }
         set {
-            assert(newValue.count <= rows.count, "Can not access more rows than available")
-            rows.forEach { $0.alignment = newValue }
+            rows.forEach {
+                assert(newValue.count <= $0.cells.count, "Can not access more columns than available")
+                $0.alignment = newValue
+            }
         }
     }
 
diff --git a/Tests/TPPDFTests/API/Table/Sections/PDFTableRowSpec.swift b/Tests/TPPDFTests/API/Table/Sections/PDFTableRowSpec.swift
new file mode 100644
index 00000000..3e1512d0
--- /dev/null
+++ b/Tests/TPPDFTests/API/Table/Sections/PDFTableRowSpec.swift
@@ -0,0 +1,30 @@
+//
+//  PDFTableRowSpec.swift
+//  TPPDF
+//
+//  Created by Philip Niedertscheider on 23.09.20.
+//
+
+import Quick
+import Nimble
+@testable import TPPDF
+
+class PDFTableRowSpec: QuickSpec {
+
+    override func spec() {
+        describe("PDFTableRow") {
+            describe("alignment") {
+                it("should set alignment for each row the same") {
+                    let table = PDFTable(rows: 3, columns: 3)
+                    table.rows.allRowsAlignment = [.left, .center, .right]
+                    expect(table.rows.alignment) == [
+                        [.left, .center, .right],
+                        [.left, .center, .right],
+                        [.left, .center, .right]
+                    ]
+                }
+            }
+        }
+    }
+}
+
diff --git a/Tests/TPPDFTests/API/Table/Sections/PDFTable_RowSubscriptsSpec.swift b/Tests/TPPDFTests/API/Table/Sections/PDFTable_RowSubscriptsSpec.swift
new file mode 100644
index 00000000..bfc760f4
--- /dev/null
+++ b/Tests/TPPDFTests/API/Table/Sections/PDFTable_RowSubscriptsSpec.swift
@@ -0,0 +1,24 @@
+//
+//  PDFTable_RowSubscriptsSpec.swift
+//  TPPDF
+//
+//  Created by Philip Niedertscheider on 23.09.20.
+//
+
+import Quick
+import Nimble
+@testable import TPPDF
+
+class PDFTable_RowSubscriptsSpec: QuickSpec {
+
+    override func spec() {
+        describe("PDFTable") {
+            describe("Row Subscripts") {
+                it("should return row by index") {
+                    let table = PDFTable(rows: 3, columns: 3)
+                    expect(table[row: 1].cells) === table.cells[1]
+                }
+            }
+        }
+    }
+}