From 98761510dd21959394bd633135bf77d0b241cf93 Mon Sep 17 00:00:00 2001 From: Maurice Parker Date: Mon, 9 Sep 2019 10:06:13 -0500 Subject: [PATCH] Reload visible cells after expanding/collapsing folders since the folder row might not get updated --- iOS/MasterFeed/MasterFeedViewController.swift | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/iOS/MasterFeed/MasterFeedViewController.swift b/iOS/MasterFeed/MasterFeedViewController.swift index 7d2cb54ae..377ba873d 100644 --- a/iOS/MasterFeed/MasterFeedViewController.swift +++ b/iOS/MasterFeed/MasterFeedViewController.swift @@ -410,25 +410,33 @@ class MasterFeedViewController: UITableViewController, UndoableCommandRunner { @objc func expandSelectedRows(_ sender: Any?) { if let indexPath = coordinator.currentFeedIndexPath { coordinator.expandFolder(indexPath) - self.applyChanges(animate: true) + self.applyChanges(animate: true) { + self.reloadAllVisibleCells() + } } } @objc func collapseSelectedRows(_ sender: Any?) { if let indexPath = coordinator.currentFeedIndexPath { coordinator.collapseFolder(indexPath) - self.applyChanges(animate: true) + self.applyChanges(animate: true) { + self.reloadAllVisibleCells() + } } } @objc func expandAll(_ sender: Any?) { coordinator.expandAllSectionsAndFolders() - self.applyChanges(animate: true) + self.applyChanges(animate: true) { + self.reloadAllVisibleCells() + } } @objc func collapseAllExceptForGroupItems(_ sender: Any?) { coordinator.collapseAllFolders() - self.applyChanges(animate: true) + self.applyChanges(animate: true) { + self.reloadAllVisibleCells() + } } // MARK: API