dorkbox/Network
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added easy ability to wait forever when unit testing

Browse Source
This commit is contained in:
Robinson 2021-04-29 11:03:47 +02:00
parent a3b5cbcc89
commit 62deb13d5a
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
test/dorkboxTest/network/BaseTest.kt
View File

@ -50,12 +50,12 @@ import dorkbox.os.OS
import dorkbox.util.entropy.Entropy import dorkbox.util.entropy.Entropy
import dorkbox.util.entropy.SimpleEntropy import dorkbox.util.entropy.SimpleEntropy
import dorkbox.util.exceptions.InitializationException import dorkbox.util.exceptions.InitializationException
import kotlinx.coroutines.delay
import kotlinx.coroutines.runBlocking import kotlinx.coroutines.runBlocking
import org.junit.After import org.junit.After
import org.junit.Assert import org.junit.Assert
import org.junit.Before import org.junit.Before
import org.slf4j.LoggerFactory import org.slf4j.LoggerFactory
import java.lang.Thread.sleep
import java.lang.reflect.Field import java.lang.reflect.Field
import java.lang.reflect.Method import java.lang.reflect.Method
import java.util.concurrent.CopyOnWriteArrayList import java.util.concurrent.CopyOnWriteArrayList
@ -171,7 +171,7 @@ abstract class BaseTest {
/** /**
* Immediately stop the endpoints * Immediately stop the endpoints
*/ */
suspend fun stopEndPoints(stopAfterMillis: Long = 0) { fun stopEndPoints(stopAfterMillis: Long = 0) {
if (isStopping) { if (isStopping) {
return return
} }
@ -179,7 +179,7 @@ abstract class BaseTest {
// not the best, but this works for our purposes. This is a TAD hacky, because we ALSO have to make sure that we // not the best, but this works for our purposes. This is a TAD hacky, because we ALSO have to make sure that we
// ARE NOT in the same thread group as netty! // ARE NOT in the same thread group as netty!
delay(stopAfterMillis) sleep(stopAfterMillis)
synchronized(lock) {} synchronized(lock) {}
@ -212,7 +212,12 @@ abstract class BaseTest {
fun waitForThreads(stopAfterSeconds: Long = AUTO_FAIL_TIMEOUT) { fun waitForThreads(stopAfterSeconds: Long = AUTO_FAIL_TIMEOUT) {
synchronized(lock) {} synchronized(lock) {}
try { try {
latch.await(stopAfterSeconds, TimeUnit.SECONDS) if (stopAfterSeconds == 0L) {
latch.await(Long.MAX_VALUE, TimeUnit.SECONDS)
} else {
latch.await(stopAfterSeconds, TimeUnit.SECONDS)
}
} catch (e: InterruptedException) { } catch (e: InterruptedException) {
e.printStackTrace() e.printStackTrace()
} }