From 280f754217772ba1ce6edfb06e6781d228f91cc7 Mon Sep 17 00:00:00 2001 From: Maurice Parker Date: Wed, 30 Oct 2019 20:38:57 -0500 Subject: [PATCH] Reload nodes who's unread counts change as that could change the cell layout --- iOS/MasterFeed/MasterFeedViewController.swift | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/iOS/MasterFeed/MasterFeedViewController.swift b/iOS/MasterFeed/MasterFeedViewController.swift index 20f92c253..d7232e61c 100644 --- a/iOS/MasterFeed/MasterFeedViewController.swift +++ b/iOS/MasterFeed/MasterFeedViewController.swift @@ -99,10 +99,8 @@ class MasterFeedViewController: UITableViewController, UndoableCommandRunner { node = coordinator.rootNode.descendantNodeRepresentingObject(representedObject as AnyObject) } - if let node = node, let indexPath = dataSource.indexPath(for: node), let unreadCountProvider = node.representedObject as? UnreadCountProvider { - if let cell = tableView.cellForRow(at: indexPath) as? MasterFeedTableViewCell { - cell.unreadCount = unreadCountProvider.unreadCount - } + if let node = node, dataSource.indexPath(for: node) != nil { + reloadNode(node) } }