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