Support attributes in HTML paths in style mappings

1 files changed, 14 insertions, 0 deletions

diff --git a/src/test/java/org/zwobble/mammoth/tests/styles/parsing/HtmlPathParsingTests.java b/src/test/java/org/zwobble/mammoth/tests/styles/parsing/HtmlPathParsingTests.java
index 23cc730..85591ad 100644
--- a/src/test/java/org/zwobble/mammoth/tests/styles/parsing/HtmlPathParsingTests.java
+++ b/src/test/java/org/zwobble/mammoth/tests/styles/parsing/HtmlPathParsingTests.java
@@ -61,6 +61,20 @@ public class HtmlPathParsingTests {
     }
 
     @Test
+    public void canReadAttributeOnElement() {
+        assertThat(
+            parseHtmlPath("p[lang='fr']"),
+            deepEquals(HtmlPath.collapsibleElement("p", map("lang", "fr"))));
+    }
+
+    @Test
+    public void canReadMultipleAttributesOnElement() {
+        assertThat(
+            parseHtmlPath("p[lang='fr'][data-x='y']"),
+            deepEquals(HtmlPath.collapsibleElement("p", map("lang", "fr", "data-x", "y"))));
+    }
+
+    @Test
     public void canReadWhenElementMustBeFresh() {
         assertThat(
             parseHtmlPath("p:fresh"),