chore: update java microgenerator to 1.0.2 Adds request initializatio…

…n to sample code (#938) Committer: @miraleung PiperOrigin-RevId: 362856902 Source-Author: Google APIs <noreply@google.com> Source-Date: Sun Mar 14 20:47:20 2021 -0700 Source-Repo: googleapis/googleapis Source-Sha: 1305ca41d554eb0725237561e34129373bb8cbc1 Source-Link: googleapis/googleapis@1305ca4
googleapis · Mar 15, 2021 · 050f855 · 050f855
1 parent 4a09405
commit 050f855
Show file tree

Hide file tree

Showing 6 changed files with 53 additions and 53 deletions.
diff --git a/...querystorage/src/test/java/com/google/cloud/bigquery/storage/v1/MockBigQueryReadImpl.java b/...querystorage/src/test/java/com/google/cloud/bigquery/storage/v1/MockBigQueryReadImpl.java
@@ -61,7 +61,7 @@ public void reset() {
   @Override
   public void createReadSession(
       CreateReadSessionRequest request, StreamObserver<ReadSession> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof ReadSession) {
       requests.add(request);
       responseObserver.onNext(((ReadSession) response));
@@ -73,15 +73,15 @@ public void createReadSession(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method CreateReadSession, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   ReadSession.class.getName(),
                   Exception.class.getName())));
     }
   }
 
   @Override
   public void readRows(ReadRowsRequest request, StreamObserver<ReadRowsResponse> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof ReadRowsResponse) {
       requests.add(request);
       responseObserver.onNext(((ReadRowsResponse) response));
@@ -93,7 +93,7 @@ public void readRows(ReadRowsRequest request, StreamObserver<ReadRowsResponse> r
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method ReadRows, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   ReadRowsResponse.class.getName(),
                   Exception.class.getName())));
     }
@@ -102,7 +102,7 @@ public void readRows(ReadRowsRequest request, StreamObserver<ReadRowsResponse> r
   @Override
   public void splitReadStream(
       SplitReadStreamRequest request, StreamObserver<SplitReadStreamResponse> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof SplitReadStreamResponse) {
       requests.add(request);
       responseObserver.onNext(((SplitReadStreamResponse) response));
@@ -114,7 +114,7 @@ public void splitReadStream(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method SplitReadStream, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   SplitReadStreamResponse.class.getName(),
                   Exception.class.getName())));
     }

diff --git a/...orage/src/test/java/com/google/cloud/bigquery/storage/v1alpha2/MockBigQueryWriteImpl.java b/...orage/src/test/java/com/google/cloud/bigquery/storage/v1alpha2/MockBigQueryWriteImpl.java
@@ -62,7 +62,7 @@ public void reset() {
   public void createWriteStream(
       Storage.CreateWriteStreamRequest request,
       StreamObserver<Stream.WriteStream> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof Stream.WriteStream) {
       requests.add(request);
       responseObserver.onNext(((Stream.WriteStream) response));
@@ -74,7 +74,7 @@ public void createWriteStream(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method CreateWriteStream, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   Stream.WriteStream.class.getName(),
                   Exception.class.getName())));
     }
@@ -98,7 +98,7 @@ public void onNext(Storage.AppendRowsRequest value) {
                   new IllegalArgumentException(
                       String.format(
                           "Unrecognized response type %s for method AppendRows, expected %s or %s",
-                          response.getClass().getName(),
+                          response == null ? "null" : response.getClass().getName(),
                           Storage.AppendRowsResponse.class.getName(),
                           Exception.class.getName())));
             }
@@ -120,7 +120,7 @@ public void onCompleted() {
   @Override
   public void getWriteStream(
       Storage.GetWriteStreamRequest request, StreamObserver<Stream.WriteStream> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof Stream.WriteStream) {
       requests.add(request);
       responseObserver.onNext(((Stream.WriteStream) response));
@@ -132,7 +132,7 @@ public void getWriteStream(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method GetWriteStream, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   Stream.WriteStream.class.getName(),
                   Exception.class.getName())));
     }
@@ -142,7 +142,7 @@ public void getWriteStream(
   public void finalizeWriteStream(
       Storage.FinalizeWriteStreamRequest request,
       StreamObserver<Storage.FinalizeWriteStreamResponse> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof Storage.FinalizeWriteStreamResponse) {
       requests.add(request);
       responseObserver.onNext(((Storage.FinalizeWriteStreamResponse) response));
@@ -154,7 +154,7 @@ public void finalizeWriteStream(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method FinalizeWriteStream, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   Storage.FinalizeWriteStreamResponse.class.getName(),
                   Exception.class.getName())));
     }
@@ -164,7 +164,7 @@ public void finalizeWriteStream(
   public void batchCommitWriteStreams(
       Storage.BatchCommitWriteStreamsRequest request,
       StreamObserver<Storage.BatchCommitWriteStreamsResponse> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof Storage.BatchCommitWriteStreamsResponse) {
       requests.add(request);
       responseObserver.onNext(((Storage.BatchCommitWriteStreamsResponse) response));
@@ -176,7 +176,7 @@ public void batchCommitWriteStreams(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method BatchCommitWriteStreams, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   Storage.BatchCommitWriteStreamsResponse.class.getName(),
                   Exception.class.getName())));
     }
@@ -186,7 +186,7 @@ public void batchCommitWriteStreams(
   public void flushRows(
       Storage.FlushRowsRequest request,
       StreamObserver<Storage.FlushRowsResponse> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof Storage.FlushRowsResponse) {
       requests.add(request);
       responseObserver.onNext(((Storage.FlushRowsResponse) response));
@@ -198,7 +198,7 @@ public void flushRows(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method FlushRows, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   Storage.FlushRowsResponse.class.getName(),
                   Exception.class.getName())));
     }

diff --git a/...rage/src/test/java/com/google/cloud/bigquery/storage/v1beta1/MockBigQueryStorageImpl.java b/...rage/src/test/java/com/google/cloud/bigquery/storage/v1beta1/MockBigQueryStorageImpl.java
@@ -63,7 +63,7 @@ public void reset() {
   public void createReadSession(
       Storage.CreateReadSessionRequest request,
       StreamObserver<Storage.ReadSession> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof Storage.ReadSession) {
       requests.add(request);
       responseObserver.onNext(((Storage.ReadSession) response));
@@ -75,7 +75,7 @@ public void createReadSession(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method CreateReadSession, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   Storage.ReadSession.class.getName(),
                   Exception.class.getName())));
     }
@@ -84,7 +84,7 @@ public void createReadSession(
   @Override
   public void readRows(
       Storage.ReadRowsRequest request, StreamObserver<Storage.ReadRowsResponse> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof Storage.ReadRowsResponse) {
       requests.add(request);
       responseObserver.onNext(((Storage.ReadRowsResponse) response));
@@ -96,7 +96,7 @@ public void readRows(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method ReadRows, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   Storage.ReadRowsResponse.class.getName(),
                   Exception.class.getName())));
     }
@@ -106,7 +106,7 @@ public void readRows(
   public void batchCreateReadSessionStreams(
       Storage.BatchCreateReadSessionStreamsRequest request,
       StreamObserver<Storage.BatchCreateReadSessionStreamsResponse> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof Storage.BatchCreateReadSessionStreamsResponse) {
       requests.add(request);
       responseObserver.onNext(((Storage.BatchCreateReadSessionStreamsResponse) response));
@@ -118,7 +118,7 @@ public void batchCreateReadSessionStreams(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method BatchCreateReadSessionStreams, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   Storage.BatchCreateReadSessionStreamsResponse.class.getName(),
                   Exception.class.getName())));
     }
@@ -127,7 +127,7 @@ public void batchCreateReadSessionStreams(
   @Override
   public void finalizeStream(
       Storage.FinalizeStreamRequest request, StreamObserver<Empty> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof Empty) {
       requests.add(request);
       responseObserver.onNext(((Empty) response));
@@ -139,7 +139,7 @@ public void finalizeStream(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method FinalizeStream, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   Empty.class.getName(),
                   Exception.class.getName())));
     }
@@ -149,7 +149,7 @@ public void finalizeStream(
   public void splitReadStream(
       Storage.SplitReadStreamRequest request,
       StreamObserver<Storage.SplitReadStreamResponse> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof Storage.SplitReadStreamResponse) {
       requests.add(request);
       responseObserver.onNext(((Storage.SplitReadStreamResponse) response));
@@ -161,7 +161,7 @@ public void splitReadStream(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method SplitReadStream, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   Storage.SplitReadStreamResponse.class.getName(),
                   Exception.class.getName())));
     }

diff --git a/...storage/src/test/java/com/google/cloud/bigquery/storage/v1beta2/MockBigQueryReadImpl.java b/...storage/src/test/java/com/google/cloud/bigquery/storage/v1beta2/MockBigQueryReadImpl.java
@@ -61,7 +61,7 @@ public void reset() {
   @Override
   public void createReadSession(
       CreateReadSessionRequest request, StreamObserver<ReadSession> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof ReadSession) {
       requests.add(request);
       responseObserver.onNext(((ReadSession) response));
@@ -73,15 +73,15 @@ public void createReadSession(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method CreateReadSession, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   ReadSession.class.getName(),
                   Exception.class.getName())));
     }
   }
 
   @Override
   public void readRows(ReadRowsRequest request, StreamObserver<ReadRowsResponse> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof ReadRowsResponse) {
       requests.add(request);
       responseObserver.onNext(((ReadRowsResponse) response));
@@ -93,7 +93,7 @@ public void readRows(ReadRowsRequest request, StreamObserver<ReadRowsResponse> r
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method ReadRows, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   ReadRowsResponse.class.getName(),
                   Exception.class.getName())));
     }
@@ -102,7 +102,7 @@ public void readRows(ReadRowsRequest request, StreamObserver<ReadRowsResponse> r
   @Override
   public void splitReadStream(
       SplitReadStreamRequest request, StreamObserver<SplitReadStreamResponse> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof SplitReadStreamResponse) {
       requests.add(request);
       responseObserver.onNext(((SplitReadStreamResponse) response));
@@ -114,7 +114,7 @@ public void splitReadStream(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method SplitReadStream, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   SplitReadStreamResponse.class.getName(),
                   Exception.class.getName())));
     }

diff --git a/...torage/src/test/java/com/google/cloud/bigquery/storage/v1beta2/MockBigQueryWriteImpl.java b/...torage/src/test/java/com/google/cloud/bigquery/storage/v1beta2/MockBigQueryWriteImpl.java
@@ -61,7 +61,7 @@ public void reset() {
   @Override
   public void createWriteStream(
       CreateWriteStreamRequest request, StreamObserver<WriteStream> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof WriteStream) {
       requests.add(request);
       responseObserver.onNext(((WriteStream) response));
@@ -73,7 +73,7 @@ public void createWriteStream(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method CreateWriteStream, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   WriteStream.class.getName(),
                   Exception.class.getName())));
     }
@@ -97,7 +97,7 @@ public void onNext(AppendRowsRequest value) {
                   new IllegalArgumentException(
                       String.format(
                           "Unrecognized response type %s for method AppendRows, expected %s or %s",
-                          response.getClass().getName(),
+                          response == null ? "null" : response.getClass().getName(),
                           AppendRowsResponse.class.getName(),
                           Exception.class.getName())));
             }
@@ -119,7 +119,7 @@ public void onCompleted() {
   @Override
   public void getWriteStream(
       GetWriteStreamRequest request, StreamObserver<WriteStream> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof WriteStream) {
       requests.add(request);
       responseObserver.onNext(((WriteStream) response));
@@ -131,7 +131,7 @@ public void getWriteStream(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method GetWriteStream, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   WriteStream.class.getName(),
                   Exception.class.getName())));
     }
@@ -141,7 +141,7 @@ public void getWriteStream(
   public void finalizeWriteStream(
       FinalizeWriteStreamRequest request,
       StreamObserver<FinalizeWriteStreamResponse> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof FinalizeWriteStreamResponse) {
       requests.add(request);
       responseObserver.onNext(((FinalizeWriteStreamResponse) response));
@@ -153,7 +153,7 @@ public void finalizeWriteStream(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method FinalizeWriteStream, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   FinalizeWriteStreamResponse.class.getName(),
                   Exception.class.getName())));
     }
@@ -163,7 +163,7 @@ public void finalizeWriteStream(
   public void batchCommitWriteStreams(
       BatchCommitWriteStreamsRequest request,
       StreamObserver<BatchCommitWriteStreamsResponse> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof BatchCommitWriteStreamsResponse) {
       requests.add(request);
       responseObserver.onNext(((BatchCommitWriteStreamsResponse) response));
@@ -175,7 +175,7 @@ public void batchCommitWriteStreams(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method BatchCommitWriteStreams, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   BatchCommitWriteStreamsResponse.class.getName(),
                   Exception.class.getName())));
     }
@@ -184,7 +184,7 @@ public void batchCommitWriteStreams(
   @Override
   public void flushRows(
       FlushRowsRequest request, StreamObserver<FlushRowsResponse> responseObserver) {
-    Object response = responses.remove();
+    Object response = responses.poll();
     if (response instanceof FlushRowsResponse) {
       requests.add(request);
       responseObserver.onNext(((FlushRowsResponse) response));
@@ -196,7 +196,7 @@ public void flushRows(
           new IllegalArgumentException(
               String.format(
                   "Unrecognized response type %s for method FlushRows, expected %s or %s",
-                  response.getClass().getName(),
+                  response == null ? "null" : response.getClass().getName(),
                   FlushRowsResponse.class.getName(),
                   Exception.class.getName())));
     }