Read classes on HTML path elements

1 files changed, 16 insertions, 0 deletions

diff --git a/tests/style_reader_tests.py b/tests/style_reader_tests.py
index 595ece1..a25d7ce 100644
--- a/tests/style_reader_tests.py
+++ b/tests/style_reader_tests.py
@@ -35,3 +35,19 @@ class ReadHtmlPathTests(object):
             html_paths.path([html_paths.element(["ul"]), html_paths.element(["li"])]),
             read_html_path("ul > li")
         )
+
+    
+    @istest
+    def can_read_class_on_element(self):
+        assert_equal(
+            html_paths.path([html_paths.element(["p"], class_names=["tip"])]),
+            read_html_path("p.tip")
+        )
+
+    
+    @istest
+    def can_read__multiple_classes_on_element(self):
+        assert_equal(
+            html_paths.path([html_paths.element(["p"], class_names=["tip", "help"])]),
+            read_html_path("p.tip.help")
+        )