Add tests for the Nth-Expression selectors
This commit is contained in:
iska
@@ -0,0 +1,352 @@
|
||||
//
|
||||
// CSSNThExpressionSelectorTests.m
|
||||
// HTMLKit
|
||||
//
|
||||
// Created by Iska on 21/10/15.
|
||||
// Copyright © 2015 BrainCookie. All rights reserved.
|
||||
//
|
||||
|
||||
#import <XCTest/XCTest.h>
|
||||
#import "CSSSelectors.h"
|
||||
#import "HTMLParser.h"
|
||||
#import "HTMLDOM.h"
|
||||
|
||||
@interface CSSNThExpressionSelectorTests : XCTestCase
|
||||
{
|
||||
HTMLDocument *_childTree;
|
||||
HTMLDocument *_typeTree;
|
||||
}
|
||||
@end
|
||||
|
||||
@implementation CSSNThExpressionSelectorTests
|
||||
|
||||
- (void)setUp
|
||||
{
|
||||
[super setUp];
|
||||
|
||||
/*
|
||||
| <div>
|
||||
| <table>
|
||||
| <tr id='11'><td>11</td></tr>
|
||||
| <tr id='12'><td>12</td></tr>
|
||||
| <tr id='13'><td>13</td></tr>
|
||||
| <tr id='14'><td>14</td></tr>
|
||||
| </table>
|
||||
| </div>
|
||||
| <div>
|
||||
| <table>
|
||||
| <tr id='21'><td>21</td></tr>
|
||||
| </table>
|
||||
| </div>
|
||||
| <div>
|
||||
| <table>
|
||||
| <tr id='31'><td>31</td></tr>
|
||||
| <tr id='32'><td>32</td></tr>
|
||||
| <tr id='33'><td>33</td></tr>
|
||||
| <tr id='34'><td>34</td></tr>
|
||||
| <tr id='35'><td>35</td></tr>
|
||||
| <tr id='36'><td>36</td></tr>
|
||||
| </table>
|
||||
| </div>
|
||||
*/
|
||||
_childTree = [[[HTMLParser alloc] initWithString:@"<div><table>"
|
||||
"<tr id='11'><td>11</td></tr>"
|
||||
"<tr id='12'><td>12</td></tr>"
|
||||
"<tr id='13'><td>13</td></tr>"
|
||||
"<tr id='14'><td>14</td></tr>"
|
||||
"</table></div>"
|
||||
"<div><table>"
|
||||
"<tr id='21'><td>21</td></tr>"
|
||||
"</table></div>"
|
||||
"<div><table>"
|
||||
"<tr id='31'><td>31</td></tr>"
|
||||
"<tr id='32'><td>32</td></tr>"
|
||||
"<tr id='33'><td>33</td></tr>"
|
||||
"<tr id='34'><td>34</td></tr>"
|
||||
"<tr id='35'><td>35</td></tr>"
|
||||
"<tr id='36'><td>36</td></tr>"
|
||||
"</table></div>"] document];
|
||||
|
||||
/*
|
||||
| <div>
|
||||
| <span id='s11'>s11</span>
|
||||
| <span id='s12'>s12</span>
|
||||
| <span id='s13'>s13</span>
|
||||
| <b id='b1'>b1</b>
|
||||
| </div>
|
||||
| <div>
|
||||
| <b id='b2'>b2</b>
|
||||
| <span id='s21'>s21</span>
|
||||
| <span id='s22'>s22</span>
|
||||
| <span id='s23'>s23</span>
|
||||
| </div>
|
||||
| <div>
|
||||
| <span id='s31'>s31</span>
|
||||
| <b id='b3'>b3</b>
|
||||
| <span id='s32'>s32</span>
|
||||
| <b id='b4'>b4</b>
|
||||
| <span id='s33'>s33</span>
|
||||
| <b id='b5'>b5</b>
|
||||
| <span id='s34'>s34</span>
|
||||
| </div>
|
||||
*/
|
||||
_typeTree = [[[HTMLParser alloc] initWithString:@"<div>"
|
||||
"<span id='s11'>s11</span>"
|
||||
"<span id='s12'>s12</span>"
|
||||
"<span id='s13'>s13</span>"
|
||||
"<b id='b1'>b1</b>"
|
||||
"</div>"
|
||||
"<div>"
|
||||
"<b id='b2'>b2</b>"
|
||||
"<span id='s21'>s21</span>"
|
||||
"<span id='s22'>s22</span>"
|
||||
"<span id='s23'>s23</span>"
|
||||
"</div>"
|
||||
"<div>"
|
||||
"<span id='s31'>s31</span>"
|
||||
"<b id='b3'>b3</b>"
|
||||
"<span id='s32'>s32</span>"
|
||||
"<b id='b4'>b4</b>"
|
||||
"<span id='s33'>s33</span>"
|
||||
"<b id='b5'>b5</b>"
|
||||
"<span id='s34'>s34</span>"
|
||||
"</div>"] document];
|
||||
}
|
||||
|
||||
- (void)testOddSelector
|
||||
{
|
||||
NSArray *elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), oddSelector()])];
|
||||
NSArray *expected = @[@"11", @"13", @"21", @"31", @"33", @"35"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)testEvenSelector
|
||||
{
|
||||
NSArray *elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), evenSlector()])];
|
||||
NSArray *expected = @[@"12", @"14", @"32", @"34", @"36"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)testFirstChildSelector
|
||||
{
|
||||
NSArray *elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), firstChildSelector()])];
|
||||
NSArray *expected = @[@"11", @"21", @"31"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)testLastChildSelector
|
||||
{
|
||||
NSArray *elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), lastChildSelector()])];
|
||||
NSArray *expected = @[@"14", @"21", @"36"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)testFirstOfTypeSelector
|
||||
{
|
||||
NSArray *elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), firstOfTypeSelector()])];
|
||||
NSArray *expected = @[@"s11", @"s21", @"s31"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)testLastOfTypeSelector
|
||||
{
|
||||
NSArray *elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), lastOfTypeSelector()])];
|
||||
NSArray *expected = @[@"s13", @"s23", @"s34"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)testOnlyChildSelector
|
||||
{
|
||||
NSArray *elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), onlyChildSelector()])];
|
||||
NSArray *expected = @[@"21"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)testOnlyOfTypeSelector
|
||||
{
|
||||
NSArray *elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), onlyOfTypeSelector()])];
|
||||
NSArray *expected = @[];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), onlyOfTypeSelector()])];
|
||||
expected = @[];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"b"), onlyOfTypeSelector()])];
|
||||
expected = @[@"b1", @"b2"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
#pragma mark - Nth-Child Selector
|
||||
|
||||
- (void)test_NthChild_B_Selector
|
||||
{
|
||||
NSArray *elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthChildSelector(CSSNthExpressionMake(0, 1))])];
|
||||
NSArray *expected = @[@"11", @"21", @"31"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthChildSelector(CSSNthExpressionMake(0, 2))])];
|
||||
expected = @[@"12", @"32"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthChildSelector(CSSNthExpressionMake(0, 3))])];
|
||||
expected = @[@"13", @"33"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)test_NthChild_An_Selector
|
||||
{
|
||||
NSArray *elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthChildSelector(CSSNthExpressionMake(2, 0))])];
|
||||
NSArray *expected = @[@"12", @"14", @"32", @"34", @"36"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthChildSelector(CSSNthExpressionMake(3, 0))])];
|
||||
expected = @[@"13", @"33", @"36"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthChildSelector(CSSNthExpressionMake(4, 0))])];
|
||||
expected = @[@"14", @"34"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)test_NthChild_An_B_Selector
|
||||
{
|
||||
NSArray *elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthChildSelector(CSSNthExpressionMake(3, 1))])];
|
||||
NSArray *expected = @[@"11", @"14", @"21", @"31", @"34"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthChildSelector(CSSNthExpressionMake(3, 2))])];
|
||||
expected = @[@"12", @"32", @"35"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
#pragma mark - Nth-Last-Child Selector
|
||||
|
||||
- (void)test_NthLastChild_B_Selector
|
||||
{
|
||||
NSArray *elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthLastChildSelector(CSSNthExpressionMake(0, 1))])];
|
||||
NSArray *expected = @[@"14", @"21", @"36"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthLastChildSelector(CSSNthExpressionMake(0, 2))])];
|
||||
expected = @[@"13", @"35"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthLastChildSelector(CSSNthExpressionMake(0, 3))])];
|
||||
expected = @[@"12", @"34"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)test_NthLastChild_An_Selector
|
||||
{
|
||||
NSArray *elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthLastChildSelector(CSSNthExpressionMake(2, 0))])];
|
||||
NSArray *expected = @[@"11", @"13", @"31", @"33", @"35"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthLastChildSelector(CSSNthExpressionMake(3, 0))])];
|
||||
expected = @[@"12", @"31", @"34"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthLastChildSelector(CSSNthExpressionMake(4, 0))])];
|
||||
expected = @[@"11", @"33"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)test_NthLastChild_An_B_Selector
|
||||
{
|
||||
NSArray *elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthLastChildSelector(CSSNthExpressionMake(3, 1))])];
|
||||
NSArray *expected = @[@"11", @"14", @"21", @"33", @"36"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_childTree elementsMatchingSelector:allOf(@[typeSelector(@"tr"), nthLastChildSelector(CSSNthExpressionMake(3, 2))])];
|
||||
expected = @[@"13", @"32", @"35"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
#pragma mark - Nth-Of-Type Selector
|
||||
|
||||
- (void)test_NthOfType_B_Selector
|
||||
{
|
||||
NSArray *elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthOfTypeSelector(CSSNthExpressionMake(0, 1))])];
|
||||
NSArray *expected = @[@"s11", @"s21", @"s31"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthOfTypeSelector(CSSNthExpressionMake(0, 2))])];
|
||||
expected = @[@"s12", @"s22", @"s32"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthOfTypeSelector(CSSNthExpressionMake(0, 3))])];
|
||||
expected = @[@"s13", @"s23", @"s33"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)test_NthOfType_An_Selector
|
||||
{
|
||||
NSArray *elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthOfTypeSelector(CSSNthExpressionMake(2, 0))])];
|
||||
NSArray *expected = @[@"s12", @"s22", @"s32", @"s34"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthOfTypeSelector(CSSNthExpressionMake(3, 0))])];
|
||||
expected = @[@"s13", @"s23", @"s33"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthOfTypeSelector(CSSNthExpressionMake(4, 0))])];
|
||||
expected = @[@"s34"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)test_NthOfType_An_B_Selector
|
||||
{
|
||||
NSArray *elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthOfTypeSelector(CSSNthExpressionMake(2, 1))])];
|
||||
NSArray *expected = @[@"s11", @"s13", @"s21", @"s23", @"s31", @"s33"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthOfTypeSelector(CSSNthExpressionMake(2, 2))])];
|
||||
expected = @[@"s12", @"s22", @"s32", @"s34"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
#pragma mark - Nth-Last-Of-Type Selector
|
||||
|
||||
- (void)test_NthLastOfType_B_Selector
|
||||
{
|
||||
NSArray *elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthLastOfTypeSelector(CSSNthExpressionMake(0, 1))])];
|
||||
NSArray *expected = @[@"s13", @"s23", @"s34"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthLastOfTypeSelector(CSSNthExpressionMake(0, 2))])];
|
||||
expected = @[@"s12", @"s22", @"s33"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthLastOfTypeSelector(CSSNthExpressionMake(0, 3))])];
|
||||
expected = @[@"s11", @"s21", @"s32"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)test_NthLastOfType_An_Selector
|
||||
{
|
||||
NSArray *elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthLastOfTypeSelector(CSSNthExpressionMake(2, 0))])];
|
||||
NSArray *expected = @[@"s12", @"s22", @"s31", @"s33"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthLastOfTypeSelector(CSSNthExpressionMake(3, 0))])];
|
||||
expected = @[@"s11", @"s21", @"s32"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthLastOfTypeSelector(CSSNthExpressionMake(4, 0))])];
|
||||
expected = @[@"s31"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
- (void)test_NthLastOfType_An_B_Selector
|
||||
{
|
||||
NSArray *elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthLastOfTypeSelector(CSSNthExpressionMake(2, 1))])];
|
||||
NSArray *expected = @[@"s11", @"s13", @"s21", @"s23", @"s32", @"s34"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
|
||||
elements = [_typeTree elementsMatchingSelector:allOf(@[typeSelector(@"span"), nthLastOfTypeSelector(CSSNthExpressionMake(2, 2))])];
|
||||
expected = @[@"s12", @"s22", @"s31", @"s33"];
|
||||
XCTAssertEqualObjects([elements valueForKeyPath:@"attributes.id"], expected);
|
||||
}
|
||||
|
||||
@end
|
||||
Reference in New Issue
Block a user