Add Test Case for PropertyPrefixSupport And PropertySuffixSupport

Author: sealkingking

src/main/kotlin/extensions/Extension.kt

@@ -44,4 +44,19 @@ abstract class Extension : IImportClassDeclarationInterceptor, IKotlinDataClassI
         return originImportClasses
     }
 
+    fun getTestHelper() = TestHelper(this)
+
+    /**
+     * Test helper for test config settings
+     */
+    class TestHelper(private val extension: Extension) {
+
+        fun setConfig(key: String, value: String) {
+            extension.setConfig(key,value)
+        }
+
+        fun getConfig(key: String): String {
+            return getConfig(key)
+        }
+    }
 }

src/test/kotlin/extensions/wu/seal/PropertyPrefixSupportTest.kt

@@ -0,0 +1,33 @@
+package extensions.wu.seal
+
+import com.winterbe.expekt.should
+import org.junit.Before
+import org.junit.Test
+import wu.seal.jsontokotlin.generateKotlinDataClasses
+import wu.seal.jsontokotlin.test.TestConfig
+
+class PropertyPrefixSupportTest {
+    val json = """{"a":1}"""
+
+
+    val expectResult = """data class Test(
+    @SerializedName("a")
+    val sealA: Int = 0 // 1
+)"""
+
+    @Before
+    fun setUp() {
+        TestConfig.setToTestInitState()
+    }
+
+    @Test
+    fun interceptTest() {
+        val kotlinDataClass =
+                json.generateKotlinDataClasses()
+        PropertyPrefixSupport.getTestHelper().setConfig("wu.seal.property_prefix_enable","true")
+        PropertyPrefixSupport.getTestHelper().setConfig("wu.seal.property_prefix","seal")
+        val generatedCode = kotlinDataClass.applyInterceptors(listOf(PropertyPrefixSupport)).getCode()
+
+        generatedCode.trimMargin().should.equal(expectResult.trimMargin())
+    }
+}

src/test/kotlin/extensions/wu/seal/PropertySuffixSupportTest.kt

@@ -0,0 +1,35 @@
+package extensions.wu.seal
+
+import com.winterbe.expekt.should
+import org.junit.Before
+import org.junit.Test
+import wu.seal.jsontokotlin.generateKotlinDataClasses
+import wu.seal.jsontokotlin.test.TestConfig
+
+class PropertySuffixSupportTest {
+
+    val json = """{"a":1}"""
+
+
+    val expectResult = """data class Test(
+    @SerializedName("a")
+    val aWu: Int = 0 // 1
+)"""
+
+    @Before
+    fun setUp() {
+        TestConfig.setToTestInitState()
+    }
+
+    @Test
+    fun interceptTest() {
+        val kotlinDataClass =
+                json.generateKotlinDataClasses()
+        PropertySuffixSupport.getTestHelper().setConfig("wu.seal.property_suffix_enable","true")
+        PropertySuffixSupport.getTestHelper().setConfig("wu.seal.property_suffix","wu")
+        val generatedCode = kotlinDataClass.applyInterceptors(listOf(PropertySuffixSupport)).getCode()
+
+        generatedCode.trimMargin().should.equal(expectResult.trimMargin())
+    }
+
+}

src/test/kotlin/wu/seal/jsontokotlin/TestUtils.kt

@@ -0,0 +1,10 @@
+package wu.seal.jsontokotlin
+
+import wu.seal.jsontokotlin.classscodestruct.KotlinDataClass
+import wu.seal.jsontokotlin.utils.classblockparse.ClassCodeParser
+
+/**
+ * Generate kotlin data classes with root name 'Test'
+ */
+fun String.generateKotlinDataClasses() =
+        KotlinDataClass.fromParsedKotlinDataClass(ClassCodeParser(KotlinCodeMaker("Test", this).makeKotlinData()).getKotlinDataClass())