Add custom exception test.

gk-brown · gk-brown · commit 5087306d5a91 · 2021-03-05T18:10:04.000-05:00
diff --git a/httprpc-test/src/test/java/org/httprpc/WebServiceProxyTest.java b/httprpc-test/src/test/java/org/httprpc/WebServiceProxyTest.java
@@ -42,6 +42,7 @@
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
 public class WebServiceProxyTest {
@@ -172,6 +173,9 @@ enum Size {
         }
     }
 
+    public static class CustomException extends IOException {
+    }
+
     private URL serverURL;
 
     private DayOfWeek dayOfWeek = DayOfWeek.MONDAY;
@@ -466,8 +470,14 @@ public void testError() throws IOException {
     }
 
     @Test
-    public void testCustomErrorHandler() throws IOException {
-        // TODO
+    public void testCustomException() throws IOException {
+        WebServiceProxy webServiceProxy = new WebServiceProxy("GET", new URL(serverURL, "test/error"));
+
+        webServiceProxy.setErrorHandler((errorStream, contentType, statusCode) -> {
+            throw new CustomException();
+        });
+
+        assertThrows(CustomException.class, webServiceProxy::invoke);
     }
 
     @Test
