more tests (ref #373)

LogNet · Sep 4, 2023 · 11d1660 · 11d1660
1 parent 13bbfbc
commit 11d1660
Showing 1 changed file with 29 additions and 3 deletions.
diff --git a/...emo/src/test/java/org/lognet/springboot/grpc/recovery/GRpcStatusRuntimeExceptionTest.java b/...emo/src/test/java/org/lognet/springboot/grpc/recovery/GRpcStatusRuntimeExceptionTest.java
@@ -49,10 +49,17 @@ static class CustomService extends CustomServiceGrpc.CustomServiceImplBase {
 
             @Override
             public void anotherCustom(Custom.CustomRequest request, StreamObserver<Custom.CustomReply> responseObserver) {
-                if("NPE".equals(request.getName())){
+                if ("NPE".equals(request.getName())) {
                     ExceptionUtilsKt.throwException(new NullPointerException());
                 }
-                ExceptionUtilsKt.throwException(Status.FAILED_PRECONDITION);
+                if ("RT_ERROR".equals(request.getName())) {
+                    responseObserver.onError(Status.NOT_FOUND.asRuntimeException());
+                }
+                if ("ERROR".equals(request.getName())) {
+                    responseObserver.onError(Status.OUT_OF_RANGE.asException());
+                } else {
+                    ExceptionUtilsKt.throwException(Status.FAILED_PRECONDITION);
+                }
             }
 
             @Override
@@ -65,8 +72,9 @@ public void custom(Custom.CustomRequest request, StreamObserver<Custom.CustomRep
             public StreamObserver<Custom.CustomRequest> customStream(StreamObserver<Custom.CustomReply> responseObserver) {
                 throw new StatusRuntimeException(Status.FAILED_PRECONDITION);
             }
+
             @GRpcExceptionHandler
-            public Status  handle(NullPointerException e, GRpcExceptionScope scope ){
+            public Status handle(NullPointerException e, GRpcExceptionScope scope) {
                 return Status.DATA_LOSS;
             }
 
@@ -129,4 +137,22 @@ public void npeExceptionTest() {
         assertThat(statusRuntimeException.getStatus(), is(Status.DATA_LOSS));
     }
 
+    @Test
+    public void errorExceptionTest() {
+        final StatusRuntimeException statusRuntimeException = assertThrows(StatusRuntimeException.class, () ->
+                CustomServiceGrpc.newBlockingStub(getChannel()).anotherCustom(
+                        Custom.CustomRequest.newBuilder().setName("ERROR").build()
+                )
+        );
+        assertThat(statusRuntimeException.getStatus(), is(Status.OUT_OF_RANGE));
+    }
+    public void rtErrorExceptionTest() {
+        final StatusRuntimeException statusRuntimeException = assertThrows(StatusRuntimeException.class, () ->
+                CustomServiceGrpc.newBlockingStub(getChannel()).anotherCustom(
+                        Custom.CustomRequest.newBuilder().setName("RT_ERROR").build()
+                )
+        );
+        assertThat(statusRuntimeException.getStatus(), is(Status.NOT_FOUND));
+    }
+
 }