diff --git a/HTMLKitTests/CSSAttributeSelectorTests.m b/HTMLKitTests/CSSAttributeSelectorTests.m index b92db7c..8811dff 100644 --- a/HTMLKitTests/CSSAttributeSelectorTests.m +++ b/HTMLKitTests/CSSAttributeSelectorTests.m @@ -221,4 +221,18 @@ XCTAssertEqual([selector acceptElement:_element], NO); } +- (void)testIdSelector +{ + _element[@"id"] = @"id"; + XCTAssertEqual([idSelector(@"id") acceptElement:_element], YES); + XCTAssertEqual([idSelector(@"other id") acceptElement:_element], NO); +} + +- (void)testClassSelector +{ + _element[@"class"] = @"class"; + XCTAssertEqual([classSelector(@"class") acceptElement:_element], YES); + XCTAssertEqual([classSelector(@"other class") acceptElement:_element], NO); +} + @end