Fix HTML5 Lib test to correctly handle ParseError token

JSON structure is not strong with this one
2014-10-26 18:10:32 +01:00
parent e73a8dc512
commit a3358dd8b0
1 changed files with 8 additions and 4 deletions
@@ -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;