Add timeout to cds loader tests

icpc · Aug 30, 2023 · e451f37 · e451f37
1 parent 13f40e3
commit e451f37
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/cds/src/test/kotlin/org/icpclive/cds/CdsLoadersTest.kt b/src/cds/src/test/kotlin/org/icpclive/cds/CdsLoadersTest.kt
@@ -2,6 +2,7 @@ package org.icpclive.cds
 
 import kotlinx.coroutines.flow.first
 import kotlinx.coroutines.runBlocking
+import kotlinx.coroutines.withTimeout
 import kotlinx.serialization.encodeToString
 import kotlinx.serialization.json.Json
 import org.approvaltests.Approvals
@@ -16,6 +17,7 @@ import org.icpclive.cds.common.ContestParseResult
 import org.icpclive.cds.settings.*
 import kotlin.test.Test
 import kotlin.text.Regex
+import kotlin.time.Duration.Companion.minutes
 
 class CdsLoadersTest {
     @Test
@@ -103,12 +105,14 @@ class CdsLoadersTest {
     private fun loaderTest(args: CDSSettings, advanced: AdvancedProperties? = null) {
         val loader = args.toFlow(emptyMap())
         val result = runBlocking {
-            val result = loader.contestState().first { it.infoAfterEvent?.status == ContestStatus.FINALIZED }.let {
-                ContestParseResult(
-                    it.infoAfterEvent!!,
-                    it.runs.values.toList(),
-                    it.analyticsMessages.values.toList()
-                )
+            val result = withTimeout(1.minutes) {
+                loader.contestState().first { it.infoAfterEvent?.status == ContestStatus.FINALIZED }.let {
+                    ContestParseResult(
+                        it.infoAfterEvent!!,
+                        it.runs.values.toList(),
+                        it.analyticsMessages.values.toList()
+                    )
+                }
             }
             if (advanced != null) {
                 result.copy(