HTMLKit/Tests/HTMLKitTests/HTMLKitTokenizerTests.m

//
//  HTMLTokenizerTests.m
//  HTMLKit
//
//  Created by Iska on 25/10/14.
//  Copyright (c) 2014 BrainCookie. All rights reserved.
//

#import <XCTest/XCTest.h>

#import "HTMLKitTestUtil.h"

#import "HTML5LibTokenizerTest.h"
#import "HTMLTokenizer.h"
#import "HTMLTokenizerStates.h"
#import "HTMLTokens.h"

#import "HTMLParser.h"
#import "HTMLDocument.h"

#pragma mark - Extensions

@implementation HTMLParseErrorToken (Testing)

- (BOOL)isEqual:(id)object
{
	return [object isKindOfClass:[HTMLParseErrorToken class]];
}

@end

#pragma mark - HTML5Lib Test Suite

@interface HTMLKitTokenizerTests : XCTestCase
@property (nonatomic, strong) NSArray *testsList;
@end

@implementation HTMLKitTokenizerTests

+ (XCTestSuite *)defaultTestSuite
{
	XCTestSuite *suite = [[XCTestSuite alloc] initWithName:NSStringFromClass(self)];

	NSDictionary *testsMap = [HTML5LibTokenizerTest loadHTML5LibTokenizerTests];
	[testsMap enumerateKeysAndObjectsUsingBlock:^(id key, id obj, BOOL *stop) {
		[self addTestCaseForTestFile:key withTests:obj toTestSuite:suite];
	}];

	return suite;
}

+ (void)addTestCaseForTestFile:(NSString *)testFile withTests:(NSArray *)tests toTestSuite:(XCTestSuite *)suite
{
	NSString *testName = [testFile.stringByDeletingPathExtension stringByReplacingOccurrencesOfString:@"-" withString:@"_"];
	testName = [NSString stringWithFormat:@"testTokenizer__%@", testName];

	NSInvocation *invocation = [HTMLKitTestUtil addTestToClass:self withName:testName block:^ (HTMLKitTokenizerTests *instance){
		[instance runTests];
	}];

	XCTestCase *testCase = [[self alloc] initWithInvocation:invocation tests:tests];
	[suite addTest:testCase];
}

#pragma mark - Instance

- (instancetype)initWithInvocation:(NSInvocation *)invocation tests:(NSArray *)tests
{
	self = [super initWithInvocation:invocation];
	if (self) {
		_testsList = tests;
	}
	return self;
}

#pragma mark - Tests

- (void)runTests
{
	for (HTML5LibTokenizerTest *test in self.testsList) {

		for (NSNumber *state in test.initialStates) {
			HTMLTokenizer *tokenizer = [[HTMLTokenizer alloc] initWithString:test.input];

			[tokenizer setValue:test.lastStartTag forKey:@"_lastStartTagName"];
			tokenizer.state = [state integerValue];

			NSArray *expectedErrors = test.errors;
			NSMutableArray *actualErrors = [NSMutableArray new];
			tokenizer.parseErrorCallback = ^(HTMLParseErrorToken *token) {
				[actualErrors addObject:token];
			};

			NSArray *expectedTokens = test.output;
			NSArray *actualTokens = tokenizer.allObjects;

			NSString *message = [NSString stringWithFormat:@"\nTest file: \'%@\'\nTitle: '%@'\nInput: '%@'\n",
								 test.testFile,
								 test.title,
								 test.input];
			XCTAssertEqualObjects(actualTokens, expectedTokens, @"%@", message);


			message = [NSString stringWithFormat:@"\nTest file: \'%@\'\nTitle: '%@'\nInput: '%@'\n",
								 test.testFile,
								 test.title,
								 test.input];
			XCTAssertEqualObjects(actualErrors, expectedErrors, @"%@", message);
		}
	}
}

@end