Swiftgram/submodules/AsyncDisplayKit/Tests/ASDisplayNodeExtrasTests.mm

//
//  ASDisplayNodeExtrasTests.mm
//  Texture
//
//  Copyright (c) Pinterest, Inc.  All rights reserved.
//  Licensed under Apache 2.0: http://www.apache.org/licenses/LICENSE-2.0
//

#import <XCTest/XCTest.h>
#import <AsyncDisplayKit/AsyncDisplayKit.h>
#import <AsyncDisplayKit/ASDisplayNodeExtras.h>

@interface ASDisplayNodeExtrasTests : XCTestCase

@end

@interface TestDisplayNode : ASDisplayNode
@end

@implementation TestDisplayNode
@end

@implementation ASDisplayNodeExtrasTests

- (void)testShallowFindSubnodesOfSubclass {
  ASDisplayNode *supernode = [[ASDisplayNode alloc] initWithLayerBlock:^CALayer * _Nonnull{
    return [CALayer layer];
  }];
  NSUInteger count = 10;
  NSMutableArray *expected = [[NSMutableArray alloc] initWithCapacity:count];
  for (NSUInteger nodeIndex = 0; nodeIndex < count; nodeIndex++) {
    TestDisplayNode *node = [[TestDisplayNode alloc] initWithLayerBlock:^CALayer * _Nonnull{
      return [CALayer layer];
    }];
    [supernode addSubnode:node];
    [expected addObject:node];
  }
  NSArray *found = ASDisplayNodeFindAllSubnodesOfClass(supernode, [TestDisplayNode class]);
  XCTAssertEqualObjects(found, expected, @"Expecting %lu %@ nodes, found %lu", (unsigned long)count, [TestDisplayNode class], (unsigned long)found.count);
}

- (void)testDeepFindSubnodesOfSubclass {
  ASDisplayNode *supernode = [[ASDisplayNode alloc] initWithLayerBlock:^CALayer * _Nonnull{
    return [CALayer layer];
  }];
  
  const NSUInteger count = 2;
  const NSUInteger levels = 2;
  const NSUInteger capacity = [[self class] capacityForCount:count levels:levels];
  NSMutableArray *expected = [[NSMutableArray alloc] initWithCapacity:capacity];
  
  [[self class] addSubnodesToNode:supernode number:count remainingLevels:levels accumulated:expected];
  
  NSArray *found = ASDisplayNodeFindAllSubnodesOfClass(supernode, [TestDisplayNode class]);
  XCTAssertEqualObjects(found, expected, @"Expecting %lu %@ nodes, found %lu", (unsigned long)count, [TestDisplayNode class], (unsigned long)found.count);
}

+ (void)addSubnodesToNode:(ASDisplayNode *)supernode number:(NSUInteger)number remainingLevels:(NSUInteger)level accumulated:(inout NSMutableArray *)expected {
  if (level == 0) return;
  for (NSUInteger nodeIndex = 0; nodeIndex < number; nodeIndex++) {
    TestDisplayNode *node = [[TestDisplayNode alloc] initWithLayerBlock:^CALayer * _Nonnull{
      return [CALayer layer];
    }];
    [supernode addSubnode:node];
    [expected addObject:node];
    [self addSubnodesToNode:node number:number remainingLevels:(level - 1) accumulated:expected];
  }
}

// Graph theory is failing me atm.
+ (NSUInteger)capacityForCount:(NSUInteger)count levels:(NSUInteger)level {
  if (level == 0) return 0;
  return pow(count, level) + [self capacityForCount:count levels:(level - 1)];
}

@end