From f28dee0ef3a0dd4a0819a3d3a1c800a83a0f07f5 Mon Sep 17 00:00:00 2001
From: Linnea Gräf <nea@nea.moe>
Date: Fri, 1 Mar 2024 10:46:54 +0100
Subject: Add more color code tests

[no changelog]
---
 .../kotlin/moe/nea/firmament/test/ColorCode.kt     | 35 ++++++++++++++++++++++
 1 file changed, 35 insertions(+)

(limited to 'src/test/kotlin/moe')

diff --git a/src/test/kotlin/moe/nea/firmament/test/ColorCode.kt b/src/test/kotlin/moe/nea/firmament/test/ColorCode.kt
index 737534e..b8968aa 100644
--- a/src/test/kotlin/moe/nea/firmament/test/ColorCode.kt
+++ b/src/test/kotlin/moe/nea/firmament/test/ColorCode.kt
@@ -25,4 +25,39 @@ class ColorCode {
         Assertions.assertEquals("b§lc§l", "§ab§l§§c§l".removeColorCodes(true).toString())
         Assertions.assertEquals("§lb§lc", "§l§ab§l§§c".removeColorCodes(true).toString())
     }
+
+    @Test
+    fun testEdging() {
+        Assertions.assertEquals("", "§".removeColorCodes())
+        Assertions.assertEquals("a", "a§".removeColorCodes())
+        Assertions.assertEquals("b", "§ab§".removeColorCodes())
+    }
+
+    @Test
+    fun `testDouble§`() {
+        Assertions.assertEquals("1", "§§1".removeColorCodes())
+    }
+
+    @Test
+    fun testKeepNonColor() {
+        Assertions.assertEquals("§k§l§m§n§o§r", "§k§l§m§f§n§o§r".removeColorCodes(true))
+    }
+
+    @Test
+    fun testPlainString() {
+        Assertions.assertEquals("bcdefgp", "bcdefgp".removeColorCodes())
+        Assertions.assertEquals("", "".removeColorCodes())
+    }
+
+    @Test
+    fun testSomeNormalTestCases() {
+        Assertions.assertEquals(
+            "You are not currently in a party.",
+            "§r§cYou are not currently in a party.§r".removeColorCodes()
+        )
+        Assertions.assertEquals(
+            "Ancient Necron's Chestplate ✪✪✪✪",
+            "§dAncient Necron's Chestplate §6✪§6✪§6✪§6✪".removeColorCodes()
+        )
+    }
 }
-- 
cgit