chore: add codegen tests

Author: WipeAir

graphql-kotlin-toolkit-codegen/build.gradle.kts

@@ -14,9 +14,10 @@ dependencies {
 
     implementation(project(":graphql-kotlin-toolkit-common"))
 
-    testImplementation("org.jetbrains.kotlin:kotlin-test")
-    testImplementation("org.jetbrains.kotlin:kotlin-test-junit")
-    testImplementation("io.kotlintest:kotlintest-runner-junit5:3.3.2")
+    testImplementation("com.github.tschuchortdev:kotlin-compile-testing:1.2.8")
+    testImplementation("org.junit.jupiter:junit-jupiter-engine:5.6.2")
+    testImplementation("org.junit.jupiter:junit-jupiter-params:5.6.2")
+    testImplementation("org.jetbrains.kotlin:kotlin-reflect")
 }
 
 tasks.dokka {

graphql-kotlin-toolkit-codegen/src/test/kotlin/com/auritylab/graphql/kotlin/toolkit/codegen/_test/AbstractCompilationTest.kt

@@ -0,0 +1,26 @@
+package com.auritylab.graphql.kotlin.toolkit.codegen._test
+
+import com.auritylab.graphql.kotlin.toolkit.codegen.generator.FileGenerator
+import com.tschuchort.compiletesting.KotlinCompilation
+import com.tschuchort.compiletesting.SourceFile
+import kotlin.reflect.KClass
+
+abstract class AbstractCompilationTest {
+
+    protected fun compile(generator: FileGenerator): KClass<*> {
+        // Generate the code of the generator.
+        val fileSpec = generator.generate()
+
+        // Create the source files
+        val source = SourceFile.kotlin("Generated.kt", fileSpec.toString())
+
+        val compilation = KotlinCompilation()
+
+        compilation.sources = listOf(source)
+        compilation.inheritClassPath = false
+
+        val compileResult = compilation.compile()
+
+        return compileResult.classLoader.loadClass(fileSpec.packageName + "." + fileSpec.name).kotlin
+    }
+}

graphql-kotlin-toolkit-codegen/src/test/kotlin/com/auritylab/graphql/kotlin/toolkit/codegen/_test/TestObject.kt

@@ -0,0 +1,26 @@
+package com.auritylab.graphql.kotlin.toolkit.codegen._test
+
+import com.auritylab.graphql.kotlin.toolkit.codegen.CodegenOptions
+import com.auritylab.graphql.kotlin.toolkit.codegen.codeblock.ArgumentCodeBlockGenerator
+import com.auritylab.graphql.kotlin.toolkit.codegen.mapper.GeneratedMapper
+import com.auritylab.graphql.kotlin.toolkit.codegen.mapper.ImplementerMapper
+import com.auritylab.graphql.kotlin.toolkit.codegen.mapper.KotlinTypeMapper
+import graphql.schema.GraphQLObjectType
+import graphql.schema.GraphQLSchema
+import java.nio.file.Path
+
+internal object TestObject {
+    val options = CodegenOptions(hashSetOf(), Path.of(""))
+    val generatedMapper = GeneratedMapper(options)
+    val kotlinTypeMapper = KotlinTypeMapper(options, generatedMapper)
+    val argumentCodeBlockGenerator = ArgumentCodeBlockGenerator(kotlinTypeMapper, generatedMapper)
+    val implementerMapper = ImplementerMapper(options, schema)
+
+    val schema: GraphQLSchema
+        get() = GraphQLSchema.newSchema()
+            .query(GraphQLObjectType.newObject().name("Query").build())
+            .mutation(GraphQLObjectType.newObject().name("Mutation").build())
+            .build()
+}
+
+

graphql-kotlin-toolkit-codegen/src/test/kotlin/com/auritylab/graphql/kotlin/toolkit/codegen/generator/EnumGeneratorTest.kt

@@ -0,0 +1,40 @@
+package com.auritylab.graphql.kotlin.toolkit.codegen.generator
+
+import com.auritylab.graphql.kotlin.toolkit.codegen._test.AbstractCompilationTest
+import com.auritylab.graphql.kotlin.toolkit.codegen._test.TestObject
+import graphql.schema.GraphQLEnumType
+import org.junit.jupiter.api.Assertions
+import org.junit.jupiter.api.Test
+import kotlin.reflect.full.isSubclassOf
+
+internal class EnumGeneratorTest : AbstractCompilationTest() {
+    @Test
+    fun shouldGenerateCompilableCode() {
+        val generator =
+            EnumGenerator(testEnum, TestObject.options, TestObject.kotlinTypeMapper, TestObject.generatedMapper)
+
+        // Compile the generator output.
+        val generatedClass = compile(generator)
+
+        // Assert that the generated class is of type enum.
+        Assertions.assertTrue(generatedClass.isSubclassOf(Enum::class))
+
+        val enumConstants = generatedClass.java.enumConstants as Array<Enum<*>>
+
+        // Assert the size of the enum constants.
+        Assertions.assertEquals(4, enumConstants.size)
+
+        // Assert against the defined values.
+        testEnum.values.forEachIndexed { i, definition ->
+            Assertions.assertEquals(definition.name, enumConstants[i].name)
+        }
+    }
+}
+
+private val testEnum = GraphQLEnumType.newEnum()
+    .name("ETestEnum")
+    .value("first")
+    .value("second")
+    .value("third")
+    .value("fourth")
+    .build()

graphql-kotlin-toolkit-codegen/src/test/kotlin/com/auritylab/graphql/kotlin/toolkit/codegen/generator/InputObjectGeneratorTest.kt

@@ -0,0 +1,84 @@
+package com.auritylab.graphql.kotlin.toolkit.codegen.generator
+
+import com.auritylab.graphql.kotlin.toolkit.codegen._test.AbstractCompilationTest
+import com.auritylab.graphql.kotlin.toolkit.codegen._test.TestObject
+import graphql.Scalars
+import graphql.schema.GraphQLInputObjectField
+import graphql.schema.GraphQLInputObjectType
+import graphql.schema.GraphQLNonNull
+import org.junit.jupiter.api.Assertions
+import org.junit.jupiter.api.BeforeAll
+import org.junit.jupiter.api.Test
+import org.junit.jupiter.api.TestInstance
+import kotlin.reflect.KClass
+import kotlin.reflect.full.companionObject
+import kotlin.reflect.full.memberFunctions
+import kotlin.reflect.full.memberProperties
+import kotlin.reflect.full.starProjectedType
+
+@TestInstance(TestInstance.Lifecycle.PER_CLASS)
+internal class InputObjectGeneratorTest : AbstractCompilationTest() {
+    val generator = InputObjectGenerator(
+        testInputObject,
+        TestObject.argumentCodeBlockGenerator,
+        TestObject.options,
+        TestObject.kotlinTypeMapper,
+        TestObject.generatedMapper
+    )
+
+    lateinit var generatedClass: KClass<*>
+
+    @BeforeAll
+    fun compileCode() {
+        // Compile the code of the generator
+        generatedClass = compile(generator)
+    }
+
+    @Test
+    fun `should generate properties correctly`() {
+        // Assert against the member properties.
+        val memberProperties = generatedClass.memberProperties.toList()
+        Assertions.assertEquals(2, memberProperties.size)
+        Assertions.assertEquals("name", memberProperties[0].name)
+        Assertions.assertEquals("number", memberProperties[1].name)
+    }
+
+    @Test
+    fun `should generate build method correctly`() {
+        Assertions.assertNotNull(generatedClass.companionObject)
+        val companionObject = generatedClass.companionObject!!
+
+        // Assert against the member functions.
+        val memberFunctions = companionObject.memberFunctions
+
+        // Search for the builder method
+        val builderMethod = memberFunctions.firstOrNull { it.name.endsWith("BuildByMap") }
+        Assertions.assertNotNull(builderMethod)
+
+        // Cast to not null.
+        builderMethod!!
+
+        // Assert that the return type of the function is the generated type.
+        Assertions.assertEquals(generatedClass.starProjectedType, builderMethod.returnType)
+
+        val builderMethodParameters = builderMethod.parameters
+        val mapBuilderMethodParameter = builderMethodParameters.firstOrNull { it.name == "map" }
+        Assertions.assertNotNull(mapBuilderMethodParameter)
+    }
+}
+
+val testInputObject = GraphQLInputObjectType.newInputObject()
+    .name("TestInput")
+    .field(
+        GraphQLInputObjectField.newInputObjectField()
+            .name("name")
+            .type(GraphQLNonNull(Scalars.GraphQLString))
+            .build()
+    )
+    .field(
+        GraphQLInputObjectField.newInputObjectField()
+            .name("number")
+            .type(Scalars.GraphQLInt)
+            .build()
+    )
+    .build()

graphql-kotlin-toolkit-codegen/src/test/kotlin/com/auritylab/graphql/kotlin/toolkit/codegen/generator/ObjectTypeGeneratorTest.kt

@@ -0,0 +1,53 @@
+package com.auritylab.graphql.kotlin.toolkit.codegen.generator
+
+import com.auritylab.graphql.kotlin.toolkit.codegen._test.AbstractCompilationTest
+import com.auritylab.graphql.kotlin.toolkit.codegen._test.TestObject
+import graphql.Scalars
+import graphql.schema.GraphQLFieldDefinition
+import graphql.schema.GraphQLObjectType
+import org.junit.jupiter.api.Assertions
+import org.junit.jupiter.api.BeforeAll
+import org.junit.jupiter.api.Test
+import org.junit.jupiter.api.TestInstance
+import kotlin.reflect.KClass
+import kotlin.reflect.full.memberProperties
+
+@TestInstance(TestInstance.Lifecycle.PER_CLASS)
+internal class ObjectTypeGeneratorTest : AbstractCompilationTest() {
+    val generator = ObjectTypeGenerator(
+        testObjectType,
+        TestObject.options,
+        TestObject.kotlinTypeMapper,
+        TestObject.generatedMapper
+    )
+
+    lateinit var generatedClass: KClass<*>
+
+    @BeforeAll
+    fun compileCode() {
+        // Compile the code of the generator
+        generatedClass = compile(generator)
+    }
+
+    @Test
+    fun `should generate properties correctly`() {
+        val memberProperties = generatedClass.memberProperties.toList()
+        Assertions.assertEquals(2, memberProperties.size)
+
+        Assertions.assertEquals("name", memberProperties[0].name)
+        Assertions.assertEquals("number", memberProperties[1].name)
+    }
+}
+
+val testObjectType = GraphQLObjectType.newObject()
+    .name("TestObjectType")
+    .field(
+        GraphQLFieldDefinition.newFieldDefinition()
+            .name("name")
+            .type(Scalars.GraphQLString)
+    ).field(
+        GraphQLFieldDefinition.newFieldDefinition()
+            .name("number")
+            .type(Scalars.GraphQLInt)
+    )
+    .build()

graphql-kotlin-toolkit-codegen/src/test/kotlin/com/auritylab/graphql/kotlin/toolkit/codegen/generator/ValueWrapperGeneratorTest.kt

@@ -0,0 +1,31 @@
+package com.auritylab.graphql.kotlin.toolkit.codegen.generator
+
+import com.auritylab.graphql.kotlin.toolkit.codegen._test.AbstractCompilationTest
+import com.auritylab.graphql.kotlin.toolkit.codegen._test.TestObject
+import com.squareup.kotlinpoet.asTypeName
+import com.squareup.kotlinpoet.asTypeVariableName
+import org.junit.jupiter.api.Assertions
+import org.junit.jupiter.api.Test
+import kotlin.reflect.full.memberProperties
+
+internal class ValueWrapperGeneratorTest: AbstractCompilationTest() {
+    @Test
+    fun shouldGenerateCompilableCode() {
+        val generator =
+            ValueWrapperGenerator(TestObject.options, TestObject.kotlinTypeMapper, TestObject.generatedMapper)
+
+        // Compile the generated code.
+        val generatedClass = compile(generator)
+
+        // Assert against the member properties.
+        val memberProperties = generatedClass.memberProperties
+        Assertions.assertEquals(1, memberProperties.size)
+        Assertions.assertEquals("value", memberProperties.first().name)
+
+        val typeParameters = generatedClass.typeParameters
+        Assertions.assertEquals(1, typeParameters.size)
+
+        // Assert that the return type of the property is equal to the type parameter of the class.
+        Assertions.assertEquals(typeParameters[0].asTypeVariableName(), memberProperties.first().returnType.asTypeName())
+    }
+}