From 06bbf18ca70982ce2f17db132a4634d5dffb1940 Mon Sep 17 00:00:00 2001
From: Pavel Punegov <32519625+PavelPunegov@users.noreply.github.com>
Date: Tue, 18 Jun 2019 14:57:14 +0300
Subject: [PATCH] Fix helper class for coroutines

---
 .../main/kotlin/org/jetbrains/kotlin/coroutineTestUtil.kt   | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/buildSrc/plugins/src/main/kotlin/org/jetbrains/kotlin/coroutineTestUtil.kt b/buildSrc/plugins/src/main/kotlin/org/jetbrains/kotlin/coroutineTestUtil.kt
index c397b112ea5..dacf96ae081 100644
--- a/buildSrc/plugins/src/main/kotlin/org/jetbrains/kotlin/coroutineTestUtil.kt
+++ b/buildSrc/plugins/src/main/kotlin/org/jetbrains/kotlin/coroutineTestUtil.kt
@@ -101,6 +101,12 @@ fun createTextForHelpers(isReleaseCoroutines: Boolean): String {
             |    var finished = false
             |
             |    var proceed: () -> Unit = {}
+            |    
+            |    fun reset() {
+            |        counter = 0
+            |        finished = false
+            |        proceed = {}
+            |    }
             |
             |    suspend fun suspendHere() = suspendCoroutine<Unit> { c ->
             |        counter++