diff --git a/HTMLKitTests/HTML5LibTest.m b/HTMLKitTests/HTML5LibTest.m
index 662f4e4..92933fc 100644
--- a/HTMLKitTests/HTML5LibTest.m
+++ b/HTMLKitTests/HTML5LibTest.m
@@ -71,13 +71,17 @@
self.ignoreErrorOrder = [fixture[@"ignoreErrorOrder"] boolValue];
}
-- (HTMLToken *)processOutputToken:(NSArray *)output doubleEscaped:(BOOL)doubleEscaped
+- (HTMLToken *)processOutputToken:(id)output doubleEscaped:(BOOL)doubleEscaped
{
+ if ([output isKindOfClass:[NSString class]] && [output isEqualToString:@"ParseError"]) {
+ return [HTMLParseErrorToken new];
+ }
+
NSString *type = [output firstObject];
NSString *data = nil;
- if (output.count > 1) {
- NSString *data = [output lastObject];
+ if ([output count] > 1) {
+ data = [output lastObject];
if (doubleEscaped) {
data = [self processDoubleEscaped:data];
}
@@ -104,7 +108,7 @@
NSString *value = [attributes objectForKey:name];
[token.attributes setObject:value forKey:name];
}
- token.selfClosing = (output.count == 4);
+ token.selfClosing = ([output count] == 4);
return token;
}
return nil;