Rename Tree Enumeration Tests to Node Iterator Tests

2015-05-28 00:42:20 +02:00
parent 45156d98fd
commit aece7138a4
2 changed files with 25 additions and 26 deletions
@@ -0,0 +1,147 @@
+//
+//  HTMLKitNodeTreeEnumratorTests.m
+//  HTMLKit
+//
+//  Created by Iska on 28/03/15.
+//  Copyright (c) 2015 BrainCookie. All rights reserved.
+//
+
+#import <XCTest/XCTest.h>
+#import "HTMLDOM.h"
+
+@interface HTMLKitNodeIteratorTests : XCTestCase
+
+@end
+
+@implementation HTMLKitNodeIteratorTests
+
+#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.nodeIterator.allObjects;
+	NSArray	*expected = @[@"div"];
+	XCTAssertEqualObjects([result valueForKey:@"name"], expected);
+}
+
+- (void)testSimpleTree
+{
+	HTMLElement *tree = self.simpleTree;
+	NSArray *result = tree.nodeIterator.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.nodeIterator.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.nodeIterator.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