OF-2628: Add unit tests

igniterealtime · Jul 21, 2023 · dc11312 · dc11312
1 parent f290e0c
commit dc11312
Showing 1 changed file with 29 additions and 2 deletions.
diff --git a/xmppserver/src/test/java/org/jivesoftware/openfire/nio/XMLLightweightParserTest.java b/xmppserver/src/test/java/org/jivesoftware/openfire/nio/XMLLightweightParserTest.java
@@ -20,8 +20,7 @@
 
 import java.nio.charset.StandardCharsets;
 
-import static org.junit.jupiter.api.Assertions.assertEquals;
-import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.*;
 
 /**
  * Unit tests that verify the functionality as implemented in {@link XMLLightweightParser}
@@ -185,4 +184,32 @@ public void testOF2329SelfTerminatingWithNewline() throws Exception
         assertNotNull( result );
         assertEquals(1, result.length );
     }
+
+    @Test
+    public void testHasIllegalCharacterReferencesFindsIllegalDecimalChars() throws Exception
+    {
+        final String input = "test &#65; test";
+        assertFalse(XMLLightweightParser.hasIllegalCharacterReferences(input));
+    }
+
+    @Test
+    public void testHasIllegalCharacterReferencesFindsLegalDecimalChars() throws Exception
+    {
+        final String input = "test &#7; test";
+        assertTrue(XMLLightweightParser.hasIllegalCharacterReferences(input));
+    }
+
+    @Test
+    public void testHasIllegalCharacterReferencesFindsIllegalHexChars() throws Exception
+    {
+        final String input = "test &#x41; test";
+        assertFalse(XMLLightweightParser.hasIllegalCharacterReferences(input));
+    }
+
+    @Test
+    public void testHasIllegalCharacterReferencesFindsLegalHexChars() throws Exception
+    {
+        final String input = "test &#x7; test";
+        assertTrue(XMLLightweightParser.hasIllegalCharacterReferences(input));
+    }
 }