Individual update snapshot func (#561)

* Add individual update snapshot functions

* Update tests

* Make private

test/src/test/kotlin/graphql/nadel/tests/next/UpdateTestSnapshots.kt

@@ -5,6 +5,7 @@ import com.squareup.kotlinpoet.AnnotationSpec
 import com.squareup.kotlinpoet.ClassName
 import com.squareup.kotlinpoet.CodeBlock
 import com.squareup.kotlinpoet.FileSpec
+import com.squareup.kotlinpoet.FunSpec
 import com.squareup.kotlinpoet.KModifier
 import com.squareup.kotlinpoet.MemberName
 import com.squareup.kotlinpoet.ParameterizedTypeName.Companion.parameterizedBy
@@ -25,14 +26,25 @@ import kotlin.reflect.KClass
 // For navigation so you can search up UpdateTestSnapshots
 private typealias UpdateTestSnapshots = Unit
 
-suspend fun main() {
+suspend inline fun <reified T : NadelIntegrationTest> update() {
+    return update(T::class)
+}
+
+suspend fun <T : NadelIntegrationTest> update(klass: KClass<T>) {
+    main(klass.qualifiedName!!)
+}
+
+private suspend fun main(vararg args: String) {
     val sourceRoot = File("test/src/test/kotlin/")
     require(sourceRoot.exists() && sourceRoot.isDirectory)
 
     getTestClassSequence()
         .onEach { klass ->
             println("Loading ${klass.qualifiedName}")
         }
+        .filter {
+            args.isEmpty() || args.contains(it.qualifiedName)
+        }
         .map {
             it to it.newInstanceNoArgConstructor()
         }
@@ -45,6 +57,7 @@ suspend fun main() {
             val outputFile = FileSpec.builder(ClassName.bestGuess(klass.qualifiedName!! + "Snapshot"))
                 .indent(' '.toString().repeat(4))
                 .addFileComment("@formatter:off")
+                .addFunction(makeUpdateSnapshotFunction(klass))
                 .addType(makeTestSnapshotClass(klass, captured))
                 .build()
                 .writeTo(sourceRoot)
@@ -65,6 +78,13 @@ suspend fun main() {
         }
 }
 
+fun makeUpdateSnapshotFunction(klass: KClass<NadelIntegrationTest>): FunSpec {
+    return FunSpec.builder("main")
+        .addModifiers(KModifier.PRIVATE, KModifier.SUSPEND)
+        .addCode("graphql.nadel.tests.next.update<%T>()", klass)
+        .build()
+}
+
 private fun getTestClassSequence(): Sequence<KClass<NadelIntegrationTest>> {
     return ClassPath.from(ClassLoader.getSystemClassLoader())
         .getTopLevelClassesRecursive("graphql.nadel.tests")

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/basic/BasicObjectSchemaTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<BasicObjectSchemaTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/basic/EchoTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<EchoTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/defer/ComprehensiveDeferQueryWithDifferentServiceCallsSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<ComprehensiveDeferQueryWithDifferentServiceCalls>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/defer/DeferOnListItemsTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<DeferOnListItemsTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/defer/DeferThrowsErrorTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<DeferThrowsErrorTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/defer/DeferWithIfFalseTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<DeferWithIfFalseTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/defer/DeferWithLabelTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<DeferWithLabelTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/defer/DeferWithoutLabelTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<DeferWithoutLabelTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/defer/MultipleDeferDirectivesTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<MultipleDeferDirectivesTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/defer/MultipleDeferWithDifferentServiceCallsSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<MultipleDeferWithDifferentServiceCalls>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/defer/MultipleFieldsinMultipleDeferDirectivesTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<MultipleFieldsinMultipleDeferDirectivesTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/defer/MultipleFieldsinSingleDeferDirectiveTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<MultipleFieldsinSingleDeferDirectiveTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/defer/NestedDefersTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<NestedDefersTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/hydration/HydrationTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<HydrationTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/hydration/defer/HydrationDeferFlagOffTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<HydrationDeferFlagOffTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/hydration/defer/HydrationDeferInRenamedFieldAndNestedHydrationsTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<HydrationDeferInRenamedFieldAndNestedHydrationsTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/hydration/defer/HydrationDeferInRenamedFieldTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<HydrationDeferInRenamedFieldTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/hydration/defer/HydrationDeferInRenamedFieldUsingRenamedFieldTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<HydrationDeferInRenamedFieldUsingRenamedFieldTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/hydration/defer/HydrationDeferIsDisabledForNestedHydrationsTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<HydrationDeferIsDisabledForNestedHydrationsTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/hydration/defer/HydrationDeferIsDisabledForRelatedIssuesTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<HydrationDeferIsDisabledForRelatedIssuesTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/hydration/defer/HydrationDeferIsDisabledInListOfRelatedIssuesForParentIssueTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<HydrationDeferIsDisabledInListOfRelatedIssuesForParentIssueTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/hydration/defer/HydrationDeferIsDisabledTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<HydrationDeferIsDisabledTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/hydration/defer/HydrationDeferTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<HydrationDeferTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/instrumentation/InstrumentationBeginExecuteOnCompleteOnFailureTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<InstrumentationBeginExecuteOnCompleteOnFailureTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/instrumentation/InstrumentationBeginExecuteOnCompleteOnSuccessTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<InstrumentationBeginExecuteOnCompleteOnSuccessTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/instrumentation/InstrumentationBeginQueryExecutionOnCompleteOnFailureTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<InstrumentationBeginQueryExecutionOnCompleteOnFailureTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *

test/src/test/kotlin/graphql/nadel/tests/next/fixtures/instrumentation/InstrumentationBeginQueryExecutionOnCompleteOnSuccessTestSnapshot.kt

@@ -9,6 +9,10 @@ import kotlin.Suppress
 import kotlin.collections.List
 import kotlin.collections.listOf
 
+private suspend fun main() {
+    graphql.nadel.tests.next.update<InstrumentationBeginQueryExecutionOnCompleteOnSuccessTest>()
+}
+
 /**
  * This class is generated. Do NOT modify.
  *