Merge branch 'master' of github.com:manGoweb/SpecTools into swift4

Author: Ondrej Rafaj

README.md

@@ -200,7 +200,7 @@ Check if all UITableViewCells available through the UITableView data source eval
 ```Swift
 // Create an enumerate closure
 let doesFitClosure: (UITableViewCell)->Bool = { (cell) -> Bool in
-	guard let cell = cell as? TableViewCell else {
+	guard let cell = cell as? CustomTableViewCell else {
 		return false
 	}
 	if cell.customLabel.text?.characters.count == 0 {
@@ -218,6 +218,30 @@ You can also ask for an array of IndexPaths that don't fit the criteria
 let indexPaths: [IndexPath] = subject.tableView.spec.check.allCells(thatDontFit: doesFitClosure)
 ```
 
+#### Checking collection view cells
+
+Check if all UICollectionViewCell available through the UICollectionView data source evaluate correctly
+```Swift
+// Create an enumerate closure
+let doesFitClosure: (UICollectionViewCell)->Bool = { (cell) -> Bool in
+	guard let cell = cell as? CustomCollectionViewCell else {
+		return false
+	}
+	if cell.customLabel.text?.characters.count == 0 {
+		return false
+	}
+	return true
+}
+
+// Test if all cells generated by the data source are ok using your closure
+let ok = subject.collectionView.spec.check.allCells(fit: doesFitClosure)
+```
+
+You can also ask for an array of IndexPaths that don't fit the criteria
+```Swift
+let indexPaths: [IndexPath] = subject.collectionView.spec.check.allCells(thatDontFit: doesFitClosure)
+```
+
 #### UIViewController checks
 
 Look for a specific view controller in your navigation stack

SpecTools/Classes/Checks/Checks+UICollectionView.swift

@@ -0,0 +1,45 @@
+//
+//  Checks+UICollectionView.swift
+//  SpecTools
+//
+//  Created by Ondrej Rafaj on 15/09/2017.
+//
+
+import Foundation
+import UIKit
+
+
+extension Check where T: UICollectionView {
+    
+    // MARK: UICollectionView
+    
+    /// Check if there are any cells that don't fit criteria specified in a closure
+    /// - Parameter fit: Closure that needs to evaluate the cell which is passed onto it
+    /// - Returns: Bool (true if no issue is found)
+    public func allCells(fit evaluateClosure: (UICollectionViewCell)->Bool) -> Bool {
+        return allCells(thatDontFit: evaluateClosure).count == 0
+    }
+    
+    /// Check if there are any cells that don't fit criteria specified in a closure
+    /// - Parameter fit: Closure that needs to evaluate the cell which is passed onto it
+    /// - Returns: [IndexPath] Index path of all cells that do not match the given criteria
+    public func allCells(thatDontFit evaluateClosure: (UICollectionViewCell)->Bool) -> [IndexPath] {
+        var indexPaths: [IndexPath] = []
+        for sectionIndex in 0...element.spec.find.numberOfSections() {
+            row: for rowIndex in 0...element.spec.find.number(ofRowsIn: sectionIndex) {
+                let indexPath = IndexPath(item: rowIndex, section: sectionIndex)
+                guard let cell = element.dataSource?.collectionView(element, cellForItemAt: indexPath) else {
+                    fatalError("Data source is not set")
+                }
+                
+                let ok = evaluateClosure(cell)
+                if !ok {
+                    indexPaths.append(indexPath)
+                }
+            }
+        }
+        return indexPaths
+    }
+    
+}
+

SpecTools/Classes/Find/Find+UICollectionView.swift

@@ -0,0 +1,29 @@
+//
+//  Find+UICollectionView.swift
+//  SpecTools
+//
+//  Created by Ondrej Rafaj on 15/09/2017.
+//
+
+import Foundation
+import UIKit
+
+
+extension Find where T: UICollectionView {
+    
+    // MARK: UICollectionView
+    
+    /// Return number of sections on a collection view data source
+    /// - Returns: Int
+    public func numberOfSections() -> Int {
+        return element.dataSource?.numberOfSections?(in: element) ?? 0
+    }
+    
+    /// Number of rows on a section in a collection view data source
+    /// - Parameter ofRowsIn: Section index
+    /// - Returns: Int
+    public func number(ofRowsIn section: Int) -> Int {
+        return element.dataSource?.collectionView(element, numberOfItemsInSection: section) ?? 0
+    }
+    
+}