Add convenience modifier

outfoxx · Dec 14, 2023 · 0d71e7e · 0d71e7e
1 parent ca3933f
commit 0d71e7e
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/src/main/java/io/outfoxx/swiftpoet/Modifier.kt b/src/main/java/io/outfoxx/swiftpoet/Modifier.kt
@@ -38,6 +38,8 @@ enum class Modifier(
 
   REQUIRED("required", Target.FUNCTION),
 
+  CONVENIENCE("convenience", Target.FUNCTION),
+
   INOUT("inout", Target.PARAMETER),
 
   WEAK("weak", Target.PROPERTY),

diff --git a/src/test/java/io/outfoxx/swiftpoet/test/FunctionSpecTests.kt b/src/test/java/io/outfoxx/swiftpoet/test/FunctionSpecTests.kt
@@ -430,6 +430,28 @@ class FunctionSpecTests {
     )
   }
 
+  @Test
+  @DisplayName("Generates constructor with convenience modifier")
+  fun testConvenienceModifier() {
+    val testFunc = FunctionSpec.constructorBuilder()
+      .addModifiers(Modifier.PUBLIC, Modifier.CONVENIENCE)
+      .build()
+
+    val out = StringWriter()
+    testFunc.emit(CodeWriter(out), setOf())
+
+    assertThat(
+      out.toString(),
+      equalTo(
+        """
+            public convenience init() {
+            }
+
+        """.trimIndent()
+      )
+    )
+  }
+
   @Test
   @DisplayName("Generates decorators formatted")
   fun testGenDecorators() {