serverscanning/capnproto-java-rpc
16
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix RpcState tests

Browse source
This commit is contained in:
Vaci Koblizek 2020-10-23 15:23:49 +01:00
parent debfda7351
commit d6a163990e
1 changed files with 17 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
runtime/src/test/java/org/capnproto/RpcStateTest.java
View file

@ -25,8 +25,13 @@ public class RpcStateTest {
class TestConnection implements VatNetwork.Connection { class TestConnection implements VatNetwork.Connection {
private CompletableFuture<IncomingRpcMessage> nextIncomingMessage = new CompletableFuture<>();
private final CompletableFuture<java.lang.Void> disconnect = new CompletableFuture<>(); private final CompletableFuture<java.lang.Void> disconnect = new CompletableFuture<>();
public void setNextIncomingMessage(IncomingRpcMessage message) {
this.nextIncomingMessage.complete(message);
}
@Override @Override
public OutgoingRpcMessage newOutgoingMessage(int firstSegmentWordSize) { public OutgoingRpcMessage newOutgoingMessage(int firstSegmentWordSize) {
var message = new MessageBuilder(); var message = new MessageBuilder();
@ -51,7 +56,7 @@ public class RpcStateTest {
@Override @Override
public CompletableFuture<IncomingRpcMessage> receiveIncomingMessage() { public CompletableFuture<IncomingRpcMessage> receiveIncomingMessage() {
return null; return this.nextIncomingMessage;
} }
@Override @Override
@ -73,24 +78,24 @@ public class RpcStateTest {
@Before @Before
public void setUp() throws Exception { public void setUp() throws Exception {
connection = new TestConnection(); this.connection = new TestConnection();
bootstrapInterface = new Capability.Client(Capability.newNullCap()); this.bootstrapInterface = new Capability.Client(Capability.newNullCap());
rpc = new RpcState(bootstrapInterface, connection, connection.disconnect); this.rpc = new RpcState(bootstrapInterface, connection, connection.disconnect);
} }
@After @After
public void tearDown() throws Exception { public void tearDown() throws Exception {
connection = null; this.connection = null;
rpc = null; this.rpc = null;
sent.clear(); this.sent.clear();
} }
@Test @Test
public void handleUnimplemented() throws RpcException { public void handleUnimplemented() throws RpcException {
var msg = new TestMessage(); var msg = new TestMessage();
msg.builder.getRoot(RpcProtocol.Message.factory).initUnimplemented(); msg.builder.getRoot(RpcProtocol.Message.factory).initUnimplemented();
rpc.handleMessage(msg); this.connection.setNextIncomingMessage(msg);
Assert.assertTrue(sent.isEmpty()); Assert.assertFalse(sent.isEmpty());
} }
@Test @Test
@ -98,7 +103,8 @@ public class RpcStateTest {
var msg = new TestMessage(); var msg = new TestMessage();
var builder = msg.builder.getRoot(RpcProtocol.Message.factory); var builder = msg.builder.getRoot(RpcProtocol.Message.factory);
RpcException.fromException(RpcException.failed("Test abort"), builder.initAbort()); RpcException.fromException(RpcException.failed("Test abort"), builder.initAbort());
Assert.assertThrows(RpcException.class, () -> rpc.handleMessage(msg)); this.connection.setNextIncomingMessage(msg);
//Assert.assertThrows(RpcException.class, () -> rpc.handleMessage(msg));
} }
@Test @Test
@ -106,7 +112,7 @@ public class RpcStateTest {
var msg = new TestMessage(); var msg = new TestMessage();
var bootstrap = msg.builder.getRoot(RpcProtocol.Message.factory).initBootstrap(); var bootstrap = msg.builder.getRoot(RpcProtocol.Message.factory).initBootstrap();
bootstrap.setQuestionId(0); bootstrap.setQuestionId(0);
rpc.handleMessage(msg); this.connection.setNextIncomingMessage(msg);
Assert.assertFalse(sent.isEmpty()); Assert.assertFalse(sent.isEmpty());
var reply = sent.remove(); var reply = sent.remove();
var rpcMsg = reply.getBody().getAs(RpcProtocol.Message.factory); var rpcMsg = reply.getBody().getAs(RpcProtocol.Message.factory);