Collection.forceUpdate() calls inner CollectionView.forceUpdate()

Resolves #568
bvaughn · Feb 14, 2017 · 4fc6999 · 4fc6999
1 parent 77414b6
commit 4fc6999
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/source/Collection/Collection.jest.js b/source/Collection/Collection.jest.js
@@ -719,4 +719,19 @@ describe('Collection', () => {
       done()
     })
   })
+
+  // See issue #568 for more
+  it('forceUpdate will also forceUpdate the inner CollectionView', () => {
+    const cellRenderer = jest.fn()
+    cellRenderer.mockImplementation(({ key }) => <div key={key} />)
+
+    const rendered = render(getMarkup({ cellRenderer }))
+
+    expect(cellRenderer).toHaveBeenCalled()
+
+    cellRenderer.mockReset()
+    rendered.forceUpdate()
+
+    expect(cellRenderer).toHaveBeenCalled()
+  })
 })
diff --git a/source/Collection/Collection.js b/source/Collection/Collection.js
@@ -64,6 +64,12 @@ export default class Collection extends PureComponent {
     this._setCollectionViewRef = this._setCollectionViewRef.bind(this)
   }
 
+  forceUpdate () {
+    if (this._collectionView !== undefined) {
+      this._collectionView.forceUpdate()
+    }
+  }
+
   /** See Collection#recomputeCellSizesAndPositions */
   recomputeCellSizesAndPositions () {
     this._cellCache = []