fix: Extract the rpc exception from IllegalStateExceptins in publish/…

…subscribe (#430)
googleapis · Dec 16, 2020 · ff05b1a · ff05b1a
1 parent 02aea67
commit ff05b1a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/google-cloud-pubsublite/src/test/java/com/google/cloud/pubsublite/it/ITVPCNegativeTest.java b/google-cloud-pubsublite/src/test/java/com/google/cloud/pubsublite/it/ITVPCNegativeTest.java
@@ -67,7 +67,6 @@
 
 /* Integration tests for VPC-SC */
 public class ITVPCNegativeTest {
-
   private static final boolean IS_VPCSC_TEST =
       System.getenv("GOOGLE_CLOUD_TESTS_IN_VPCSC") != null
           && System.getenv("GOOGLE_CLOUD_TESTS_IN_VPCSC").equalsIgnoreCase("true");
@@ -330,7 +329,7 @@ public void deniedPublish() {
     } catch (TimeoutException t) {
       fail("Expected PERMISSION_DENIED CheckedApiException but got: " + t.toString());
     } catch (IllegalStateException e) {
-      checkExceptionForVPCError(toCanonical(e));
+      checkExceptionForVPCError(toCanonical(e.getCause()));
     }
   }
 
@@ -367,7 +366,7 @@ public void deniedSubscriber() {
     } catch (TimeoutException t) {
       fail("Expected PERMISSION_DENIED CheckedApiException but got: " + t.toString());
     } catch (IllegalStateException e) {
-      checkExceptionForVPCError(toCanonical(e));
+      checkExceptionForVPCError(toCanonical(e.getCause()));
     }
   }
 }