From c57d3453fcaa730dc33d1309faf3b481e614993c Mon Sep 17 00:00:00 2001
From: iska <iska.gp@gmail.com>
Date: Mon, 26 Oct 2015 22:27:00 +0100
Subject: [PATCH] Add id & class selector tests

---
 HTMLKitTests/CSSAttributeSelectorTests.m | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

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