Fixed the test failures.

Author: Vinodh Swamy

VSCollectionKit/CollectionKitTestApp/AlbumsCollectionController.swift

@@ -22,7 +22,8 @@ class AlbumsCollectionController: VSCollectionViewController {
 
     override func willAddSectionControllers() {
         super.willAddSectionControllers()
-        sectionHandler.addSectionHandler(handler: PhotosSectionHandler())
+        let photSectionHandler = PhotosSectionHandler()
+        sectionHandler.addSectionHandler(handler: photSectionHandler)
     }
 
     override func viewDidLoad() {

VSCollectionKit/VSCollectionKit/VSCollectionViewController/VSCollectionViewDataSource.swift

@@ -10,9 +10,9 @@ import UIKit
 
 public class VSCollectionViewDataSource: NSObject, UICollectionViewDataSource {
 
-    private var collectionView: UICollectionView
+    unowned private var collectionView: UICollectionView
     private var data: VSCollectionViewData?
-    private var sectionHandler: VSCollectionViewSectionHandller
+    unowned private var sectionHandler: VSCollectionViewSectionHandller
     public init(collectionView: UICollectionView,
          sectionHandler: VSCollectionViewSectionHandller) {
         self.collectionView = collectionView
@@ -68,7 +68,7 @@ public class VSCollectionViewDataSource: NSObject, UICollectionViewDataSource {
 
     private func handleCollectionView(collectionData: VSCollectionViewData) {
 
-        collectionView.performBatchUpdates({
+        collectionView.performBatchUpdates({  
 
             data = collectionData
             let updates = collectionData.update.updates

VSCollectionKit/VSCollectionKit/VSCollectionViewController/VSCollectionViewDelegate.swift

@@ -10,9 +10,9 @@ import UIKit
 
 public class VSCollectionViewDelegate: NSObject, UICollectionViewDelegate {
 
-    private var collectionView: UICollectionView
+    unowned private var collectionView: UICollectionView
     public var data: VSCollectionViewData?
-    private var sectionHandler: VSCollectionViewSectionHandller
+    unowned private var sectionHandler: VSCollectionViewSectionHandller
 
     public init(collectionView: UICollectionView,
          sectionHandler: VSCollectionViewSectionHandller) {

VSCollectionKit/VSCollectionKit/VSCollectionViewController/VSCollectionViewLayoutProvider.swift

@@ -10,8 +10,8 @@ import UIKit
 
 public class VSCollectionViewLayoutProvider {
 
-    private var collectionView: UICollectionView
-    private var sectionHandler: VSCollectionViewSectionHandller
+    unowned private var collectionView: UICollectionView
+    unowned private var sectionHandler: VSCollectionViewSectionHandller
     public var data: VSCollectionViewData?
 
     public init(collectionView: UICollectionView,

VSCollectionKit/VSCollectionKit/VSCollectionViewController/VSCollectionViewSectionHandler.swift

@@ -18,8 +18,8 @@ public class VSCollectionViewSectionHandller {
         sectionHandlers[handler.type] = handler
     }
 
-    public func removeSectionHandler(handler: SectionHandler) {
-        sectionHandlers[handler.type] = nil
+    public func removeSectionHandler(type: String) {
+        sectionHandlers[type] = nil
     }
 
     public func registerCells(for collectionView: UICollectionView) {

VSCollectionKit/VSCollectionKit/VSCollectionViewController/VSSectionHandlerProtocol.swift

@@ -37,13 +37,6 @@ public protocol SectionLayoutInfo: AnyObject {
                                _ environment: NSCollectionLayoutEnvironment) -> NSCollectionLayoutSection?
 }
 
-//public extension SectionLayoutInfo  {
-//    func sectionLayoutProvider(_ sectionModel: SectionModel,
-//                               _ environment: NSCollectionLayoutEnvironment) -> NSCollectionLayoutSection? {
-//        return nil
-//    }
-//}
-
 public protocol SectionDelegateHandler: AnyObject {
     func didSelect(_ collectionView: UICollectionView,
                    _ indexPath: IndexPath,

VSCollectionKit/VSCollectionKitTests/VSCollectionViewLayoutProviderTests.swift

@@ -31,7 +31,8 @@ class VSCollectionViewLayoutProviderTests: XCTestCase {
     func testExample() {
         let layoutProvider = VSCollectionViewLayoutProvider(collectionView: collectionView,
                                                             sectionHandler: mockSectionHandler)
-        XCTAssertNotNil(layoutProvider.collectionLayout(for: MockSectionModel(sectionType: "MockSection", sectionName: "Mock Section Name"),
+        layoutProvider.data = mockCollectionViewData()
+        XCTAssertNotNil(layoutProvider.collectionLayout(for: 0,
                                                         environment: MockLayoutEnvironment()))
     }