1 files changed, 71 insertions, 11 deletions
diff --git a/src/test/model/HtmlParserTest.java b/src/test/model/HtmlParserTest.java
index 58b4555..36c5935 100644
--- a/src/test/model/HtmlParserTest.java
+++ b/src/test/model/HtmlParserTest.java
@@ -2,39 +2,99 @@ package model;
 
 import model.html.ElementNode;
 import model.html.HtmlParser;
+import model.html.TextNode;
 import model.util.Node;
+import org.javatuples.Pair;
 import org.junit.jupiter.api.Test;
 
 import java.util.*;
 
+import static org.junit.jupiter.api.Assertions.*;
+
 public class HtmlParserTest {
 
     String idiomaticHtml = "<!DOCTYPE html><html><head></head><body><p>Hello,  world!</p></body></html>";
-    String brokenHtml = "<html><foo><bar></bar><ba";
-    String trailingTextHtml = "<html><foo><bar></bar>ba";
+    String brokenHtml = "<html><foo><bar></bar><><ba";
+    String trailingTextHtml = "bot<html><foo><bar></bar>ba";
+    String attributesHtml = "<html><attr hello=\"world\" foo='bar' strange=\"cha\\\"rm\" up='do\\'wn'></attr></html>";
 
     @Test
     void testIdiomaticHtml() {
-        String[] idiomaticHtmlArray = {"<!DOCTYPE html>","<html>","<head>","</head>","<body>","<p>","Hello,world!","</p>","</body>","</html>"};
+        ArrayList<Node> expected = new ArrayList<>();
+        ArrayList<Node> expectedChildren = new ArrayList<>();
+        ArrayList<Node> expectedGrandChildren = new ArrayList<>();
+        ArrayList<Node> expectedGreatGrandChildren = new ArrayList<>();
+        expected.add(new ElementNode("html", new ArrayList<>(), expectedChildren));
+        expectedChildren.add(new ElementNode("head"));
+        expectedChildren.add(new ElementNode("body", new ArrayList<>(), expectedGrandChildren));
+        expectedGrandChildren.add(new ElementNode("p", new ArrayList<>(), expectedGreatGrandChildren));
+        expectedGreatGrandChildren.add(new TextNode("Hello, world!"));
+
         HtmlParser parser = new HtmlParser();
-        displayHtmlTree(parser.parseHtml(idiomaticHtml));
-//        assertEquals(HtmlParser.parseHtmlLL(idiomaticHtml), Arrays.asList(idiomaticHtmlArray));
+        assertEqualsHtml(parser.parseHtml(idiomaticHtml), expected);
+        // displayHtmlTree(parser.parseHtml(idiomaticHtml));
     }
 
     @Test
     void testBrokenHtml() {
-        String[] brokenHtmlArray = {"<html>","<foo>","<bar>","</bar>","<ba>"};
+        ArrayList<Node> expected = new ArrayList<>();
+        ArrayList<Node> expectedChildren = new ArrayList<>();
+        ArrayList<Node> expectedGrandChildren = new ArrayList<>();
+        expected.add(new ElementNode("html", new ArrayList<>(), expectedChildren));
+        expectedChildren.add(new ElementNode("foo", new ArrayList<>(), expectedGrandChildren));
+        expectedGrandChildren.add(new ElementNode("bar", new ArrayList<>()));
+        expectedGrandChildren.add(new TextNode("<>"));
+
         HtmlParser parser = new HtmlParser();
-        displayHtmlTree(parser.parseHtml(brokenHtml));
-//        assertEquals(HtmlParser.parseHtmlLL(brokenHtml), Arrays.asList(brokenHtmlArray));
+        assertEqualsHtml(parser.parseHtml(brokenHtml), expected);
+        // displayHtmlTree(parser.parseHtml(brokenHtml));
     }
 
     @Test
     void testTrailingTextHtml() {
-        String[] trailingTextHtmlArray = {"<html>","<foo>","<bar>","</bar>","ba"};
+        ArrayList<Node> expected = new ArrayList<>();
+        ArrayList<Node> expectedChildren = new ArrayList<>();
+        ArrayList<Node> expectedGrandChildren = new ArrayList<>();
+        expected.add(new TextNode("bot"));
+        expected.add(new ElementNode("html", new ArrayList<>(), expectedChildren));
+        expected.add(new TextNode("ba"));
+        expectedChildren.add(new ElementNode("foo", new ArrayList<>(), expectedGrandChildren));
+        expectedGrandChildren.add(new ElementNode("bar", new ArrayList<>()));
+
         HtmlParser parser = new HtmlParser();
-        displayHtmlTree(parser.parseHtml(trailingTextHtml));
-//        assertEquals(HtmlParser.parseHtmlLL(trailingTextHtml), Arrays.asList(trailingTextHtmlArray));
+        assertEqualsHtml(parser.parseHtml(trailingTextHtml), expected);
+        // displayHtmlTree(parser.parseHtml(trailingTextHtml));
+    }
+
+    @Test
+    void testAttributesHtml() {
+        ArrayList<Node> expected = new ArrayList<>();
+        ArrayList<Node> expectedChildren = new ArrayList<>();
+        ArrayList<Pair<String, String>> expectedAttributes = new ArrayList<>();
+        expected.add(new ElementNode("html", new ArrayList<>(), expectedChildren));
+        expectedChildren.add(new ElementNode("attr", expectedAttributes));
+        expectedAttributes.add(new Pair<>("hello", "world"));
+        expectedAttributes.add(new Pair<>("foo", "bar"));
+        expectedAttributes.add(new Pair<>("strange", "cha\"rm"));
+        expectedAttributes.add(new Pair<>("up", "do'wn"));
+
+        HtmlParser parser = new HtmlParser();
+        var parsed = parser.parseHtml(attributesHtml);
+        displayHtmlTree(parsed);
+        assertEqualsHtml(parsed, expected);
+    }
+
+    /**
+     * Complicated helper function for tests.
+     */
+    private static void assertEqualsHtml(ArrayList<Node> html, ArrayList<Node> expected) {
+        for (int i = 0; i < html.size(); i++) {
+            assertEquals(html.get(i).getData(), expected.get(i).getData());
+            // System.out.println(html.get(i).getData() + " " + expected.get(i).getData());
+            if (html.get(i) instanceof ElementNode) {
+                assertEqualsHtml(((ElementNode) html.get(i)).getChildren(), ((ElementNode) expected.get(i)).getChildren());
+            }
+        }
     }
 
     /**