Skip to content
This repository was archived by the owner on Feb 24, 2026. It is now read-only.

Commit 437b29b

Browse files
yoshi-automationyoshi-automation
committed
chore: update java microgenerator to 1.0.2 Adds request initialization to sample code
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
1 parent 4a09405 commit 437b29b

File tree

6 files changed

+53
-53
lines changed

6 files changed

+53
-53
lines changed

‎google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/MockBigQueryReadImpl.java‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ public void reset() {
6161
@Override
6262
public void createReadSession(
6363
CreateReadSessionRequest request, StreamObserver<ReadSession> responseObserver) {
64-
Object response = responses.remove();
64+
Object response = responses.poll();
6565
if (response instanceof ReadSession) {
6666
requests.add(request);
6767
responseObserver.onNext(((ReadSession) response));
@@ -73,15 +73,15 @@ public void createReadSession(
7373
new IllegalArgumentException(
7474
String.format(
7575
"Unrecognized response type %s for method CreateReadSession, expected %s or %s",
76-
response.getClass().getName(),
76+
response == null ? "null" : response.getClass().getName(),
7777
ReadSession.class.getName(),
7878
Exception.class.getName())));
7979
}
8080
}
8181

8282
@Override
8383
public void readRows(ReadRowsRequest request, StreamObserver<ReadRowsResponse> responseObserver) {
84-
Object response = responses.remove();
84+
Object response = responses.poll();
8585
if (response instanceof ReadRowsResponse) {
8686
requests.add(request);
8787
responseObserver.onNext(((ReadRowsResponse) response));
@@ -93,7 +93,7 @@ public void readRows(ReadRowsRequest request, StreamObserver<ReadRowsResponse> r
9393
new IllegalArgumentException(
9494
String.format(
9595
"Unrecognized response type %s for method ReadRows, expected %s or %s",
96-
response.getClass().getName(),
96+
response == null ? "null" : response.getClass().getName(),
9797
ReadRowsResponse.class.getName(),
9898
Exception.class.getName())));
9999
}
@@ -102,7 +102,7 @@ public void readRows(ReadRowsRequest request, StreamObserver<ReadRowsResponse> r
102102
@Override
103103
public void splitReadStream(
104104
SplitReadStreamRequest request, StreamObserver<SplitReadStreamResponse> responseObserver) {
105-
Object response = responses.remove();
105+
Object response = responses.poll();
106106
if (response instanceof SplitReadStreamResponse) {
107107
requests.add(request);
108108
responseObserver.onNext(((SplitReadStreamResponse) response));
@@ -114,7 +114,7 @@ public void splitReadStream(
114114
new IllegalArgumentException(
115115
String.format(
116116
"Unrecognized response type %s for method SplitReadStream, expected %s or %s",
117-
response.getClass().getName(),
117+
response == null ? "null" : response.getClass().getName(),
118118
SplitReadStreamResponse.class.getName(),
119119
Exception.class.getName())));
120120
}

‎google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1alpha2/MockBigQueryWriteImpl.java‎

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ public void reset() {
6262
public void createWriteStream(
6363
Storage.CreateWriteStreamRequest request,
6464
StreamObserver<Stream.WriteStream> responseObserver) {
65-
Object response = responses.remove();
65+
Object response = responses.poll();
6666
if (response instanceof Stream.WriteStream) {
6767
requests.add(request);
6868
responseObserver.onNext(((Stream.WriteStream) response));
@@ -74,7 +74,7 @@ public void createWriteStream(
7474
new IllegalArgumentException(
7575
String.format(
7676
"Unrecognized response type %s for method CreateWriteStream, expected %s or %s",
77-
response.getClass().getName(),
77+
response == null ? "null" : response.getClass().getName(),
7878
Stream.WriteStream.class.getName(),
7979
Exception.class.getName())));
8080
}
@@ -98,7 +98,7 @@ public void onNext(Storage.AppendRowsRequest value) {
9898
new IllegalArgumentException(
9999
String.format(
100100
"Unrecognized response type %s for method AppendRows, expected %s or %s",
101-
response.getClass().getName(),
101+
response == null ? "null" : response.getClass().getName(),
102102
Storage.AppendRowsResponse.class.getName(),
103103
Exception.class.getName())));
104104
}
@@ -120,7 +120,7 @@ public void onCompleted() {
120120
@Override
121121
public void getWriteStream(
122122
Storage.GetWriteStreamRequest request, StreamObserver<Stream.WriteStream> responseObserver) {
123-
Object response = responses.remove();
123+
Object response = responses.poll();
124124
if (response instanceof Stream.WriteStream) {
125125
requests.add(request);
126126
responseObserver.onNext(((Stream.WriteStream) response));
@@ -132,7 +132,7 @@ public void getWriteStream(
132132
new IllegalArgumentException(
133133
String.format(
134134
"Unrecognized response type %s for method GetWriteStream, expected %s or %s",
135-
response.getClass().getName(),
135+
response == null ? "null" : response.getClass().getName(),
136136
Stream.WriteStream.class.getName(),
137137
Exception.class.getName())));
138138
}
@@ -142,7 +142,7 @@ public void getWriteStream(
142142
public void finalizeWriteStream(
143143
Storage.FinalizeWriteStreamRequest request,
144144
StreamObserver<Storage.FinalizeWriteStreamResponse> responseObserver) {
145-
Object response = responses.remove();
145+
Object response = responses.poll();
146146
if (response instanceof Storage.FinalizeWriteStreamResponse) {
147147
requests.add(request);
148148
responseObserver.onNext(((Storage.FinalizeWriteStreamResponse) response));
@@ -154,7 +154,7 @@ public void finalizeWriteStream(
154154
new IllegalArgumentException(
155155
String.format(
156156
"Unrecognized response type %s for method FinalizeWriteStream, expected %s or %s",
157-
response.getClass().getName(),
157+
response == null ? "null" : response.getClass().getName(),
158158
Storage.FinalizeWriteStreamResponse.class.getName(),
159159
Exception.class.getName())));
160160
}
@@ -164,7 +164,7 @@ public void finalizeWriteStream(
164164
public void batchCommitWriteStreams(
165165
Storage.BatchCommitWriteStreamsRequest request,
166166
StreamObserver<Storage.BatchCommitWriteStreamsResponse> responseObserver) {
167-
Object response = responses.remove();
167+
Object response = responses.poll();
168168
if (response instanceof Storage.BatchCommitWriteStreamsResponse) {
169169
requests.add(request);
170170
responseObserver.onNext(((Storage.BatchCommitWriteStreamsResponse) response));
@@ -176,7 +176,7 @@ public void batchCommitWriteStreams(
176176
new IllegalArgumentException(
177177
String.format(
178178
"Unrecognized response type %s for method BatchCommitWriteStreams, expected %s or %s",
179-
response.getClass().getName(),
179+
response == null ? "null" : response.getClass().getName(),
180180
Storage.BatchCommitWriteStreamsResponse.class.getName(),
181181
Exception.class.getName())));
182182
}
@@ -186,7 +186,7 @@ public void batchCommitWriteStreams(
186186
public void flushRows(
187187
Storage.FlushRowsRequest request,
188188
StreamObserver<Storage.FlushRowsResponse> responseObserver) {
189-
Object response = responses.remove();
189+
Object response = responses.poll();
190190
if (response instanceof Storage.FlushRowsResponse) {
191191
requests.add(request);
192192
responseObserver.onNext(((Storage.FlushRowsResponse) response));
@@ -198,7 +198,7 @@ public void flushRows(
198198
new IllegalArgumentException(
199199
String.format(
200200
"Unrecognized response type %s for method FlushRows, expected %s or %s",
201-
response.getClass().getName(),
201+
response == null ? "null" : response.getClass().getName(),
202202
Storage.FlushRowsResponse.class.getName(),
203203
Exception.class.getName())));
204204
}

‎google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1beta1/MockBigQueryStorageImpl.java‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ public void reset() {
6363
public void createReadSession(
6464
Storage.CreateReadSessionRequest request,
6565
StreamObserver<Storage.ReadSession> responseObserver) {
66-
Object response = responses.remove();
66+
Object response = responses.poll();
6767
if (response instanceof Storage.ReadSession) {
6868
requests.add(request);
6969
responseObserver.onNext(((Storage.ReadSession) response));
@@ -75,7 +75,7 @@ public void createReadSession(
7575
new IllegalArgumentException(
7676
String.format(
7777
"Unrecognized response type %s for method CreateReadSession, expected %s or %s",
78-
response.getClass().getName(),
78+
response == null ? "null" : response.getClass().getName(),
7979
Storage.ReadSession.class.getName(),
8080
Exception.class.getName())));
8181
}
@@ -84,7 +84,7 @@ public void createReadSession(
8484
@Override
8585
public void readRows(
8686
Storage.ReadRowsRequest request, StreamObserver<Storage.ReadRowsResponse> responseObserver) {
87-
Object response = responses.remove();
87+
Object response = responses.poll();
8888
if (response instanceof Storage.ReadRowsResponse) {
8989
requests.add(request);
9090
responseObserver.onNext(((Storage.ReadRowsResponse) response));
@@ -96,7 +96,7 @@ public void readRows(
9696
new IllegalArgumentException(
9797
String.format(
9898
"Unrecognized response type %s for method ReadRows, expected %s or %s",
99-
response.getClass().getName(),
99+
response == null ? "null" : response.getClass().getName(),
100100
Storage.ReadRowsResponse.class.getName(),
101101
Exception.class.getName())));
102102
}
@@ -106,7 +106,7 @@ public void readRows(
106106
public void batchCreateReadSessionStreams(
107107
Storage.BatchCreateReadSessionStreamsRequest request,
108108
StreamObserver<Storage.BatchCreateReadSessionStreamsResponse> responseObserver) {
109-
Object response = responses.remove();
109+
Object response = responses.poll();
110110
if (response instanceof Storage.BatchCreateReadSessionStreamsResponse) {
111111
requests.add(request);
112112
responseObserver.onNext(((Storage.BatchCreateReadSessionStreamsResponse) response));
@@ -118,7 +118,7 @@ public void batchCreateReadSessionStreams(
118118
new IllegalArgumentException(
119119
String.format(
120120
"Unrecognized response type %s for method BatchCreateReadSessionStreams, expected %s or %s",
121-
response.getClass().getName(),
121+
response == null ? "null" : response.getClass().getName(),
122122
Storage.BatchCreateReadSessionStreamsResponse.class.getName(),
123123
Exception.class.getName())));
124124
}
@@ -127,7 +127,7 @@ public void batchCreateReadSessionStreams(
127127
@Override
128128
public void finalizeStream(
129129
Storage.FinalizeStreamRequest request, StreamObserver<Empty> responseObserver) {
130-
Object response = responses.remove();
130+
Object response = responses.poll();
131131
if (response instanceof Empty) {
132132
requests.add(request);
133133
responseObserver.onNext(((Empty) response));
@@ -139,7 +139,7 @@ public void finalizeStream(
139139
new IllegalArgumentException(
140140
String.format(
141141
"Unrecognized response type %s for method FinalizeStream, expected %s or %s",
142-
response.getClass().getName(),
142+
response == null ? "null" : response.getClass().getName(),
143143
Empty.class.getName(),
144144
Exception.class.getName())));
145145
}
@@ -149,7 +149,7 @@ public void finalizeStream(
149149
public void splitReadStream(
150150
Storage.SplitReadStreamRequest request,
151151
StreamObserver<Storage.SplitReadStreamResponse> responseObserver) {
152-
Object response = responses.remove();
152+
Object response = responses.poll();
153153
if (response instanceof Storage.SplitReadStreamResponse) {
154154
requests.add(request);
155155
responseObserver.onNext(((Storage.SplitReadStreamResponse) response));
@@ -161,7 +161,7 @@ public void splitReadStream(
161161
new IllegalArgumentException(
162162
String.format(
163163
"Unrecognized response type %s for method SplitReadStream, expected %s or %s",
164-
response.getClass().getName(),
164+
response == null ? "null" : response.getClass().getName(),
165165
Storage.SplitReadStreamResponse.class.getName(),
166166
Exception.class.getName())));
167167
}

‎google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1beta2/MockBigQueryReadImpl.java‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ public void reset() {
6161
@Override
6262
public void createReadSession(
6363
CreateReadSessionRequest request, StreamObserver<ReadSession> responseObserver) {
64-
Object response = responses.remove();
64+
Object response = responses.poll();
6565
if (response instanceof ReadSession) {
6666
requests.add(request);
6767
responseObserver.onNext(((ReadSession) response));
@@ -73,15 +73,15 @@ public void createReadSession(
7373
new IllegalArgumentException(
7474
String.format(
7575
"Unrecognized response type %s for method CreateReadSession, expected %s or %s",
76-
response.getClass().getName(),
76+
response == null ? "null" : response.getClass().getName(),
7777
ReadSession.class.getName(),
7878
Exception.class.getName())));
7979
}
8080
}
8181

8282
@Override
8383
public void readRows(ReadRowsRequest request, StreamObserver<ReadRowsResponse> responseObserver) {
84-
Object response = responses.remove();
84+
Object response = responses.poll();
8585
if (response instanceof ReadRowsResponse) {
8686
requests.add(request);
8787
responseObserver.onNext(((ReadRowsResponse) response));
@@ -93,7 +93,7 @@ public void readRows(ReadRowsRequest request, StreamObserver<ReadRowsResponse> r
9393
new IllegalArgumentException(
9494
String.format(
9595
"Unrecognized response type %s for method ReadRows, expected %s or %s",
96-
response.getClass().getName(),
96+
response == null ? "null" : response.getClass().getName(),
9797
ReadRowsResponse.class.getName(),
9898
Exception.class.getName())));
9999
}
@@ -102,7 +102,7 @@ public void readRows(ReadRowsRequest request, StreamObserver<ReadRowsResponse> r
102102
@Override
103103
public void splitReadStream(
104104
SplitReadStreamRequest request, StreamObserver<SplitReadStreamResponse> responseObserver) {
105-
Object response = responses.remove();
105+
Object response = responses.poll();
106106
if (response instanceof SplitReadStreamResponse) {
107107
requests.add(request);
108108
responseObserver.onNext(((SplitReadStreamResponse) response));
@@ -114,7 +114,7 @@ public void splitReadStream(
114114
new IllegalArgumentException(
115115
String.format(
116116
"Unrecognized response type %s for method SplitReadStream, expected %s or %s",
117-
response.getClass().getName(),
117+
response == null ? "null" : response.getClass().getName(),
118118
SplitReadStreamResponse.class.getName(),
119119
Exception.class.getName())));
120120
}

‎google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1beta2/MockBigQueryWriteImpl.java‎

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ public void reset() {
6161
@Override
6262
public void createWriteStream(
6363
CreateWriteStreamRequest request, StreamObserver<WriteStream> responseObserver) {
64-
Object response = responses.remove();
64+
Object response = responses.poll();
6565
if (response instanceof WriteStream) {
6666
requests.add(request);
6767
responseObserver.onNext(((WriteStream) response));
@@ -73,7 +73,7 @@ public void createWriteStream(
7373
new IllegalArgumentException(
7474
String.format(
7575
"Unrecognized response type %s for method CreateWriteStream, expected %s or %s",
76-
response.getClass().getName(),
76+
response == null ? "null" : response.getClass().getName(),
7777
WriteStream.class.getName(),
7878
Exception.class.getName())));
7979
}
@@ -97,7 +97,7 @@ public void onNext(AppendRowsRequest value) {
9797
new IllegalArgumentException(
9898
String.format(
9999
"Unrecognized response type %s for method AppendRows, expected %s or %s",
100-
response.getClass().getName(),
100+
response == null ? "null" : response.getClass().getName(),
101101
AppendRowsResponse.class.getName(),
102102
Exception.class.getName())));
103103
}
@@ -119,7 +119,7 @@ public void onCompleted() {
119119
@Override
120120
public void getWriteStream(
121121
GetWriteStreamRequest request, StreamObserver<WriteStream> responseObserver) {
122-
Object response = responses.remove();
122+
Object response = responses.poll();
123123
if (response instanceof WriteStream) {
124124
requests.add(request);
125125
responseObserver.onNext(((WriteStream) response));
@@ -131,7 +131,7 @@ public void getWriteStream(
131131
new IllegalArgumentException(
132132
String.format(
133133
"Unrecognized response type %s for method GetWriteStream, expected %s or %s",
134-
response.getClass().getName(),
134+
response == null ? "null" : response.getClass().getName(),
135135
WriteStream.class.getName(),
136136
Exception.class.getName())));
137137
}
@@ -141,7 +141,7 @@ public void getWriteStream(
141141
public void finalizeWriteStream(
142142
FinalizeWriteStreamRequest request,
143143
StreamObserver<FinalizeWriteStreamResponse> responseObserver) {
144-
Object response = responses.remove();
144+
Object response = responses.poll();
145145
if (response instanceof FinalizeWriteStreamResponse) {
146146
requests.add(request);
147147
responseObserver.onNext(((FinalizeWriteStreamResponse) response));
@@ -153,7 +153,7 @@ public void finalizeWriteStream(
153153
new IllegalArgumentException(
154154
String.format(
155155
"Unrecognized response type %s for method FinalizeWriteStream, expected %s or %s",
156-
response.getClass().getName(),
156+
response == null ? "null" : response.getClass().getName(),
157157
FinalizeWriteStreamResponse.class.getName(),
158158
Exception.class.getName())));
159159
}
@@ -163,7 +163,7 @@ public void finalizeWriteStream(
163163
public void batchCommitWriteStreams(
164164
BatchCommitWriteStreamsRequest request,
165165
StreamObserver<BatchCommitWriteStreamsResponse> responseObserver) {
166-
Object response = responses.remove();
166+
Object response = responses.poll();
167167
if (response instanceof BatchCommitWriteStreamsResponse) {
168168
requests.add(request);
169169
responseObserver.onNext(((BatchCommitWriteStreamsResponse) response));
@@ -175,7 +175,7 @@ public void batchCommitWriteStreams(
175175
new IllegalArgumentException(
176176
String.format(
177177
"Unrecognized response type %s for method BatchCommitWriteStreams, expected %s or %s",
178-
response.getClass().getName(),
178+
response == null ? "null" : response.getClass().getName(),
179179
BatchCommitWriteStreamsResponse.class.getName(),
180180
Exception.class.getName())));
181181
}
@@ -184,7 +184,7 @@ public void batchCommitWriteStreams(
184184
@Override
185185
public void flushRows(
186186
FlushRowsRequest request, StreamObserver<FlushRowsResponse> responseObserver) {
187-
Object response = responses.remove();
187+
Object response = responses.poll();
188188
if (response instanceof FlushRowsResponse) {
189189
requests.add(request);
190190
responseObserver.onNext(((FlushRowsResponse) response));
@@ -196,7 +196,7 @@ public void flushRows(
196196
new IllegalArgumentException(
197197
String.format(
198198
"Unrecognized response type %s for method FlushRows, expected %s or %s",
199-
response.getClass().getName(),
199+
response == null ? "null" : response.getClass().getName(),
200200
FlushRowsResponse.class.getName(),
201201
Exception.class.getName())));
202202
}

0 commit comments

Comments
 (0)