Add test cases for TriNode

AY2324S2-CS2103T-F13-1 · Mar 19, 2024 · fccdf65 · fccdf65
1 parent cca24d9
commit fccdf65
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/src/main/java/seedu/address/commons/util/TrieNode.java b/src/main/java/seedu/address/commons/util/TrieNode.java
@@ -88,6 +88,9 @@ public int hashCode() {
 
     @Override
     public String toString() {
-        return String.format("TrieNode: %s", this.character);
+        return String.format("TrieNode: %s%s",
+            this.character,
+            this.isEndOfWord ? "\0" : ""
+        );
     }
 }
diff --git a/src/test/java/seedu/address/commons/util/TrieNodeTest.java b/src/test/java/seedu/address/commons/util/TrieNodeTest.java
@@ -41,11 +41,33 @@ void getChildren() {
     @Test
     void testEquals() {
         TrieNode trieNode = new TrieNode('a');
+
+        assertEquals(trieNode, trieNode);
+        assertNotEquals(trieNode, null);
+        assertNotEquals(trieNode, "a");
+
         TrieNode trieNode2 = new TrieNode('a');
 
         assertEquals(trieNode, trieNode2);
 
         trieNode.setEndOfWord(true);
         assertNotEquals(trieNode, trieNode2);
     }
+
+    @Test
+    void testHashCode() {
+        TrieNode trieNode = new TrieNode('a');
+        TrieNode trieNode2 = new TrieNode('a');
+
+        assertEquals(trieNode.hashCode(), trieNode2.hashCode());
+    }
+
+    @Test
+    void testToString() {
+        TrieNode trieNode = new TrieNode('a');
+        assertEquals("TrieNode: a", trieNode.toString());
+
+        trieNode.setEndOfWord(true);
+        assertEquals("TrieNode: a\0", trieNode.toString());
+    }
 }