diff --git a/HTMLKit/HTMLTokenizer.m b/HTMLKit/HTMLTokenizer.m index f4e207c..d64cba9 100644 --- a/HTMLKit/HTMLTokenizer.m +++ b/HTMLKit/HTMLTokenizer.m @@ -103,14 +103,6 @@ return nextToken; } -- (NSArray *)allObjects -{ - while (_eof == NO) { - [self read]; - } - return _tokens; -} - - (void)read { SEL selector = [[_states objectForKey:@(_currentState)] pointerValue]; diff --git a/HTMLKitTests/HTMLKitTokenizerTests.m b/HTMLKitTests/HTMLKitTokenizerTests.m index b480a3b..dbf40aa 100644 --- a/HTMLKitTests/HTMLKitTokenizerTests.m +++ b/HTMLKitTests/HTMLKitTokenizerTests.m @@ -90,15 +90,15 @@ - (void)testTokenizer { for (HTML5LibTokenizerTest *test in self.testsList) { - HTMLTokenizer *tokenizer = [[HTMLTokenizer alloc] initWithString:test.input]; - [tokenizer setValue:test.lastStartTag forKey:@"_lastStartTagName"]; for (NSNumber *state in test.initialStates) { + HTMLTokenizer *tokenizer = [[HTMLTokenizer alloc] initWithString:test.input]; + [tokenizer setValue:test.lastStartTag forKey:@"_lastStartTagName"]; tokenizer.state = [state integerValue]; NSArray *expectedTokens = test.output; - NSArray *tokens = [tokenizer allObjects]; + NSArray *tokens = tokenizer.allObjects; XCTAssertEqualObjects(tokens, expectedTokens, @"%@", test.title); } }