299 lines
12 KiB
Kotlin
299 lines
12 KiB
Kotlin
/*
|
|
* Copyright 2020 dorkbox, llc
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
* Copyright (c) 2008, Nathan Sweet
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following
|
|
* conditions are met:
|
|
*
|
|
* - Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
|
|
* - Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following
|
|
* disclaimer in the documentation and/or other materials provided with the distribution.
|
|
* - Neither the name of Esoteric Software nor the names of its contributors may be used to endorse or promote products derived
|
|
* from this software without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
|
|
* BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT
|
|
* SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
|
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
package dorkboxTest.network
|
|
|
|
import ch.qos.logback.classic.Level
|
|
import ch.qos.logback.classic.Logger
|
|
import ch.qos.logback.classic.encoder.PatternLayoutEncoder
|
|
import ch.qos.logback.classic.joran.JoranConfigurator
|
|
import ch.qos.logback.classic.spi.ILoggingEvent
|
|
import ch.qos.logback.core.ConsoleAppender
|
|
import dorkbox.network.Client
|
|
import dorkbox.network.ClientConfiguration
|
|
import dorkbox.network.Configuration
|
|
import dorkbox.network.Server
|
|
import dorkbox.network.ServerConfiguration
|
|
import dorkbox.network.connection.EndPoint
|
|
import dorkbox.os.OS
|
|
import dorkbox.storage.Storage
|
|
import dorkbox.util.entropy.Entropy
|
|
import dorkbox.util.entropy.SimpleEntropy
|
|
import dorkbox.util.exceptions.InitializationException
|
|
import kotlinx.coroutines.runBlocking
|
|
import org.junit.After
|
|
import org.junit.Assert
|
|
import org.junit.Before
|
|
import org.slf4j.LoggerFactory
|
|
import java.io.File
|
|
import java.lang.Thread.sleep
|
|
import java.lang.reflect.Field
|
|
import java.lang.reflect.Method
|
|
import java.util.concurrent.*
|
|
|
|
abstract class BaseTest {
|
|
companion object {
|
|
const val LOCALHOST = "localhost"
|
|
|
|
// wait minimum of 3 minutes before we automatically fail the unit test.
|
|
var AUTO_FAIL_TIMEOUT: Long = 180L
|
|
|
|
init {
|
|
if (OS.javaVersion >= 9) {
|
|
// disableAccessWarnings
|
|
try {
|
|
val unsafeClass = Class.forName("sun.misc.Unsafe")
|
|
val field: Field = unsafeClass.getDeclaredField("theUnsafe")
|
|
field.isAccessible = true
|
|
val unsafe: Any = field.get(null)
|
|
val putObjectVolatile: Method = unsafeClass.getDeclaredMethod("putObjectVolatile", Any::class.java, Long::class.javaPrimitiveType, Any::class.java)
|
|
val staticFieldOffset: Method = unsafeClass.getDeclaredMethod("staticFieldOffset", Field::class.java)
|
|
val loggerClass = Class.forName("jdk.internal.module.IllegalAccessLogger")
|
|
val loggerField: Field = loggerClass.getDeclaredField("logger")
|
|
val offset = staticFieldOffset.invoke(unsafe, loggerField) as Long
|
|
putObjectVolatile.invoke(unsafe, loggerClass, offset, null)
|
|
} catch (ignored: Exception) {
|
|
}
|
|
}
|
|
|
|
// if (System.getProperty("logback.configurationFile") == null) {
|
|
// val file = File("logback.xml")
|
|
// if (file.canRead()) {
|
|
// System.setProperty("logback.configurationFile", file.toPath().toRealPath().toFile().toString())
|
|
// } else {
|
|
// System.setProperty("logback.configurationFile", "logback.xml")
|
|
// }
|
|
// }
|
|
|
|
setLogLevel(Level.TRACE)
|
|
// setLogLevel(Level.ERROR)
|
|
// setLogLevel(Level.DEBUG)
|
|
|
|
// we want our entropy generation to be simple (ie, no user interaction to generate)
|
|
try {
|
|
Entropy.init(SimpleEntropy::class.java)
|
|
} catch (e: InitializationException) {
|
|
e.printStackTrace()
|
|
}
|
|
}
|
|
|
|
fun clientConfig(block: Configuration.() -> Unit = {}): ClientConfiguration {
|
|
|
|
val configuration = ClientConfiguration()
|
|
configuration.settingsStore = Storage.Memory() // don't want to persist anything on disk!
|
|
configuration.port = 2200
|
|
|
|
configuration.enableIpc = false
|
|
|
|
block(configuration)
|
|
return configuration
|
|
}
|
|
|
|
fun serverConfig(block: ServerConfiguration.() -> Unit = {}): ServerConfiguration {
|
|
val configuration = ServerConfiguration()
|
|
configuration.settingsStore = Storage.Memory() // don't want to persist anything on disk!
|
|
configuration.port = 2200
|
|
|
|
configuration.enableIpc = false
|
|
|
|
configuration.maxClientCount = 50
|
|
configuration.maxConnectionsPerIpAddress = 50
|
|
|
|
block(configuration)
|
|
return configuration
|
|
}
|
|
|
|
fun setLogLevel(level: Level) {
|
|
println("Log level: $level")
|
|
|
|
// assume SLF4J is bound to logback in the current environment
|
|
val rootLogger = LoggerFactory.getLogger(org.slf4j.Logger.ROOT_LOGGER_NAME) as Logger
|
|
rootLogger.detachAndStopAllAppenders()
|
|
rootLogger.level = level
|
|
|
|
val context = rootLogger.loggerContext
|
|
|
|
val jc = JoranConfigurator()
|
|
jc.context = context
|
|
jc.doConfigure(File("logback.xml").absoluteFile)
|
|
context.reset() // override default configuration
|
|
|
|
|
|
// we only want error messages
|
|
val nettyLogger = LoggerFactory.getLogger("io.netty") as Logger
|
|
nettyLogger.level = Level.ERROR
|
|
|
|
// we only want error messages
|
|
val kryoLogger = LoggerFactory.getLogger("com.esotericsoftware") as Logger
|
|
kryoLogger.level = Level.ERROR
|
|
|
|
|
|
val encoder = PatternLayoutEncoder()
|
|
encoder.context = context
|
|
encoder.pattern = "%date{HH:mm:ss.SSS} %-5level [%logger{35}] %msg%n"
|
|
encoder.start()
|
|
val consoleAppender = ConsoleAppender<ILoggingEvent>()
|
|
consoleAppender.context = context
|
|
consoleAppender.encoder = encoder
|
|
consoleAppender.start()
|
|
rootLogger.addAppender(consoleAppender)
|
|
}
|
|
}
|
|
|
|
@Volatile
|
|
private var latch = CountDownLatch(1)
|
|
|
|
@Volatile
|
|
private var autoFailThread: Thread? = null
|
|
|
|
private val endPointConnections: MutableList<EndPoint<*>> = CopyOnWriteArrayList()
|
|
|
|
@Volatile
|
|
private var isStopping = false
|
|
|
|
init {
|
|
println("---- " + this.javaClass.simpleName)
|
|
|
|
// we must always make sure that aeron is shut-down before starting again.
|
|
while (Server.isRunning(serverConfig())) {
|
|
println("Aeron was still running. Waiting for it to stop...")
|
|
sleep(2000)
|
|
}
|
|
}
|
|
|
|
fun addEndPoint(endPointConnection: EndPoint<*>) {
|
|
endPointConnections.add(endPointConnection)
|
|
latch = CountDownLatch(endPointConnections.size + 1)
|
|
}
|
|
|
|
/**
|
|
* Immediately stop the endpoints
|
|
*/
|
|
fun stopEndPoints(stopAfterMillis: Long = 0L) {
|
|
if (isStopping) {
|
|
return
|
|
}
|
|
isStopping = true
|
|
|
|
// 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!
|
|
if (stopAfterMillis > 0L) {
|
|
sleep(stopAfterMillis)
|
|
}
|
|
|
|
// we start with "1", so make sure adjust if we want an accurate count
|
|
println("Shutting down ${endPointConnections.size} (${latch.count - 1}) endpoints...")
|
|
|
|
val remainingConnections = mutableListOf<EndPoint<*>>()
|
|
|
|
// shutdown clients first
|
|
endPointConnections.forEach { endPoint ->
|
|
if (endPoint is Client) {
|
|
endPoint.close()
|
|
latch.countDown()
|
|
println("Done closing: ${endPoint.type.simpleName}")
|
|
} else {
|
|
remainingConnections.add(endPoint)
|
|
}
|
|
}
|
|
|
|
// shutdown everything else (should only be servers) last
|
|
println("Shutting down ${remainingConnections.size} (${latch.count - 1}) endpoints...")
|
|
remainingConnections.forEach {
|
|
it.close()
|
|
latch.countDown()
|
|
}
|
|
|
|
// we start with "1", so make sure to end it
|
|
latch.countDown()
|
|
endPointConnections.clear()
|
|
}
|
|
/**
|
|
* Wait for network client/server threads to shutdown for the specified time. 0 will wait forever
|
|
*
|
|
* it should close as close to naturally as possible, otherwise there are problems
|
|
*
|
|
* @param stopAfterSeconds how many seconds to wait, the default is 2 minutes.
|
|
*/
|
|
fun waitForThreads(stopAfterSeconds: Long = AUTO_FAIL_TIMEOUT, preShutdownAction: () -> Unit = {}) {
|
|
val latchTriggered = try {
|
|
if (stopAfterSeconds == 0L) {
|
|
latch.await(Long.MAX_VALUE, TimeUnit.SECONDS)
|
|
} else {
|
|
latch.await(stopAfterSeconds, TimeUnit.SECONDS)
|
|
}
|
|
} catch (e: InterruptedException) {
|
|
e.printStackTrace()
|
|
false
|
|
}
|
|
|
|
// run actions before we actually shutdown, but after we wait
|
|
if (!latchTriggered) {
|
|
preShutdownAction()
|
|
}
|
|
|
|
// always stop the endpoints
|
|
stopEndPoints()
|
|
}
|
|
|
|
@Before
|
|
fun setupFailureCheck() {
|
|
autoFailThread = Thread(Runnable {
|
|
// 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!
|
|
try {
|
|
Thread.sleep(AUTO_FAIL_TIMEOUT * 1000L)
|
|
|
|
// if the thread is interrupted, then it means we finished the test.
|
|
System.err.println("Test did not complete in a timely manner...")
|
|
runBlocking {
|
|
stopEndPoints(0L)
|
|
}
|
|
Assert.fail("Test did not complete in a timely manner.")
|
|
} catch (ignored: InterruptedException) {
|
|
}
|
|
}, "UnitTest timeout fail condition")
|
|
autoFailThread!!.isDaemon = true
|
|
// autoFailThread.start();
|
|
}
|
|
|
|
@After
|
|
fun cancelFailureCheck() {
|
|
if (autoFailThread != null) {
|
|
autoFailThread!!.interrupt()
|
|
autoFailThread = null
|
|
}
|
|
}
|
|
}
|