//
//  HTMLKitNodeTreeEnumratorTests.m
//  HTMLKit
//
//  Created by Iska on 28/03/15.
//  Copyright (c) 2015 BrainCookie. All rights reserved.
//

#import <XCTest/XCTest.h>

#import "HTMLNodes.h"

@interface HTMLKitNodeTreeEnumratorTests : XCTestCase

@end

@implementation HTMLKitNodeTreeEnumratorTests

#pragma mark - Elements

- (HTMLElement *)div
{
	return [[HTMLElement alloc] initWithTagName:@"div"];
}

- (HTMLElement *)simpleTree
{
	/*
	 | div
	 |   a
	 |   b
	 |   c
	 */
	HTMLElement *div = self.div;
	[div appendNode:[[HTMLElement alloc] initWithTagName:@"a"]];
	[div appendNode:[[HTMLElement alloc] initWithTagName:@"b"]];
	[div appendNode:[[HTMLElement alloc] initWithTagName:@"c"]];
	return div;
}

- (HTMLElement *)nestedSimpleTree
{
	/*
	 | div
	 |   div
	 |     a
	 |     b
	 |     c
	 |   div
	 |     a
	 |     b
	 |     c
	 */
	HTMLElement *div = self.div;
	[div appendNode:self.simpleTree];
	[div appendNode:self.simpleTree];
	return div;
}

- (HTMLElement *)complexTree
{
	/*
	 | div
	 |   div
	 |     div
	 |       a
	 |       b
	 |       c
	 |   e
	 |     f
	 |   div
	 |     a
	 |     b
	 |     c
	 */
	HTMLElement *root = self.div;

	HTMLElement *div = self.div;
	[div appendNode:self.simpleTree];
	[root appendNode:div];

	HTMLElement *e = [[HTMLElement alloc] initWithTagName:@"e"];
	[e appendNode:[[HTMLElement alloc] initWithTagName:@"f"]];
	[root appendNode:e];
	[root appendNode:self.simpleTree];

	return root;
}

#pragma mark - Tests

- (void)testSingle
{
	HTMLElement *div = self.div;
	NSArray *result = div.treeEnumerator.allObjects;
	NSArray	*expected = @[@"div"];
	XCTAssertEqualObjects([result valueForKey:@"name"], expected);
}

- (void)testSimpleTree
{
	HTMLElement *tree = self.simpleTree;
	NSArray *result = tree.treeEnumerator.allObjects;
	NSArray	*expected = @[@"div", @"a", @"b", @"c"];
	XCTAssertEqualObjects([result valueForKey:@"name"], expected);
}

- (void)testSimpleTreeReversed
{
	HTMLElement *tree = self.simpleTree;
	NSArray *result = tree.reverseTreeEnumerator.allObjects;
	NSArray	*expected = @[@"div", @"c", @"b", @"a"];
	XCTAssertEqualObjects([result valueForKey:@"name"], expected);
}

- (void)testNestedSimpleTree
{
	HTMLElement *tree = self.nestedSimpleTree;
	NSArray *result = tree.treeEnumerator.allObjects;
	NSArray	*expected = @[@"div", @"div", @"a", @"b", @"c", @"div", @"a", @"b", @"c"];
	XCTAssertEqualObjects([result valueForKey:@"name"], expected);
}

- (void)testNestedSimpleTreeReversed
{
	HTMLElement *tree = self.nestedSimpleTree;
	NSArray *result = tree.reverseTreeEnumerator.allObjects;
	NSArray	*expected = @[@"div", @"div", @"c", @"b", @"a", @"div", @"c", @"b", @"a"];
	XCTAssertEqualObjects([result valueForKey:@"name"], expected);
}

- (void)testComplexSimpleTree
{
	HTMLElement *tree = self.complexTree;
	NSArray *result = tree.treeEnumerator.allObjects;
	NSArray	*expected = @[@"div", @"div",@"div", @"a", @"b", @"c", @"e", @"f", @"div", @"a", @"b", @"c"];
	XCTAssertEqualObjects([result valueForKey:@"name"], expected);
}

- (void)testComplexSimpleTreeReversed
{
	HTMLElement *tree = self.complexTree;
	NSArray *result = tree.reverseTreeEnumerator.allObjects;
	NSArray	*expected = @[@"div", @"div", @"c", @"b", @"a", @"e", @"f", @"div", @"div", @"c", @"b", @"a"];
	XCTAssertEqualObjects([result valueForKey:@"name"], expected);
}

@end