Feat: add tests for UICollectionView contentSize

Oni-zerone · Jul 6, 2019 · d598b59 · d598b59
1 parent 100ef49
commit d598b59
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/Example/Tests/StoriesLayout/StoriesCollectionViewLayoutSpec.swift b/Example/Tests/StoriesLayout/StoriesCollectionViewLayoutSpec.swift
@@ -48,6 +48,13 @@ class StoriesCollectionViewLayoutSpec: QuickSpec {
                 collectionView.reloadData()
             }
 
+            it("calculates contentSize") {
+                let numberOfItems = CGFloat(collectionView.numberOfItems(inSection: 0))
+                let contentSize = sut.collectionViewContentSize
+                expect(contentSize.height).to(equal(collectionView.bounds.height))
+                expect(contentSize.width).to(equal(collectionView.bounds.width * numberOfItems))
+            }
+
             it("should always invalidate layout") {
                 let shouldInvalidate = sut.shouldInvalidateLayout(forBoundsChange: collectionView.bounds)
                 expect(shouldInvalidate).to(beTrue())