Fixed a bug where adding a section while others are open would cause a crash and wrote a test for it.

Author: kgaidis

FZAccordionTableView/FZAccordionTableView.m

@@ -150,6 +150,15 @@ - (void)setDataSource:(id<UITableViewDataSource>)dataSource {
     super.dataSource = self;
 }
 
+- (void)insertSections:(NSIndexSet *)sections withRowAnimation:(UITableViewRowAnimation)animation {
+    [sections enumerateIndexesUsingBlock:^(NSUInteger section, BOOL * _Nonnull stop) {
+       FZAccordionTableViewSectionInfo *sectionInfo = [[FZAccordionTableViewSectionInfo alloc] initWithNumberOfRows:0];
+        [self.sectionInfos insertObject:sectionInfo atIndex:section];
+    }];
+    
+    [super insertSections:sections withRowAnimation:animation];
+}
+
 - (void)deleteSections:(NSIndexSet *)sections withRowAnimation:(UITableViewRowAnimation)animation {
 
     // Remove section info in reverse order to prevent array from
@@ -431,7 +440,11 @@ - (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger
 
     [self.sectionInfos[section] setNumberOfRows:numOfRows];
 
-    return ([self isSectionOpen:section]) ? numOfRows : 0;
+    if (![self isSectionOpen:section]) {
+        numOfRows = 0;
+    }
+    
+    return numOfRows;
 }
 
 - (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath {

Tests/FZAccordionTableViewUnitTests/FZAccordionTableViewGeneralTests.m

@@ -203,6 +203,32 @@ - (void)testAddingSection {
     XCTAssert(initialSectionCount+1 == self.tableView.numberOfSections, @"The section counts should match up.");
 }
 
+- (void)testAddingSectionWhileOtherSectionsAreOpen {
+    
+    // First, open all of the sections
+    self.tableView.allowMultipleSectionsOpen = YES;
+    for (NSInteger i = 0; i < [self.tableView numberOfSections];  i++) {
+        [self waitForHeaderViewInSection:i];
+        [self.tableView toggleSection:i];
+        
+        XCTAssert([self.tableView isSectionOpen:i], @"Section %d should be open.", (int)i);
+    }
+    
+    NSInteger section = 1;
+    NSInteger numberOfRows = 33;
+    NSInteger initialSectionCount = self.mainViewController.sections.count;
+    
+    // Add data into data source
+    [self.mainViewController.sections insertObject:@(numberOfRows) atIndex:section];
+    
+    // Add to tableview
+    NSIndexSet *indexSet = [NSIndexSet indexSetWithIndex:section];
+    [self.tableView insertSections:indexSet withRowAnimation:UITableViewRowAnimationFade];
+    
+    XCTAssert([self.mainViewController.sections[section] integerValue] == [self.tableView.sectionInfos[section] numberOfRows], @"The data source section rows should match up.");
+    XCTAssert(initialSectionCount+1 == self.tableView.numberOfSections, @"The section counts should match up.");
+}
+
 #pragma mark - Delete Section -
 
 - (void)testDeleteClosedSection {