From aeec0b1a145edda87e6a8598a0b40c0bbf06f65f Mon Sep 17 00:00:00 2001
From: Adlai Holler <guy.on.internet@me.com>
Date: Mon, 25 Jul 2016 17:28:24 -0700
Subject: [PATCH] [ASChangeSetDataController] Enable some assertions (#1984)

---
 AsyncDisplayKit/Details/ASChangeSetDataController.mm | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/AsyncDisplayKit/Details/ASChangeSetDataController.mm b/AsyncDisplayKit/Details/ASChangeSetDataController.mm
index 0660b87d49..b85aacf817 100644
--- a/AsyncDisplayKit/Details/ASChangeSetDataController.mm
+++ b/AsyncDisplayKit/Details/ASChangeSetDataController.mm
@@ -24,8 +24,7 @@
 
 - (void)beginUpdates
 {
-  // NOTE: This assertion is failing in some apps and will be enabled soon.
-//  ASDisplayNodeAssertMainThread();
+  ASDisplayNodeAssertMainThread();
   if (_changeSetBatchUpdateCounter <= 0) {
     _changeSetBatchUpdateCounter = 0;
     _changeSet = [[_ASHierarchyChangeSet alloc] initWithOldData:[self itemCountsFromDataSource]];
@@ -35,13 +34,11 @@
 
 - (void)endUpdatesAnimated:(BOOL)animated completion:(void (^)(BOOL))completion
 {
-  // NOTE: This assertion is failing in some apps and will be enabled soon.
-//  ASDisplayNodeAssertMainThread();
+  ASDisplayNodeAssertMainThread();
   _changeSetBatchUpdateCounter--;
   
   // Prevent calling endUpdatesAnimated:completion: in an unbalanced way
-  // NOTE: This assertion is failing in some apps and will be enabled soon.
-//  NSAssert(_changeSetBatchUpdateCounter >= 0, @"endUpdatesAnimated:completion: called without having a balanced beginUpdates call");
+  NSAssert(_changeSetBatchUpdateCounter >= 0, @"endUpdatesAnimated:completion: called without having a balanced beginUpdates call");
   
   if (_changeSetBatchUpdateCounter == 0) {
     if (!self.initialReloadDataHasBeenCalled) {