Add a test for #1035

square · May 14, 2022 · aeefaa8 · aeefaa8
1 parent 2b2b160
commit aeefaa8
Showing 1 changed file with 29 additions and 3 deletions.
diff --git a/kotlinpoet/src/test/java/com/squareup/kotlinpoet/TypeSpecTest.kt b/kotlinpoet/src/test/java/com/squareup/kotlinpoet/TypeSpecTest.kt
@@ -3938,7 +3938,7 @@ class TypeSpecTest {
 
   @Test fun classHeaderFormatting() {
     val typeSpec = TypeSpec.classBuilder("Person")
-      .addModifiers(KModifier.DATA)
+      .addModifiers(DATA)
       .primaryConstructor(
         FunSpec.constructorBuilder()
           .addParameter("id", Int::class)
@@ -3987,7 +3987,7 @@ class TypeSpecTest {
       .build()
 
     val typeSpec = TypeSpec.classBuilder("Person")
-      .addModifiers(KModifier.DATA)
+      .addModifiers(DATA)
       .primaryConstructor(
         FunSpec.constructorBuilder()
           .addParameter(idParameterSpec)
@@ -4402,7 +4402,7 @@ class TypeSpecTest {
     val source = FileSpec.builder("com.squareup.tacos", "Taco")
       .addType(
         TypeSpec.classBuilder("Taco")
-          .addModifiers(KModifier.DATA)
+          .addModifiers(DATA)
           .addProperty(
             PropertySpec.builder("madeFreshDatabaseDate", sqlTaco)
               .initializer("madeFreshDatabaseDate")
@@ -5123,6 +5123,32 @@ class TypeSpecTest {
     )
   }
 
+  // https://github.com/square/kotlinpoet/issues/1035
+  @Test fun dataClassWithKeywordProperty() {
+    val parameter = ParameterSpec.builder("data", STRING).build()
+    val typeSpec = TypeSpec.classBuilder("Example")
+      .addModifiers(DATA)
+      .primaryConstructor(
+        FunSpec.constructorBuilder()
+          .addParameter(parameter)
+          .build()
+      )
+      .addProperty(
+        PropertySpec.builder(parameter.name, STRING)
+          .initializer("%N", parameter)
+          .build()
+      )
+      .build()
+    assertThat(typeSpec.toString()).isEqualTo(
+      """
+      public data class Example(
+        public val `data`: kotlin.String,
+      )
+
+      """.trimIndent()
+    )
+  }
+
   companion object {
     private const val donutsPackage = "com.squareup.donuts"
   }