fix: Attempt to start the publisher and publish in the ITVPC test (#420)

googleapis · Dec 11, 2020 · 8ebc9f1 · 8ebc9f1
1 parent 0c0d928
commit 8ebc9f1
Showing 1 changed file with 8 additions and 0 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
@@ -51,6 +51,7 @@
 import com.google.cloud.pubsublite.proto.Topic.RetentionConfig;
 import com.google.protobuf.FieldMask;
 import com.google.protobuf.util.Durations;
+import com.google.pubsub.v1.PubsubMessage;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.UUID;
@@ -320,9 +321,16 @@ public void deniedPublish() {
           PublisherSettings.newBuilder().setTopicPath(topicPath).build();
 
       Publisher publisher = Publisher.create(publisherSettings);
+      publisher.startAsync().awaitRunning();
+      publisher.publish(PubsubMessage.newBuilder().build());
+      publisher.awaitTerminated(30, TimeUnit.SECONDS);
       fail("Expected PERMISSION_DENIED CheckedApiException");
     } catch (ApiException e) {
       checkExceptionForVPCError(new CheckedApiException(e));
+    } catch (TimeoutException t) {
+      fail("Expected PERMISSION_DENIED CheckedApiException but got: " + t.toString());
+    } catch (IllegalStateException e) {
+      checkExceptionForVPCError(toCanonical(e.getCause()));
     }
   }