From b3f1ecf368fa1d77fd39399e32314b33ee0fe70b Mon Sep 17 00:00:00 2001
From: Robinson <robinson@dorkbox.com>
Date: Fri, 2 Jul 2021 16:18:40 +0200
Subject: [PATCH] Safer blocking calls

---
 .../network/rmi/RmiDelayedInvocationTest.kt    | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/test/dorkboxTest/network/rmi/RmiDelayedInvocationTest.kt b/test/dorkboxTest/network/rmi/RmiDelayedInvocationTest.kt
index 6d4ee74e..504d6b08 100644
--- a/test/dorkboxTest/network/rmi/RmiDelayedInvocationTest.kt
+++ b/test/dorkboxTest/network/rmi/RmiDelayedInvocationTest.kt
@@ -21,6 +21,7 @@ import dorkbox.network.Server
 import dorkbox.network.connection.Connection
 import dorkbox.network.serialization.Serialization
 import dorkboxTest.network.BaseTest
+import kotlinx.coroutines.runBlocking
 import org.junit.Test
 import java.util.concurrent.atomic.AtomicInteger
 
@@ -85,13 +86,16 @@ class RmiDelayedInvocationTest : BaseTest() {
 
                     // sometimes, this method is never called right away.
                     remoteObject.setOther(i.toFloat())
-                    synchronized(iterateLock) {
-                        try {
-                            (iterateLock as Object).wait(1)
-                        } catch (e: InterruptedException) {
-                            logger.error("Failed after: $i")
-                            e.printStackTrace()
-                            abort = true
+
+                    runBlocking {
+                        synchronized(iterateLock) {
+                            try {
+                                (iterateLock as Object).wait(1)
+                            } catch (e: InterruptedException) {
+                                logger.error("Failed after: $i")
+                                e.printStackTrace()
+                                abort = true
+                            }
                         }
                     }
                 }