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

Added unit test for manually starting/stopping the aeron media driver outside of the network

Browse Source
This commit is contained in:
nathan 2020-09-15 12:01:29 +02:00
parent 3c4cfe805f
commit 3b9eb00de9
1 changed files with 62 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
test/dorkboxTest/network/DisconnectReconnectTest.kt
View File

@ -2,6 +2,7 @@ package dorkboxTest.network
import dorkbox.network.Client
import dorkbox.network.Server
import dorkbox.network.aeron.AeronConfig
import dorkbox.network.connection.Connection
import kotlinx.atomicfu.atomic
import kotlinx.coroutines.delay
@ -68,6 +69,66 @@ class DisconnectReconnectTest : BaseTest() {
Assert.assertEquals(4, reconnectCount.value)
}
@Test
fun manualMediaDriverAndReconnectClient() {
val serverConfiguration = serverConfig()
val mediaDriver = runBlocking {
AeronConfig.startDriver(serverConfiguration)
}
run {
val server: Server<Connection> = Server(serverConfiguration)
addEndPoint(server)
server.bind()
server.onConnect { connection ->
connection.logger.error("Disconnecting after 2 seconds.")
delay(2000)
connection.logger.error("Disconnecting....")
connection.close()
}
}
run {
val config = clientConfig()
val client: Client<Connection> = Client(config)
addEndPoint(client)
client.onDisconnect { connection ->
connection.logger.error("Disconnected!")
val count = reconnectCount.getAndIncrement()
if (count == 3) {
connection.logger.error("Shutting down")
stopEndPoints()
}
else {
connection.logger.error("Reconnecting: $count")
try {
client.connect(LOOPBACK)
} catch (e: IOException) {
e.printStackTrace()
}
}
}
runBlocking {
client.connect(LOOPBACK)
}
}
waitForThreads()
AeronConfig.stopDriver(mediaDriver)
System.err.println("Connection count (after reconnecting) is: " + reconnectCount.value)
Assert.assertEquals(4, reconnectCount.value)
}
@Test
fun reconnectWithFallbackClient() {
run {
@ -127,7 +188,7 @@ class DisconnectReconnectTest : BaseTest() {
}
@Test
fun disconenctedMediaDriver() {
fun disconnectedMediaDriver() {
val server: Server<Connection>
run {
val config = serverConfig()