Add attributes to self closing element

1 files changed, 8 insertions, 4 deletions

diff --git a/src/main/java/org/zwobble/mammoth/html/HtmlSelfClosingElement.java b/src/main/java/org/zwobble/mammoth/html/HtmlSelfClosingElement.java
index f3ef3f8..73f4734 100644
--- a/src/main/java/org/zwobble/mammoth/html/HtmlSelfClosingElement.java
+++ b/src/main/java/org/zwobble/mammoth/html/HtmlSelfClosingElement.java
@@ -1,16 +1,20 @@
 package org.zwobble.mammoth.html;
 
+import java.util.Map;
+
 public class HtmlSelfClosingElement implements HtmlNode {
     private final String tagName;
+    private final Map<String, String> attributes;
 
-    public HtmlSelfClosingElement(String tagName) {
+    public HtmlSelfClosingElement(String tagName, Map<String, String> attributes) {
         this.tagName = tagName;
+        this.attributes = attributes;
     }
 
     @Override
     public void write(StringBuilder builder) {
-        builder.append("<")
-            .append(tagName)
-            .append(" />");
+        builder.append("<").append(tagName);
+        HtmlWriter.generateAttributes(attributes, builder);
+        builder.append(" />");
     }
 }