add tests and refactor

Author: muyexi

DemoTests/DemoTests.swift

@@ -39,4 +39,20 @@ class DemoTests: XCTestCase {
         XCTAssert(number == 0)
     }
 
+    func testIsHidden() {
+        viewController.set(cells: viewController.hideMeCell3, hidden: true)
+        viewController.reloadData(animated: true)
+        
+        let isHidden = viewController.isHidden(cell: viewController.hideMeCell3)
+        XCTAssert(isHidden)
+    }
+    
+    func testChangeHeight() {
+        viewController.set(cells: viewController.hideMeCell3, height: 90)
+        viewController.reloadData(animated: true)
+        
+        let height = viewController.hideMeCell3.frame.size.height
+        XCTAssert(height == 90)
+    }
+    
 }

Source/OriginalRow.swift

@@ -10,21 +10,6 @@ class OriginalRow {
         self.cell = cell
         self.originalIndexPath = originalIndexPath
     }
-
-    var hidden: Bool {
-        get {
-            return hiddenPlanned
-        }
-        set {
-            if !hiddenReal && newValue {
-                batchOperation = .delete
-            } else if hiddenReal && !newValue {
-                batchOperation = .insert
-            }
-            
-            hiddenPlanned = newValue
-        }
-    }
 
     var hiddenReal: Bool = false
 
@@ -39,9 +24,19 @@ class OriginalRow {
     var height: CGFloat = CGFloat.greatestFiniteMagnitude
 
     func update() {
-        if !hidden && batchOperation == .none {
+        if !hiddenPlanned && batchOperation == .none {
             batchOperation = .update
         }
     }
 
+    func set(hidden: Bool) {
+        if !hiddenReal && hidden {
+            batchOperation = .delete
+        } else if hiddenReal && !hidden {
+            batchOperation = .insert
+        }
+        
+        hiddenPlanned = hidden
+    }
+    
 }

Source/OriginalSection.swift

@@ -5,7 +5,7 @@ class OriginalSection {
     var rows: [OriginalRow] = []
 
     func numberOfVissibleRows() -> Int {
-        return rows.filter { !$0.hidden }.count
+        return rows.filter { !$0.hiddenPlanned }.count
     }
 
 }

Source/OriginalTable.swift

@@ -60,7 +60,7 @@ class OriginalTable {
 
     func vissibleOriginalRowWithIndexPath(_ indexPath: IndexPath) -> OriginalRow {
         let section = sections[indexPath.section]
-        let visibleRows = section.rows.filter { !$0.hidden }
+        let visibleRows = section.rows.filter { !$0.hiddenPlanned }
 
         return visibleRows[indexPath.row]
     }
@@ -76,7 +76,7 @@ class OriginalTable {
         var indexRow = originalRow.originalIndexPath.row
 
         let section = sections[indexSection]
-        indexRow = section.rows[0..<indexRow].filter { !$0.hidden }.count
+        indexRow = section.rows[0..<indexRow].filter { !$0.hiddenPlanned }.count
 
         return IndexPath(row: indexRow, section: indexSection)
     }

Source/StaticTableViewController.swift

@@ -26,7 +26,7 @@ open class StaticTableViewController: UITableViewController, OriginalTableConfig
     open func set(cells: UITableViewCell..., hidden: Bool) {
         cells.forEach { cell in
             let row = originalTable!.originalRowWithTableViewCell(cell)
-            row.hidden = hidden
+            row.set(hidden: hidden)
         }
     }
 
@@ -38,7 +38,7 @@ open class StaticTableViewController: UITableViewController, OriginalTableConfig
     }
 
     open func isHidden(cell: UITableViewCell) -> Bool {
-        return originalTable!.originalRowWithTableViewCell(cell).hidden
+        return originalTable!.originalRowWithTableViewCell(cell).hiddenReal
     }
 
     open func reloadData(animated: Bool) {