Network/test/dorkboxTest/network/rmi/RmiNestedTest.kt

432 lines
14 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.
*/
package dorkboxTest.network.rmi
import dorkbox.network.Client
import dorkbox.network.Server
import dorkbox.network.connection.Connection
import dorkboxTest.network.BaseTest
import kotlinx.coroutines.runBlocking
import org.junit.Assert
import org.junit.Test
import java.util.concurrent.atomic.AtomicInteger
@Suppress("unused", "RedundantSuspendModifier")
class RmiNestedTest : BaseTest() {
companion object {
private val idCounter = AtomicInteger()
}
/**
* In this test the server has two objects in an object space.
*
* The client uses the first remote object to get the second remote object.
*
*
* The MAJOR difference in this version, is that we use an interface to override the methods, so that we can have the RMI system pass
* in the connection object.
*
* Specifically, from CachedMethod.java
*
* In situations where we want to pass in the Connection (to an RMI method), we have to be able to override method A, with method B.
* This is to support calling RMI methods from an interface (that does pass the connection reference) to
* an implType, that DOES pass the connection reference. The remote side (that initiates the RMI calls), MUST use
* the interface, and the implType may override the method, so that we add the connection as the first in
* the list of parameters.
*
* for example:
* Interface: foo(String x)
* Impl: foo(Connection c, String x)
*
* The implType (if it exists, with the same name, and with the same signature + connection parameter) will be called from the interface
* instead of the method that would NORMALLY be called.
*/
@Test
fun biDirectionalDoubleRmi() {
run {
val configuration = serverConfig()
configuration.serialization.registerRmi(TestObject::class.java, TestObjectAnnotImpl::class.java)
configuration.serialization.registerRmi(OtherObject::class.java, OtherObjectImpl::class.java)
val server = Server<Connection>(configuration)
addEndPoint(server)
server.onMessage<OtherObject> { connection, message ->
// The test is complete when the client sends the OtherObject instance.
// this 'object' is the REAL object, not a proxy, because this object is created within this connection.
if (message.value() == 12.34f) {
stopEndPoints()
} else {
Assert.fail("Incorrect object value")
}
}
server.bind()
}
run {
val configuration = clientConfig()
val client = Client<Connection>(configuration)
addEndPoint(client)
client.onConnect { connection ->
connection.logger.error("Connected")
connection.createObject<TestObject> {
connection.logger.error("Starting test")
setValue(43.21f)
// Normal remote method call.
Assert.assertEquals(43.21f, other(), .0001f)
// Make a remote method call that returns another remote proxy object.
// the "test" object exists in the REMOTE side, as does the "OtherObject" that is created.
// here we have a proxy to both of them.
val otherObject: OtherObject = getOtherObject()
// Normal remote method call on the second object.
otherObject.setValue(12.34f)
val value = otherObject.value()
Assert.assertEquals(12.34f, value, .0001f)
// make sure the "local" object and the "remote" object have the same values
Assert.assertEquals(12.34f, getOtherValue(), .0001f)
// When a proxy object is sent, the other side receives its ACTUAL object (not a proxy of it), because
// that is where that object actually exists.
connection.send(otherObject)
}
}
runBlocking {
client.connect(LOOPBACK, 5000)
}
}
waitForThreads()
}
@Test
fun doubleRmi() {
run {
val configuration = serverConfig()
configuration.serialization.registerRmi(TestObject::class.java, TestObjectAnnotImpl::class.java)
configuration.serialization.registerRmi(OtherObject::class.java, OtherObjectImpl::class.java)
val server = Server<Connection>(configuration)
addEndPoint(server)
server.onMessage<OtherObject> { connection, message ->
// The test is complete when the client sends the OtherObject instance.
// this 'object' is the REAL object, not a proxy, because this object is created within this connection.
if (message.value() == 12.34f) {
stopEndPoints()
} else {
Assert.fail("Incorrect object value")
}
}
server.bind()
}
run {
val configuration = clientConfig()
val client = Client<Connection>(configuration)
addEndPoint(client)
client.onConnect { connection ->
connection.logger.error("Connected")
connection.createObject<TestObject> {
connection.logger.error("Starting test")
setValue(43.21f)
// Normal remote method call.
Assert.assertEquals(43.21f, other(), .0001f)
// Make a remote method call that returns another remote proxy object.
// the "test" object exists in the REMOTE side, as does the "OtherObject" that is created.
// here we have a proxy to both of them.
val otherObject: OtherObject = getOtherObject()
// Normal remote method call on the second object.
otherObject.setValue(12.34f)
val value = otherObject.value()
Assert.assertEquals(12.34f, value, .0001f)
// make sure the "local" object and the "remote" object have the same values
Assert.assertEquals(12.34f, getOtherValue(), .0001f)
// When a proxy object is sent, the other side receives its ACTUAL object (not a proxy of it), because
// that is where that object actually exists.
connection.send(otherObject)
}
}
runBlocking {
client.connect(LOOPBACK, 5000)
}
}
waitForThreads()
}
@Test
fun singleRmi() {
run {
val configuration = serverConfig()
configuration.serialization.registerRmi(TestObject::class.java, TestObjectImpl::class.java)
configuration.serialization.register(OtherObjectImpl::class.java)
val server = Server<Connection>(configuration)
addEndPoint(server)
server.onMessage<OtherObject> { connection, message ->
// The test is complete when the client sends the OtherObject instance.
// this 'object' is the REAL object
if (message.value() == 43.21f) {
stopEndPoints()
} else {
Assert.fail("Incorrect object value")
}
}
server.bind()
}
run {
val configuration = clientConfig()
val client = Client<Connection>(configuration)
addEndPoint(client)
client.onConnect { connection ->
connection.logger.error("Connected")
connection.createObject<TestObject> {
connection.logger.error("Starting test")
setOtherValue(43.21f)
// Normal remote method call.
Assert.assertEquals(43.21f, getOtherValue(), .0001f)
// real object
val otherObject: OtherObject = getOtherObject()
// Normal remote method call on the second object.
val value = otherObject.value()
Assert.assertEquals(43.21f, value, .0001f)
// When a proxy object is sent, the other side receives its ACTUAL object (not a proxy of it), because
// that is where that object actually exists.
connection.send(otherObject)
}
}
runBlocking {
client.connect(LOOPBACK, 5000)
}
}
waitForThreads()
}
@Test
fun singleReverseRmi() {
run {
val configuration = serverConfig()
configuration.serialization.registerRmi(TestObject::class.java, null)
configuration.serialization.register(OtherObjectImpl::class.java)
val server = Server<Connection>(configuration)
addEndPoint(server)
server.onConnect { connection ->
connection.logger.error("Connected")
connection.createObject<TestObject> {
connection.logger.error("Starting test")
setOtherValue(43.21f)
// Normal remote method call.
Assert.assertEquals(43.21f, getOtherValue(), .0001f)
// real object
val otherObject: OtherObject = getOtherObject()
// Normal remote method call on the second object.
val value = otherObject.value()
Assert.assertEquals(43.21f, value, .0001f)
// When a proxy object is sent, the other side receives its ACTUAL object (not a proxy of it), because
// that is where that object actually exists.
connection.send(otherObject)
}
}
server.bind()
}
run {
val configuration = clientConfig()
configuration.serialization.registerRmi(TestObject::class.java, TestObjectImpl::class.java)
val client = Client<Connection>(configuration)
addEndPoint(client)
client.onMessage<OtherObject> { connection, message ->
// The test is complete when the client sends the OtherObject instance.
// this 'object' is the REAL object
if (message.value() == 43.21f) {
stopEndPoints()
} else {
Assert.fail("Incorrect object value")
}
}
runBlocking {
client.connect(LOOPBACK, 5000)
}
}
waitForThreads()
}
private interface TestObject {
suspend fun setValue(aFloat: Float)
suspend fun setOtherValue(aFloat: Float)
suspend fun getOtherValue(): Float
fun other(): Float
fun getOtherObject(): OtherObject
}
private interface OtherObject {
fun setValue(aFloat: Float)
fun value(): Float
}
private class TestObjectImpl : TestObject {
@Transient
private val ID = idCounter.getAndIncrement()
private val otherObject: OtherObject = OtherObjectImpl()
private var aFloat = 0f
override suspend fun setValue(aFloat: Float) {
throw RuntimeException("Whoops!")
}
suspend fun setValue(connection: Connection, aFloat: Float) {
connection.logger.error("receiving")
this.aFloat = aFloat
}
override suspend fun setOtherValue(aFloat: Float) {
otherObject.setValue(aFloat)
}
override suspend fun getOtherValue(): Float {
return otherObject.value()
}
override fun other(): Float {
throw RuntimeException("Whoops!")
}
fun other(connection: Connection): Float {
return aFloat
}
override fun getOtherObject(): OtherObject {
return otherObject
}
override fun hashCode(): Int {
return ID
}
}
private class TestObjectAnnotImpl : TestObject {
@Transient
private val ID = idCounter.getAndIncrement()
private val otherObject: OtherObject = OtherObjectImpl()
private var aFloat = 0f
override suspend fun setValue(aFloat: Float) {
throw RuntimeException("Whoops!")
}
suspend fun setValue(connection: Connection, aFloat: Float) {
connection.logger.error("receiving")
this.aFloat = aFloat
}
override suspend fun setOtherValue(aFloat: Float) {
otherObject.setValue(aFloat)
}
override suspend fun getOtherValue(): Float {
return otherObject.value()
}
override fun other(): Float {
throw RuntimeException("Whoops!")
}
fun other(connection: Connection): Float {
return aFloat
}
override fun getOtherObject(): OtherObject {
return otherObject
}
override fun hashCode(): Int {
return ID
}
}
class OtherObjectImpl : OtherObject {
@Transient
private val ID = idCounter.getAndIncrement()
private var aFloat = 0f
override fun setValue(aFloat: Float) {
this.aFloat = aFloat
}
override fun value(): Float {
return aFloat
}
override fun hashCode(): Int {
return ID
}
}
}