Increase test coverage

theEvilReaper · Jan 12, 2024 · 5999e28 · 5999e28
1 parent 15041fe
commit 5999e28
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/test/kotlin/net/theevilreaper/dartpoet/enumeration/EnumPropertySpecTest.kt b/src/test/kotlin/net/theevilreaper/dartpoet/enumeration/EnumPropertySpecTest.kt
@@ -2,6 +2,7 @@ package net.theevilreaper.dartpoet.enumeration
 
 import com.google.common.truth.Truth
 import net.theevilreaper.dartpoet.enum.EnumPropertySpec
+import net.theevilreaper.dartpoet.type.ClassName
 import org.junit.jupiter.api.Test
 import org.junit.jupiter.params.ParameterizedTest
 import org.junit.jupiter.params.provider.Arguments
@@ -27,6 +28,12 @@ class EnumPropertySpecTest {
                 "navigation('/dashboard')"
             )
         )
+
+        @JvmStatic
+        private fun genericEnumProperties(): Stream<Arguments> = Stream.of(
+            Arguments.of(EnumPropertySpec.builder("test").generic(String::class).build(), "test<String>"),
+            Arguments.of(EnumPropertySpec.builder("test").generic(ClassName("E")).build(), "test<E>"),
+        )
     }
 
     @ParameterizedTest
@@ -35,6 +42,12 @@ class EnumPropertySpecTest {
         Truth.assertThat(spec.toString()).isEqualTo(excepted)
     }
 
+    @ParameterizedTest
+    @MethodSource("genericEnumProperties")
+    fun `test generic enum property generation`(spec: EnumPropertySpec, excepted: String) {
+        Truth.assertThat(spec.toString()).isEqualTo(excepted)
+    }
+
     @Test
     fun `test spec conversion to a builder`() {
         val propertySpec = EnumPropertySpec