1 /* 2 * Copyright 2022 Google LLC 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * https://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package io.grafeas.v1; 18 19 import com.google.api.core.BetaApi; 20 import com.google.protobuf.AbstractMessage; 21 import com.google.protobuf.Empty; 22 import io.grafeas.v1.GrafeasGrpc.GrafeasImplBase; 23 import io.grpc.stub.StreamObserver; 24 import java.util.ArrayList; 25 import java.util.LinkedList; 26 import java.util.List; 27 import java.util.Queue; 28 import javax.annotation.Generated; 29 30 @BetaApi 31 @Generated("by gapic-generator-java") 32 public class MockGrafeasImpl extends GrafeasImplBase { 33 private List<AbstractMessage> requests; 34 private Queue<Object> responses; 35 MockGrafeasImpl()36 public MockGrafeasImpl() { 37 requests = new ArrayList<>(); 38 responses = new LinkedList<>(); 39 } 40 getRequests()41 public List<AbstractMessage> getRequests() { 42 return requests; 43 } 44 addResponse(AbstractMessage response)45 public void addResponse(AbstractMessage response) { 46 responses.add(response); 47 } 48 setResponses(List<AbstractMessage> responses)49 public void setResponses(List<AbstractMessage> responses) { 50 this.responses = new LinkedList<Object>(responses); 51 } 52 addException(Exception exception)53 public void addException(Exception exception) { 54 responses.add(exception); 55 } 56 reset()57 public void reset() { 58 requests = new ArrayList<>(); 59 responses = new LinkedList<>(); 60 } 61 62 @Override getOccurrence( GetOccurrenceRequest request, StreamObserver<Occurrence> responseObserver)63 public void getOccurrence( 64 GetOccurrenceRequest request, StreamObserver<Occurrence> responseObserver) { 65 Object response = responses.poll(); 66 if (response instanceof Occurrence) { 67 requests.add(request); 68 responseObserver.onNext(((Occurrence) response)); 69 responseObserver.onCompleted(); 70 } else if (response instanceof Exception) { 71 responseObserver.onError(((Exception) response)); 72 } else { 73 responseObserver.onError( 74 new IllegalArgumentException( 75 String.format( 76 "Unrecognized response type %s for method GetOccurrence, expected %s or %s", 77 response == null ? "null" : response.getClass().getName(), 78 Occurrence.class.getName(), 79 Exception.class.getName()))); 80 } 81 } 82 83 @Override listOccurrences( ListOccurrencesRequest request, StreamObserver<ListOccurrencesResponse> responseObserver)84 public void listOccurrences( 85 ListOccurrencesRequest request, StreamObserver<ListOccurrencesResponse> responseObserver) { 86 Object response = responses.poll(); 87 if (response instanceof ListOccurrencesResponse) { 88 requests.add(request); 89 responseObserver.onNext(((ListOccurrencesResponse) response)); 90 responseObserver.onCompleted(); 91 } else if (response instanceof Exception) { 92 responseObserver.onError(((Exception) response)); 93 } else { 94 responseObserver.onError( 95 new IllegalArgumentException( 96 String.format( 97 "Unrecognized response type %s for method ListOccurrences, expected %s or %s", 98 response == null ? "null" : response.getClass().getName(), 99 ListOccurrencesResponse.class.getName(), 100 Exception.class.getName()))); 101 } 102 } 103 104 @Override deleteOccurrence( DeleteOccurrenceRequest request, StreamObserver<Empty> responseObserver)105 public void deleteOccurrence( 106 DeleteOccurrenceRequest request, StreamObserver<Empty> responseObserver) { 107 Object response = responses.poll(); 108 if (response instanceof Empty) { 109 requests.add(request); 110 responseObserver.onNext(((Empty) response)); 111 responseObserver.onCompleted(); 112 } else if (response instanceof Exception) { 113 responseObserver.onError(((Exception) response)); 114 } else { 115 responseObserver.onError( 116 new IllegalArgumentException( 117 String.format( 118 "Unrecognized response type %s for method DeleteOccurrence, expected %s or %s", 119 response == null ? "null" : response.getClass().getName(), 120 Empty.class.getName(), 121 Exception.class.getName()))); 122 } 123 } 124 125 @Override createOccurrence( CreateOccurrenceRequest request, StreamObserver<Occurrence> responseObserver)126 public void createOccurrence( 127 CreateOccurrenceRequest request, StreamObserver<Occurrence> responseObserver) { 128 Object response = responses.poll(); 129 if (response instanceof Occurrence) { 130 requests.add(request); 131 responseObserver.onNext(((Occurrence) response)); 132 responseObserver.onCompleted(); 133 } else if (response instanceof Exception) { 134 responseObserver.onError(((Exception) response)); 135 } else { 136 responseObserver.onError( 137 new IllegalArgumentException( 138 String.format( 139 "Unrecognized response type %s for method CreateOccurrence, expected %s or %s", 140 response == null ? "null" : response.getClass().getName(), 141 Occurrence.class.getName(), 142 Exception.class.getName()))); 143 } 144 } 145 146 @Override batchCreateOccurrences( BatchCreateOccurrencesRequest request, StreamObserver<BatchCreateOccurrencesResponse> responseObserver)147 public void batchCreateOccurrences( 148 BatchCreateOccurrencesRequest request, 149 StreamObserver<BatchCreateOccurrencesResponse> responseObserver) { 150 Object response = responses.poll(); 151 if (response instanceof BatchCreateOccurrencesResponse) { 152 requests.add(request); 153 responseObserver.onNext(((BatchCreateOccurrencesResponse) response)); 154 responseObserver.onCompleted(); 155 } else if (response instanceof Exception) { 156 responseObserver.onError(((Exception) response)); 157 } else { 158 responseObserver.onError( 159 new IllegalArgumentException( 160 String.format( 161 "Unrecognized response type %s for method BatchCreateOccurrences, expected %s or %s", 162 response == null ? "null" : response.getClass().getName(), 163 BatchCreateOccurrencesResponse.class.getName(), 164 Exception.class.getName()))); 165 } 166 } 167 168 @Override updateOccurrence( UpdateOccurrenceRequest request, StreamObserver<Occurrence> responseObserver)169 public void updateOccurrence( 170 UpdateOccurrenceRequest request, StreamObserver<Occurrence> responseObserver) { 171 Object response = responses.poll(); 172 if (response instanceof Occurrence) { 173 requests.add(request); 174 responseObserver.onNext(((Occurrence) response)); 175 responseObserver.onCompleted(); 176 } else if (response instanceof Exception) { 177 responseObserver.onError(((Exception) response)); 178 } else { 179 responseObserver.onError( 180 new IllegalArgumentException( 181 String.format( 182 "Unrecognized response type %s for method UpdateOccurrence, expected %s or %s", 183 response == null ? "null" : response.getClass().getName(), 184 Occurrence.class.getName(), 185 Exception.class.getName()))); 186 } 187 } 188 189 @Override getOccurrenceNote( GetOccurrenceNoteRequest request, StreamObserver<Note> responseObserver)190 public void getOccurrenceNote( 191 GetOccurrenceNoteRequest request, StreamObserver<Note> responseObserver) { 192 Object response = responses.poll(); 193 if (response instanceof Note) { 194 requests.add(request); 195 responseObserver.onNext(((Note) response)); 196 responseObserver.onCompleted(); 197 } else if (response instanceof Exception) { 198 responseObserver.onError(((Exception) response)); 199 } else { 200 responseObserver.onError( 201 new IllegalArgumentException( 202 String.format( 203 "Unrecognized response type %s for method GetOccurrenceNote, expected %s or %s", 204 response == null ? "null" : response.getClass().getName(), 205 Note.class.getName(), 206 Exception.class.getName()))); 207 } 208 } 209 210 @Override getNote(GetNoteRequest request, StreamObserver<Note> responseObserver)211 public void getNote(GetNoteRequest request, StreamObserver<Note> responseObserver) { 212 Object response = responses.poll(); 213 if (response instanceof Note) { 214 requests.add(request); 215 responseObserver.onNext(((Note) response)); 216 responseObserver.onCompleted(); 217 } else if (response instanceof Exception) { 218 responseObserver.onError(((Exception) response)); 219 } else { 220 responseObserver.onError( 221 new IllegalArgumentException( 222 String.format( 223 "Unrecognized response type %s for method GetNote, expected %s or %s", 224 response == null ? "null" : response.getClass().getName(), 225 Note.class.getName(), 226 Exception.class.getName()))); 227 } 228 } 229 230 @Override listNotes( ListNotesRequest request, StreamObserver<ListNotesResponse> responseObserver)231 public void listNotes( 232 ListNotesRequest request, StreamObserver<ListNotesResponse> responseObserver) { 233 Object response = responses.poll(); 234 if (response instanceof ListNotesResponse) { 235 requests.add(request); 236 responseObserver.onNext(((ListNotesResponse) response)); 237 responseObserver.onCompleted(); 238 } else if (response instanceof Exception) { 239 responseObserver.onError(((Exception) response)); 240 } else { 241 responseObserver.onError( 242 new IllegalArgumentException( 243 String.format( 244 "Unrecognized response type %s for method ListNotes, expected %s or %s", 245 response == null ? "null" : response.getClass().getName(), 246 ListNotesResponse.class.getName(), 247 Exception.class.getName()))); 248 } 249 } 250 251 @Override deleteNote(DeleteNoteRequest request, StreamObserver<Empty> responseObserver)252 public void deleteNote(DeleteNoteRequest request, StreamObserver<Empty> responseObserver) { 253 Object response = responses.poll(); 254 if (response instanceof Empty) { 255 requests.add(request); 256 responseObserver.onNext(((Empty) response)); 257 responseObserver.onCompleted(); 258 } else if (response instanceof Exception) { 259 responseObserver.onError(((Exception) response)); 260 } else { 261 responseObserver.onError( 262 new IllegalArgumentException( 263 String.format( 264 "Unrecognized response type %s for method DeleteNote, expected %s or %s", 265 response == null ? "null" : response.getClass().getName(), 266 Empty.class.getName(), 267 Exception.class.getName()))); 268 } 269 } 270 271 @Override createNote(CreateNoteRequest request, StreamObserver<Note> responseObserver)272 public void createNote(CreateNoteRequest request, StreamObserver<Note> responseObserver) { 273 Object response = responses.poll(); 274 if (response instanceof Note) { 275 requests.add(request); 276 responseObserver.onNext(((Note) response)); 277 responseObserver.onCompleted(); 278 } else if (response instanceof Exception) { 279 responseObserver.onError(((Exception) response)); 280 } else { 281 responseObserver.onError( 282 new IllegalArgumentException( 283 String.format( 284 "Unrecognized response type %s for method CreateNote, expected %s or %s", 285 response == null ? "null" : response.getClass().getName(), 286 Note.class.getName(), 287 Exception.class.getName()))); 288 } 289 } 290 291 @Override batchCreateNotes( BatchCreateNotesRequest request, StreamObserver<BatchCreateNotesResponse> responseObserver)292 public void batchCreateNotes( 293 BatchCreateNotesRequest request, StreamObserver<BatchCreateNotesResponse> responseObserver) { 294 Object response = responses.poll(); 295 if (response instanceof BatchCreateNotesResponse) { 296 requests.add(request); 297 responseObserver.onNext(((BatchCreateNotesResponse) response)); 298 responseObserver.onCompleted(); 299 } else if (response instanceof Exception) { 300 responseObserver.onError(((Exception) response)); 301 } else { 302 responseObserver.onError( 303 new IllegalArgumentException( 304 String.format( 305 "Unrecognized response type %s for method BatchCreateNotes, expected %s or %s", 306 response == null ? "null" : response.getClass().getName(), 307 BatchCreateNotesResponse.class.getName(), 308 Exception.class.getName()))); 309 } 310 } 311 312 @Override updateNote(UpdateNoteRequest request, StreamObserver<Note> responseObserver)313 public void updateNote(UpdateNoteRequest request, StreamObserver<Note> responseObserver) { 314 Object response = responses.poll(); 315 if (response instanceof Note) { 316 requests.add(request); 317 responseObserver.onNext(((Note) response)); 318 responseObserver.onCompleted(); 319 } else if (response instanceof Exception) { 320 responseObserver.onError(((Exception) response)); 321 } else { 322 responseObserver.onError( 323 new IllegalArgumentException( 324 String.format( 325 "Unrecognized response type %s for method UpdateNote, expected %s or %s", 326 response == null ? "null" : response.getClass().getName(), 327 Note.class.getName(), 328 Exception.class.getName()))); 329 } 330 } 331 332 @Override listNoteOccurrences( ListNoteOccurrencesRequest request, StreamObserver<ListNoteOccurrencesResponse> responseObserver)333 public void listNoteOccurrences( 334 ListNoteOccurrencesRequest request, 335 StreamObserver<ListNoteOccurrencesResponse> responseObserver) { 336 Object response = responses.poll(); 337 if (response instanceof ListNoteOccurrencesResponse) { 338 requests.add(request); 339 responseObserver.onNext(((ListNoteOccurrencesResponse) response)); 340 responseObserver.onCompleted(); 341 } else if (response instanceof Exception) { 342 responseObserver.onError(((Exception) response)); 343 } else { 344 responseObserver.onError( 345 new IllegalArgumentException( 346 String.format( 347 "Unrecognized response type %s for method ListNoteOccurrences, expected %s or %s", 348 response == null ? "null" : response.getClass().getName(), 349 ListNoteOccurrencesResponse.class.getName(), 350 Exception.class.getName()))); 351 } 352 } 353 } 354