Fix test case

pixelib · Aug 24, 2023 · 3785581 · 3785581
1 parent 47dfab4
commit 3785581
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/rpcnis-core/src/main/java/com/rpcnis/core/Rpcnis.java b/rpcnis-core/src/main/java/com/rpcnis/core/Rpcnis.java
@@ -10,6 +10,7 @@
 
 import java.util.Timer;
 import java.util.UUID;
+import java.util.concurrent.CompletionException;
 import java.util.concurrent.ConcurrentHashMap;
 
 public class Rpcnis {
@@ -51,7 +52,7 @@ public Rpcnis(RpcTransport transport) {
         this(new RpcOptions(), new GsonSerializer(), transport);
     }
 
-    public <T> T invoke(InvocationDescriptor invocationDescriptor, Class<T> returnType) throws InvocationTimedOutException {
+    public <T> T invoke(InvocationDescriptor invocationDescriptor, Class<T> returnType) throws Throwable {
         // create a pending invocation
         PendingInvocation<T> pendingInvocation = new PendingInvocation<>(this, invocationDescriptor, () -> {
             // remove the pending invocation from the map
@@ -64,7 +65,11 @@ public <T> T invoke(InvocationDescriptor invocationDescriptor, Class<T> returnTy
         // TODO: transmit
 
         // wait for response or timeout
-        return pendingInvocation.waitForResponse();
+        try {
+            return pendingInvocation.waitForResponse();
+        } catch (CompletionException e) {
+            throw e.getCause();
+        }
     }
 
     public void completeInvocation(InvocationDescriptor invocationDescriptor, Object value) {

diff --git a/rpcnis-core/src/main/java/com/rpcnis/core/proxy/ProxyInvocHandler.java b/rpcnis-core/src/main/java/com/rpcnis/core/proxy/ProxyInvocHandler.java
@@ -18,7 +18,7 @@ public ProxyInvocHandler(Rpcnis rpcnis, String targetName) {
     }
 
     @Override
-    public Object invoke(Object proxy, Method method, Object[] args) throws InvocationTimedOutException {
+    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
         // build invocation descriptor
         Class<?>[] argTypes = new Class<?>[args.length];
         for (int i = 0; i < args.length; i++)

diff --git a/rpcnis-core/src/test/java/com/npcnis/cor/invocations/PendingInvocationTest.java b/rpcnis-core/src/test/java/com/npcnis/cor/invocations/PendingInvocationTest.java
@@ -28,7 +28,7 @@ public static void tearDown() {
     }
 
     @Test
-    public void testPendingInvocation() {
+    public void testPendingInvocation() throws Throwable {
         // base instance
         Rpcnis rpcnis = new Rpcnis(new LoopbackTransport());
 
@@ -47,7 +47,6 @@ public void testPendingInvocation() {
         });
 
         String response = rpcnis.invoke(invocationDescriptor, String.class);
-
         assert response.equals(testString);
     }