#import <XCTest/XCTest.h>

#import "ASCollectionView.h"

#import "ASCollectionDataController.h"

#import "ASCollectionViewFlowLayoutInspector.h"

@interface ASCollectionViewTestDelegate : NSObject <ASCollectionViewDataSource, ASCollectionViewDelegate>

@property (nonatomic, assign) NSInteger numberOfSections;

@property (nonatomic, assign) NSInteger numberOfItemsInSection;

@implementation ASCollectionViewTestDelegate

- (id)initWithNumberOfSections:(NSInteger)numberOfSections numberOfItemsInSection:(NSInteger)numberOfItemsInSection {

if (self = [super init]) {

_numberOfSections = numberOfSections;

_numberOfItemsInSection = numberOfItemsInSection;

}

return self;

}

- (ASCellNode *)collectionView:(ASCollectionView *)collectionView nodeForItemAtIndexPath:(NSIndexPath *)indexPath {

ASTextCellNode *textCellNode = [ASTextCellNode new];

textCellNode.text = indexPath.description;

return textCellNode;

}

- (NSInteger)numberOfSectionsInCollectionView:(UICollectionView *)collectionView {

return self.numberOfSections;

}

- (NSInteger)collectionView:(UICollectionView *)collectionView numberOfItemsInSection:(NSInteger)section {

return self.numberOfItemsInSection;

}

@interface ASCollectionViewTestController: UIViewController

@property (nonatomic, strong) ASCollectionViewTestDelegate *asyncDelegate;

@property (nonatomic, strong) ASCollectionView *collectionView;

@implementation ASCollectionViewTestController

- (void)viewDidLoad {

[super viewDidLoad];

self.asyncDelegate = [[ASCollectionViewTestDelegate alloc] initWithNumberOfSections:10 numberOfItemsInSection:10];

self.collectionView = [[ASCollectionView alloc] initWithFrame:self.view.bounds

collectionViewLayout:[UICollectionViewFlowLayout new]];

self.collectionView.asyncDataSource = self.asyncDelegate;

self.collectionView.asyncDelegate = self.asyncDelegate;

[self.view addSubview:self.collectionView];

}

- (void)viewWillLayoutSubviews {

[super viewWillLayoutSubviews];

self.collectionView.frame = self.view.bounds;

}

@interface ASCollectionView (InternalTesting)

- (NSArray *)supplementaryNodeKindsInDataController:(ASCollectionDataController *)dataController;

@interface ASCollectionViewTests : XCTestCase

@implementation ASCollectionViewTests

- (void)testThatItSetsALayoutInspectorForFlowLayouts

{

UICollectionViewFlowLayout *layout = [[UICollectionViewFlowLayout alloc] init];

ASCollectionView *collectionView = [[ASCollectionView alloc] initWithFrame:CGRectZero collectionViewLayout:layout];

XCTAssert(collectionView.layoutInspector != nil, @"should automatically set a layout delegate for flow layouts");

XCTAssert([collectionView.layoutInspector isKindOfClass:[ASCollectionViewFlowLayoutInspector class]], @"should have a flow layout inspector by default");

}

- (void)testThatItDoesNotSetALayoutInspectorForCustomLayouts

{

UICollectionViewLayout *layout = [[UICollectionViewLayout alloc] init];

ASCollectionView *collectionView = [[ASCollectionView alloc] initWithFrame:CGRectZero collectionViewLayout:layout];

XCTAssert(collectionView.layoutInspector == nil, @"should not set a layout delegate for custom layouts");

}

- (void)testThatRegisteringASupplementaryNodeStoresItForIntrospection

{

UICollectionViewFlowLayout *layout = [[UICollectionViewFlowLayout alloc] init];

ASCollectionView *collectionView = [[ASCollectionView alloc] initWithFrame:CGRectZero collectionViewLayout:layout];

[collectionView registerSupplementaryNodeOfKind:UICollectionElementKindSectionHeader];

XCTAssertEqualObjects([collectionView supplementaryNodeKindsInDataController:nil], @[UICollectionElementKindSectionHeader]);

}

- (void)DISABLED_testCollectionViewController

{

ASCollectionViewTestController *testController = [[ASCollectionViewTestController alloc] initWithNibName:nil bundle:nil];

UIView *containerView = [[UIView alloc] initWithFrame:CGRectMake(0, 0, 100, 100)];

[containerView addSubview:testController.view];

[testController.collectionView reloadData];

[[NSRunLoop currentRunLoop] runUntilDate:[NSDate dateWithTimeIntervalSinceNow:1.0]];

}