From 4128c51ea6ea9c8ef187d7d06a4f233945966b1f Mon Sep 17 00:00:00 2001
From: Jiho Lee <jiho.lee17@gmail.com>
Date: Thu, 25 Jun 2026 14:25:07 -0700
Subject: [PATCH] Fix tests to support generated by default

---
 .../kotlin/com/netflix/graphql/dgs/codegen/CodeGenTest.kt    | 5 +++--
 .../com/netflix/graphql/dgs/codegen/KotlinCodeGenTest.kt     | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/graphql-dgs-codegen-core/src/test/kotlin/com/netflix/graphql/dgs/codegen/CodeGenTest.kt b/graphql-dgs-codegen-core/src/test/kotlin/com/netflix/graphql/dgs/codegen/CodeGenTest.kt
index e527fa19..81662fe3 100644
--- a/graphql-dgs-codegen-core/src/test/kotlin/com/netflix/graphql/dgs/codegen/CodeGenTest.kt
+++ b/graphql-dgs-codegen-core/src/test/kotlin/com/netflix/graphql/dgs/codegen/CodeGenTest.kt
@@ -2238,7 +2238,8 @@ class CodeGenTest {
             }
             """.trimIndent()
 
-        val (dataTypes) = CodeGen(CodeGenConfig(schemas = setOf(schema), packageName = BASE_PACKAGE_NAME)).generate()
+        val codeGenResult = CodeGen(CodeGenConfig(schemas = setOf(schema), packageName = BASE_PACKAGE_NAME)).generate()
+        val (dataTypes) = codeGenResult
         assertThat(dataTypes).hasSize(1)
 
         val data = dataTypes[0]
@@ -2252,7 +2253,7 @@ class CodeGenTest {
         assertThat(maximumField).isNotNull
         assertThat(maximumField!!.initializer().toString()).isEqualTo("null")
 
-        assertCompilesJava(dataTypes)
+        assertCompilesJava(codeGenResult)
     }
 
     @Test
diff --git a/graphql-dgs-codegen-core/src/test/kotlin/com/netflix/graphql/dgs/codegen/KotlinCodeGenTest.kt b/graphql-dgs-codegen-core/src/test/kotlin/com/netflix/graphql/dgs/codegen/KotlinCodeGenTest.kt
index 4b2bdb39..bba3afdd 100644
--- a/graphql-dgs-codegen-core/src/test/kotlin/com/netflix/graphql/dgs/codegen/KotlinCodeGenTest.kt
+++ b/graphql-dgs-codegen-core/src/test/kotlin/com/netflix/graphql/dgs/codegen/KotlinCodeGenTest.kt
@@ -1833,7 +1833,7 @@ class KotlinCodeGenTest {
         assertThat(maximumParam!!.defaultValue).isNotNull
         assertThat(maximumParam.defaultValue.toString()).isEqualTo("null")
 
-        assertCompilesKotlin(dataTypes)
+        assertCompilesKotlin(codeGenResult)
     }
 
     @Test