Do not generate StateMachineChecker if CHECK_STATE_MACHINE directive is

not present.
Otherwise, since it uses suspend functions, it breaks IR tests.

compiler/testData/codegen/boxInline/suspend/stateMachine/innerLambda.kt

@@ -4,6 +4,7 @@
 // WITH_RUNTIME
 // WITH_COROUTINES
 // NO_CHECK_LAMBDA_INLINING
+// CHECK_STATE_MACHINE
 
 suspend inline fun crossinlineMe(crossinline c: suspend () -> Unit) {
     val l: suspend () -> Unit = {

compiler/testData/codegen/boxInline/suspend/stateMachine/innerLambdaInsideLambda.kt

@@ -4,6 +4,7 @@
 // WITH_RUNTIME
 // WITH_COROUTINES
 // NO_CHECK_LAMBDA_INLINING
+// CHECK_STATE_MACHINE
 
 suspend inline fun crossinlineMe(crossinline c: suspend () -> Unit) {
     val l: suspend () -> Unit = {

compiler/testData/codegen/boxInline/suspend/stateMachine/innerLambdaWithoutCrossinline.kt

@@ -4,6 +4,7 @@
 // WITH_RUNTIME
 // WITH_COROUTINES
 // NO_CHECK_LAMBDA_INLINING
+// CHECK_STATE_MACHINE
 
 import COROUTINES_PACKAGE.intrinsics.*
 import helpers.*

compiler/testData/codegen/boxInline/suspend/stateMachine/innerMadness.kt

@@ -4,6 +4,7 @@
 // WITH_RUNTIME
 // WITH_COROUTINES
 // NO_CHECK_LAMBDA_INLINING
+// CHECK_STATE_MACHINE
 
 interface SuspendRunnable {
     suspend fun run()

compiler/testData/codegen/boxInline/suspend/stateMachine/innerMadnessCallSite.kt

@@ -4,6 +4,7 @@
 // WITH_RUNTIME
 // WITH_COROUTINES
 // NO_CHECK_LAMBDA_INLINING
+// CHECK_STATE_MACHINE
 
 interface SuspendRunnable {
     suspend fun run()

compiler/testData/codegen/boxInline/suspend/stateMachine/innerObject.kt

@@ -4,6 +4,7 @@
 // WITH_RUNTIME
 // WITH_COROUTINES
 // NO_CHECK_LAMBDA_INLINING
+// CHECK_STATE_MACHINE
 
 interface SuspendRunnable {
     suspend fun run()

compiler/testData/codegen/boxInline/suspend/stateMachine/innerObjectInsideInnerObject.kt

@@ -4,6 +4,7 @@
 // WITH_RUNTIME
 // WITH_COROUTINES
 // NO_CHECK_LAMBDA_INLINING
+// CHECK_STATE_MACHINE
 
 interface SuspendRunnable {
     suspend fun run()

compiler/testData/codegen/boxInline/suspend/stateMachine/innerObjectSeveralFunctions.kt

@@ -4,6 +4,7 @@
 // WITH_RUNTIME
 // WITH_COROUTINES
 // NO_CHECK_LAMBDA_INLINING
+// CHECK_STATE_MACHINE
 
 interface SuspendRunnable {
     suspend fun run1()

compiler/testData/codegen/boxInline/suspend/stateMachine/innerObjectWithoutCapturingCrossinline.kt

@@ -4,6 +4,7 @@
 // WITH_RUNTIME
 // WITH_COROUTINES
 // NO_CHECK_LAMBDA_INLINING
+// CHECK_STATE_MACHINE
 
 import COROUTINES_PACKAGE.intrinsics.*
 import helpers.*

compiler/testData/codegen/boxInline/suspend/stateMachine/normalInline.kt

@@ -4,6 +4,7 @@
 // WITH_RUNTIME
 // WITH_COROUTINES
 // NO_CHECK_LAMBDA_INLINING
+// CHECK_STATE_MACHINE
 
 suspend inline fun crossinlineMe(crossinline c: suspend () -> Unit) {
     c()

compiler/testData/codegen/boxInline/suspend/stateMachine/numberOfSuspentions.kt

@@ -4,6 +4,7 @@
 // WITH_RUNTIME
 // WITH_COROUTINES
 // NO_CHECK_LAMBDA_INLINING
+// CHECK_STATE_MACHINE
 
 suspend inline fun crossinlineMe(crossinline c: suspend () -> Unit) {
     val l: suspend () -> Unit = { c() }

compiler/tests-common/tests/org/jetbrains/kotlin/coroutineTestUtil.kt

@@ -7,7 +7,7 @@ package org.jetbrains.kotlin
 
 import org.jetbrains.kotlin.resolve.DescriptorUtils
 
-fun createTextForHelpers(isReleaseCoroutines: Boolean): String {
+fun createTextForHelpers(isReleaseCoroutines: Boolean, checkStateMachine: Boolean): String {
     val coroutinesPackage =
         if (isReleaseCoroutines)
             DescriptorUtils.COROUTINES_PACKAGE_FQ_NAME_RELEASE.asString()
@@ -76,7 +76,7 @@ fun createTextForHelpers(isReleaseCoroutines: Boolean): String {
         else
             ""
 
-    val checkStateMachine = """
+    val checkStateMachineString = """
     object StateMachineChecker {
         private var counter = 0
         var finished = false
@@ -141,6 +141,6 @@ fun createTextForHelpers(isReleaseCoroutines: Boolean): String {
             |    $continuationAdapterBody
             |}
             |
-            |$checkStateMachine
+            |${if (checkStateMachine) checkStateMachineString else ""}
         """.trimMargin()
 }

compiler/tests-common/tests/org/jetbrains/kotlin/test/KotlinTestUtils.java

@@ -806,9 +806,11 @@ public class KotlinTestUtils {
                     !coroutinesPackage.contains("experimental") &&
                     !isDirectiveDefined(expectedText, "!LANGUAGE: -ReleaseCoroutines");
 
+            boolean checkStateMachine = isDirectiveDefined(expectedText, "CHECK_STATE_MACHINE");
+
             testFiles.add(factory.createFile(supportModule,
                                              "CoroutineUtil.kt",
-                                             CoroutineTestUtilKt.createTextForHelpers(isReleaseCoroutines),
+                                             CoroutineTestUtilKt.createTextForHelpers(isReleaseCoroutines, checkStateMachine),
                                              directives
             ));
         }

compiler/tests-common/tests/org/jetbrains/kotlin/test/KotlinTestUtils.java.as32

@@ -803,9 +803,11 @@ public class KotlinTestUtils {
                     !coroutinesPackage.contains("experimental") &&
                     !isDirectiveDefined(expectedText, "!LANGUAGE: -ReleaseCoroutines");
 
+            boolean checkStateMachine = isDirectiveDefined(expectedText, "CHECK_STATE_MACHINE");
+
             testFiles.add(factory.createFile(supportModule,
                                              "CoroutineUtil.kt",
-                                             CoroutineTestUtilKt.createTextForHelpers(isReleaseCoroutines),
+                                             CoroutineTestUtilKt.createTextForHelpers(isReleaseCoroutines, checkStateMachine),
                                              directives
             ));
         }