Merge pull request #43 from jeantimex/simplify-heightForRowAt

Simplify heightForRowAt.

Author: jeantimex

CollapsibleTableSectionViewController.podspec

@@ -16,7 +16,7 @@ Pod::Spec.new do |s|
   #
 
   s.name         = "CollapsibleTableSectionViewController"
-  s.version      = "1.0.0"
+  s.version      = "1.0.1"
   s.summary      = "Swift 3.0 library to support collapsible sections in a table view."
 
   # This description is used to generate tags and improve search results.

CollapsibleTableSectionViewController/CollapsibleTableSectionViewController.swift

@@ -109,8 +109,7 @@ extension CollapsibleTableSectionViewController: UITableViewDataSource, UITableV
     }
 
     public func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
-        let section = (indexPath as NSIndexPath).section
-        return isSectionCollapsed(section) ? 0 : UITableViewAutomaticDimension
+        return UITableViewAutomaticDimension
     }
 
     public func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {

CollapsibleTableSectionViewControllerTests/CollapsibleTableSectionViewControllerTests.swift

@@ -150,29 +150,7 @@ class CollapsibleTableSectionViewControllerTests: XCTestCase {
     // Test heightForRowAt
     //
 
-    func testReturnsZeroWhenSectionIsCollapsed() {
-        class MockCollapsibleTableSectionViewController: CollapsibleTableSectionViewController {
-            override public func isSectionCollapsed(_ section: Int) -> Bool {
-                return true
-            }
-        }
-        
-        viewController = MockCollapsibleTableSectionViewController()
-        
-        let heightForRowAt = viewController.tableView(tableView, heightForRowAt: IndexPath(row: 0, section: 0))
-        
-        XCTAssertEqual(heightForRowAt, 0)
-    }
-    
     func testReturnsUITableViewAutomaticDimension() {
-        class MockCollapsibleTableSectionViewController: CollapsibleTableSectionViewController {
-            override public func isSectionCollapsed(_ section: Int) -> Bool {
-                return false
-            }
-        }
-        
-        viewController = MockCollapsibleTableSectionViewController()
-        
         let heightForRowAt = viewController.tableView(tableView, heightForRowAt: IndexPath(row: 0, section: 0))
 
         XCTAssertEqual(heightForRowAt, UITableViewAutomaticDimension)