Add JUnit tests for KotlinLangUtils

Yelp · Jul 11, 2019 · 9857ec7 · 9857ec7
1 parent 9d6a86d
commit 9857ec7
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/plugin/src/test/java/com/yelp/codegen/KotlinLangUtilsTest.kt b/plugin/src/test/java/com/yelp/codegen/KotlinLangUtilsTest.kt
@@ -0,0 +1,33 @@
+package com.yelp.codegen
+
+import com.yelp.codegen.utils.sanitizeKotlinSpecificNames
+import org.junit.Assert.assertEquals
+import org.junit.Test
+
+class KotlinLangUtilsTest {
+
+    @Test
+    fun sanitizeKotlinSpecificNames_withReplacementInMap() {
+        assertEquals("valueSemicolon", "value;".sanitizeKotlinSpecificNames(mapOf(";" to "Semicolon")))
+    }
+
+    @Test
+    fun sanitizeKotlinSpecificNames_withDash() {
+        assertEquals("value_", "value-".sanitizeKotlinSpecificNames(emptyMap()))
+    }
+
+    @Test
+    fun sanitizeKotlinSpecificNames_withStartingNumber() {
+        assertEquals("_42value", "42value".sanitizeKotlinSpecificNames(emptyMap()))
+    }
+
+    @Test
+    fun sanitizeKotlinSpecificNames_withUnderscore() {
+        assertEquals("Underscore", "_".sanitizeKotlinSpecificNames(emptyMap()))
+    }
+
+    @Test
+    fun sanitizeKotlinSpecificNames_withMultipleUnderscores() {
+        assertEquals("UnderscoreUnderscoreUnderscore", "___".sanitizeKotlinSpecificNames(emptyMap()))
+    }
+}