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 com.google.cloud.aiplatform.v1beta1; 18 19 import com.google.api.core.BetaApi; 20 import com.google.cloud.aiplatform.v1beta1.ModelServiceGrpc.ModelServiceImplBase; 21 import com.google.longrunning.Operation; 22 import com.google.protobuf.AbstractMessage; 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 MockModelServiceImpl extends ModelServiceImplBase { 33 private List<AbstractMessage> requests; 34 private Queue<Object> responses; 35 MockModelServiceImpl()36 public MockModelServiceImpl() { 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 uploadModel(UploadModelRequest request, StreamObserver<Operation> responseObserver)63 public void uploadModel(UploadModelRequest request, StreamObserver<Operation> responseObserver) { 64 Object response = responses.poll(); 65 if (response instanceof Operation) { 66 requests.add(request); 67 responseObserver.onNext(((Operation) response)); 68 responseObserver.onCompleted(); 69 } else if (response instanceof Exception) { 70 responseObserver.onError(((Exception) response)); 71 } else { 72 responseObserver.onError( 73 new IllegalArgumentException( 74 String.format( 75 "Unrecognized response type %s for method UploadModel, expected %s or %s", 76 response == null ? "null" : response.getClass().getName(), 77 Operation.class.getName(), 78 Exception.class.getName()))); 79 } 80 } 81 82 @Override getModel(GetModelRequest request, StreamObserver<Model> responseObserver)83 public void getModel(GetModelRequest request, StreamObserver<Model> responseObserver) { 84 Object response = responses.poll(); 85 if (response instanceof Model) { 86 requests.add(request); 87 responseObserver.onNext(((Model) response)); 88 responseObserver.onCompleted(); 89 } else if (response instanceof Exception) { 90 responseObserver.onError(((Exception) response)); 91 } else { 92 responseObserver.onError( 93 new IllegalArgumentException( 94 String.format( 95 "Unrecognized response type %s for method GetModel, expected %s or %s", 96 response == null ? "null" : response.getClass().getName(), 97 Model.class.getName(), 98 Exception.class.getName()))); 99 } 100 } 101 102 @Override listModels( ListModelsRequest request, StreamObserver<ListModelsResponse> responseObserver)103 public void listModels( 104 ListModelsRequest request, StreamObserver<ListModelsResponse> responseObserver) { 105 Object response = responses.poll(); 106 if (response instanceof ListModelsResponse) { 107 requests.add(request); 108 responseObserver.onNext(((ListModelsResponse) response)); 109 responseObserver.onCompleted(); 110 } else if (response instanceof Exception) { 111 responseObserver.onError(((Exception) response)); 112 } else { 113 responseObserver.onError( 114 new IllegalArgumentException( 115 String.format( 116 "Unrecognized response type %s for method ListModels, expected %s or %s", 117 response == null ? "null" : response.getClass().getName(), 118 ListModelsResponse.class.getName(), 119 Exception.class.getName()))); 120 } 121 } 122 123 @Override listModelVersions( ListModelVersionsRequest request, StreamObserver<ListModelVersionsResponse> responseObserver)124 public void listModelVersions( 125 ListModelVersionsRequest request, 126 StreamObserver<ListModelVersionsResponse> responseObserver) { 127 Object response = responses.poll(); 128 if (response instanceof ListModelVersionsResponse) { 129 requests.add(request); 130 responseObserver.onNext(((ListModelVersionsResponse) response)); 131 responseObserver.onCompleted(); 132 } else if (response instanceof Exception) { 133 responseObserver.onError(((Exception) response)); 134 } else { 135 responseObserver.onError( 136 new IllegalArgumentException( 137 String.format( 138 "Unrecognized response type %s for method ListModelVersions, expected %s or %s", 139 response == null ? "null" : response.getClass().getName(), 140 ListModelVersionsResponse.class.getName(), 141 Exception.class.getName()))); 142 } 143 } 144 145 @Override updateModel(UpdateModelRequest request, StreamObserver<Model> responseObserver)146 public void updateModel(UpdateModelRequest request, StreamObserver<Model> responseObserver) { 147 Object response = responses.poll(); 148 if (response instanceof Model) { 149 requests.add(request); 150 responseObserver.onNext(((Model) response)); 151 responseObserver.onCompleted(); 152 } else if (response instanceof Exception) { 153 responseObserver.onError(((Exception) response)); 154 } else { 155 responseObserver.onError( 156 new IllegalArgumentException( 157 String.format( 158 "Unrecognized response type %s for method UpdateModel, expected %s or %s", 159 response == null ? "null" : response.getClass().getName(), 160 Model.class.getName(), 161 Exception.class.getName()))); 162 } 163 } 164 165 @Override updateExplanationDataset( UpdateExplanationDatasetRequest request, StreamObserver<Operation> responseObserver)166 public void updateExplanationDataset( 167 UpdateExplanationDatasetRequest request, StreamObserver<Operation> responseObserver) { 168 Object response = responses.poll(); 169 if (response instanceof Operation) { 170 requests.add(request); 171 responseObserver.onNext(((Operation) response)); 172 responseObserver.onCompleted(); 173 } else if (response instanceof Exception) { 174 responseObserver.onError(((Exception) response)); 175 } else { 176 responseObserver.onError( 177 new IllegalArgumentException( 178 String.format( 179 "Unrecognized response type %s for method UpdateExplanationDataset, expected %s or %s", 180 response == null ? "null" : response.getClass().getName(), 181 Operation.class.getName(), 182 Exception.class.getName()))); 183 } 184 } 185 186 @Override deleteModel(DeleteModelRequest request, StreamObserver<Operation> responseObserver)187 public void deleteModel(DeleteModelRequest request, StreamObserver<Operation> responseObserver) { 188 Object response = responses.poll(); 189 if (response instanceof Operation) { 190 requests.add(request); 191 responseObserver.onNext(((Operation) response)); 192 responseObserver.onCompleted(); 193 } else if (response instanceof Exception) { 194 responseObserver.onError(((Exception) response)); 195 } else { 196 responseObserver.onError( 197 new IllegalArgumentException( 198 String.format( 199 "Unrecognized response type %s for method DeleteModel, expected %s or %s", 200 response == null ? "null" : response.getClass().getName(), 201 Operation.class.getName(), 202 Exception.class.getName()))); 203 } 204 } 205 206 @Override deleteModelVersion( DeleteModelVersionRequest request, StreamObserver<Operation> responseObserver)207 public void deleteModelVersion( 208 DeleteModelVersionRequest request, StreamObserver<Operation> responseObserver) { 209 Object response = responses.poll(); 210 if (response instanceof Operation) { 211 requests.add(request); 212 responseObserver.onNext(((Operation) response)); 213 responseObserver.onCompleted(); 214 } else if (response instanceof Exception) { 215 responseObserver.onError(((Exception) response)); 216 } else { 217 responseObserver.onError( 218 new IllegalArgumentException( 219 String.format( 220 "Unrecognized response type %s for method DeleteModelVersion, expected %s or %s", 221 response == null ? "null" : response.getClass().getName(), 222 Operation.class.getName(), 223 Exception.class.getName()))); 224 } 225 } 226 227 @Override mergeVersionAliases( MergeVersionAliasesRequest request, StreamObserver<Model> responseObserver)228 public void mergeVersionAliases( 229 MergeVersionAliasesRequest request, StreamObserver<Model> responseObserver) { 230 Object response = responses.poll(); 231 if (response instanceof Model) { 232 requests.add(request); 233 responseObserver.onNext(((Model) response)); 234 responseObserver.onCompleted(); 235 } else if (response instanceof Exception) { 236 responseObserver.onError(((Exception) response)); 237 } else { 238 responseObserver.onError( 239 new IllegalArgumentException( 240 String.format( 241 "Unrecognized response type %s for method MergeVersionAliases, expected %s or %s", 242 response == null ? "null" : response.getClass().getName(), 243 Model.class.getName(), 244 Exception.class.getName()))); 245 } 246 } 247 248 @Override exportModel(ExportModelRequest request, StreamObserver<Operation> responseObserver)249 public void exportModel(ExportModelRequest request, StreamObserver<Operation> responseObserver) { 250 Object response = responses.poll(); 251 if (response instanceof Operation) { 252 requests.add(request); 253 responseObserver.onNext(((Operation) response)); 254 responseObserver.onCompleted(); 255 } else if (response instanceof Exception) { 256 responseObserver.onError(((Exception) response)); 257 } else { 258 responseObserver.onError( 259 new IllegalArgumentException( 260 String.format( 261 "Unrecognized response type %s for method ExportModel, expected %s or %s", 262 response == null ? "null" : response.getClass().getName(), 263 Operation.class.getName(), 264 Exception.class.getName()))); 265 } 266 } 267 268 @Override copyModel(CopyModelRequest request, StreamObserver<Operation> responseObserver)269 public void copyModel(CopyModelRequest request, StreamObserver<Operation> responseObserver) { 270 Object response = responses.poll(); 271 if (response instanceof Operation) { 272 requests.add(request); 273 responseObserver.onNext(((Operation) response)); 274 responseObserver.onCompleted(); 275 } else if (response instanceof Exception) { 276 responseObserver.onError(((Exception) response)); 277 } else { 278 responseObserver.onError( 279 new IllegalArgumentException( 280 String.format( 281 "Unrecognized response type %s for method CopyModel, expected %s or %s", 282 response == null ? "null" : response.getClass().getName(), 283 Operation.class.getName(), 284 Exception.class.getName()))); 285 } 286 } 287 288 @Override importModelEvaluation( ImportModelEvaluationRequest request, StreamObserver<ModelEvaluation> responseObserver)289 public void importModelEvaluation( 290 ImportModelEvaluationRequest request, StreamObserver<ModelEvaluation> responseObserver) { 291 Object response = responses.poll(); 292 if (response instanceof ModelEvaluation) { 293 requests.add(request); 294 responseObserver.onNext(((ModelEvaluation) response)); 295 responseObserver.onCompleted(); 296 } else if (response instanceof Exception) { 297 responseObserver.onError(((Exception) response)); 298 } else { 299 responseObserver.onError( 300 new IllegalArgumentException( 301 String.format( 302 "Unrecognized response type %s for method ImportModelEvaluation, expected %s or %s", 303 response == null ? "null" : response.getClass().getName(), 304 ModelEvaluation.class.getName(), 305 Exception.class.getName()))); 306 } 307 } 308 309 @Override batchImportModelEvaluationSlices( BatchImportModelEvaluationSlicesRequest request, StreamObserver<BatchImportModelEvaluationSlicesResponse> responseObserver)310 public void batchImportModelEvaluationSlices( 311 BatchImportModelEvaluationSlicesRequest request, 312 StreamObserver<BatchImportModelEvaluationSlicesResponse> responseObserver) { 313 Object response = responses.poll(); 314 if (response instanceof BatchImportModelEvaluationSlicesResponse) { 315 requests.add(request); 316 responseObserver.onNext(((BatchImportModelEvaluationSlicesResponse) response)); 317 responseObserver.onCompleted(); 318 } else if (response instanceof Exception) { 319 responseObserver.onError(((Exception) response)); 320 } else { 321 responseObserver.onError( 322 new IllegalArgumentException( 323 String.format( 324 "Unrecognized response type %s for method BatchImportModelEvaluationSlices, expected %s or %s", 325 response == null ? "null" : response.getClass().getName(), 326 BatchImportModelEvaluationSlicesResponse.class.getName(), 327 Exception.class.getName()))); 328 } 329 } 330 331 @Override batchImportEvaluatedAnnotations( BatchImportEvaluatedAnnotationsRequest request, StreamObserver<BatchImportEvaluatedAnnotationsResponse> responseObserver)332 public void batchImportEvaluatedAnnotations( 333 BatchImportEvaluatedAnnotationsRequest request, 334 StreamObserver<BatchImportEvaluatedAnnotationsResponse> responseObserver) { 335 Object response = responses.poll(); 336 if (response instanceof BatchImportEvaluatedAnnotationsResponse) { 337 requests.add(request); 338 responseObserver.onNext(((BatchImportEvaluatedAnnotationsResponse) response)); 339 responseObserver.onCompleted(); 340 } else if (response instanceof Exception) { 341 responseObserver.onError(((Exception) response)); 342 } else { 343 responseObserver.onError( 344 new IllegalArgumentException( 345 String.format( 346 "Unrecognized response type %s for method BatchImportEvaluatedAnnotations, expected %s or %s", 347 response == null ? "null" : response.getClass().getName(), 348 BatchImportEvaluatedAnnotationsResponse.class.getName(), 349 Exception.class.getName()))); 350 } 351 } 352 353 @Override getModelEvaluation( GetModelEvaluationRequest request, StreamObserver<ModelEvaluation> responseObserver)354 public void getModelEvaluation( 355 GetModelEvaluationRequest request, StreamObserver<ModelEvaluation> responseObserver) { 356 Object response = responses.poll(); 357 if (response instanceof ModelEvaluation) { 358 requests.add(request); 359 responseObserver.onNext(((ModelEvaluation) response)); 360 responseObserver.onCompleted(); 361 } else if (response instanceof Exception) { 362 responseObserver.onError(((Exception) response)); 363 } else { 364 responseObserver.onError( 365 new IllegalArgumentException( 366 String.format( 367 "Unrecognized response type %s for method GetModelEvaluation, expected %s or %s", 368 response == null ? "null" : response.getClass().getName(), 369 ModelEvaluation.class.getName(), 370 Exception.class.getName()))); 371 } 372 } 373 374 @Override listModelEvaluations( ListModelEvaluationsRequest request, StreamObserver<ListModelEvaluationsResponse> responseObserver)375 public void listModelEvaluations( 376 ListModelEvaluationsRequest request, 377 StreamObserver<ListModelEvaluationsResponse> responseObserver) { 378 Object response = responses.poll(); 379 if (response instanceof ListModelEvaluationsResponse) { 380 requests.add(request); 381 responseObserver.onNext(((ListModelEvaluationsResponse) response)); 382 responseObserver.onCompleted(); 383 } else if (response instanceof Exception) { 384 responseObserver.onError(((Exception) response)); 385 } else { 386 responseObserver.onError( 387 new IllegalArgumentException( 388 String.format( 389 "Unrecognized response type %s for method ListModelEvaluations, expected %s or %s", 390 response == null ? "null" : response.getClass().getName(), 391 ListModelEvaluationsResponse.class.getName(), 392 Exception.class.getName()))); 393 } 394 } 395 396 @Override getModelEvaluationSlice( GetModelEvaluationSliceRequest request, StreamObserver<ModelEvaluationSlice> responseObserver)397 public void getModelEvaluationSlice( 398 GetModelEvaluationSliceRequest request, 399 StreamObserver<ModelEvaluationSlice> responseObserver) { 400 Object response = responses.poll(); 401 if (response instanceof ModelEvaluationSlice) { 402 requests.add(request); 403 responseObserver.onNext(((ModelEvaluationSlice) response)); 404 responseObserver.onCompleted(); 405 } else if (response instanceof Exception) { 406 responseObserver.onError(((Exception) response)); 407 } else { 408 responseObserver.onError( 409 new IllegalArgumentException( 410 String.format( 411 "Unrecognized response type %s for method GetModelEvaluationSlice, expected %s or %s", 412 response == null ? "null" : response.getClass().getName(), 413 ModelEvaluationSlice.class.getName(), 414 Exception.class.getName()))); 415 } 416 } 417 418 @Override listModelEvaluationSlices( ListModelEvaluationSlicesRequest request, StreamObserver<ListModelEvaluationSlicesResponse> responseObserver)419 public void listModelEvaluationSlices( 420 ListModelEvaluationSlicesRequest request, 421 StreamObserver<ListModelEvaluationSlicesResponse> responseObserver) { 422 Object response = responses.poll(); 423 if (response instanceof ListModelEvaluationSlicesResponse) { 424 requests.add(request); 425 responseObserver.onNext(((ListModelEvaluationSlicesResponse) response)); 426 responseObserver.onCompleted(); 427 } else if (response instanceof Exception) { 428 responseObserver.onError(((Exception) response)); 429 } else { 430 responseObserver.onError( 431 new IllegalArgumentException( 432 String.format( 433 "Unrecognized response type %s for method ListModelEvaluationSlices, expected %s or %s", 434 response == null ? "null" : response.getClass().getName(), 435 ListModelEvaluationSlicesResponse.class.getName(), 436 Exception.class.getName()))); 437 } 438 } 439 } 440