Add cleaning for reverse, underline, and italic formatting.

[xudocci.git] / src / test / java / net / pterodactylus / irc / util / MessageCleanerTest.java

diff --git a/src/test/java/net/pterodactylus/irc/util/MessageCleanerTest.java b/src/test/java/net/pterodactylus/irc/util/MessageCleanerTest.java
index 6ab7b0a..bd5a2a2 100644 (file)
--- a/src/test/java/net/pterodactylus/irc/util/MessageCleanerTest.java
+++ b/src/test/java/net/pterodactylus/irc/util/MessageCleanerTest.java
@@ -24,7 +24,7 @@ import java.util.EnumSet;
 
 import net.pterodactylus.irc.util.MessageCleaner.Attributes;
 
-import org.testng.annotations.Test;
+import org.junit.Test;
 
 /**
  * Tests for {@link MessageCleaner}.
@@ -91,4 +91,22 @@ public class MessageCleanerTest {
                assertThat(clean, is("This contains bold and color and clear characters."));
        }
 
+       @Test
+       public void removeUnderlineFormatting() {
+               MessageCleaner messageCleaner = MessageCleaner.getDefaultInstance();
+               assertThat(messageCleaner.clean("Text\u0016underline"), is("Textunderline"));
+       }
+
+       @Test
+       public void removeReverseFormatting() {
+               MessageCleaner messageCleaner = MessageCleaner.getDefaultInstance();
+               assertThat(messageCleaner.clean("Text\u001freverse"), is("Textreverse"));
+       }
+
+       @Test
+       public void removeItalicFormatting() {
+               MessageCleaner messageCleaner = MessageCleaner.getDefaultInstance();
+               assertThat(messageCleaner.clean("Text\u001ditalics"), is("Textitalics"));
+       }
+
 }