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.v1; 18 19 import com.google.api.core.BetaApi; 20 import com.google.cloud.aiplatform.v1.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 deleteModel(DeleteModelRequest request, StreamObserver<Operation> responseObserver)166 public void deleteModel(DeleteModelRequest request, StreamObserver<Operation> responseObserver) { 167 Object response = responses.poll(); 168 if (response instanceof Operation) { 169 requests.add(request); 170 responseObserver.onNext(((Operation) response)); 171 responseObserver.onCompleted(); 172 } else if (response instanceof Exception) { 173 responseObserver.onError(((Exception) response)); 174 } else { 175 responseObserver.onError( 176 new IllegalArgumentException( 177 String.format( 178 "Unrecognized response type %s for method DeleteModel, expected %s or %s", 179 response == null ? "null" : response.getClass().getName(), 180 Operation.class.getName(), 181 Exception.class.getName()))); 182 } 183 } 184 185 @Override deleteModelVersion( DeleteModelVersionRequest request, StreamObserver<Operation> responseObserver)186 public void deleteModelVersion( 187 DeleteModelVersionRequest 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 DeleteModelVersion, expected %s or %s", 200 response == null ? "null" : response.getClass().getName(), 201 Operation.class.getName(), 202 Exception.class.getName()))); 203 } 204 } 205 206 @Override mergeVersionAliases( MergeVersionAliasesRequest request, StreamObserver<Model> responseObserver)207 public void mergeVersionAliases( 208 MergeVersionAliasesRequest request, StreamObserver<Model> responseObserver) { 209 Object response = responses.poll(); 210 if (response instanceof Model) { 211 requests.add(request); 212 responseObserver.onNext(((Model) 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 MergeVersionAliases, expected %s or %s", 221 response == null ? "null" : response.getClass().getName(), 222 Model.class.getName(), 223 Exception.class.getName()))); 224 } 225 } 226 227 @Override exportModel(ExportModelRequest request, StreamObserver<Operation> responseObserver)228 public void exportModel(ExportModelRequest request, StreamObserver<Operation> responseObserver) { 229 Object response = responses.poll(); 230 if (response instanceof Operation) { 231 requests.add(request); 232 responseObserver.onNext(((Operation) response)); 233 responseObserver.onCompleted(); 234 } else if (response instanceof Exception) { 235 responseObserver.onError(((Exception) response)); 236 } else { 237 responseObserver.onError( 238 new IllegalArgumentException( 239 String.format( 240 "Unrecognized response type %s for method ExportModel, expected %s or %s", 241 response == null ? "null" : response.getClass().getName(), 242 Operation.class.getName(), 243 Exception.class.getName()))); 244 } 245 } 246 247 @Override copyModel(CopyModelRequest request, StreamObserver<Operation> responseObserver)248 public void copyModel(CopyModelRequest request, StreamObserver<Operation> responseObserver) { 249 Object response = responses.poll(); 250 if (response instanceof Operation) { 251 requests.add(request); 252 responseObserver.onNext(((Operation) response)); 253 responseObserver.onCompleted(); 254 } else if (response instanceof Exception) { 255 responseObserver.onError(((Exception) response)); 256 } else { 257 responseObserver.onError( 258 new IllegalArgumentException( 259 String.format( 260 "Unrecognized response type %s for method CopyModel, expected %s or %s", 261 response == null ? "null" : response.getClass().getName(), 262 Operation.class.getName(), 263 Exception.class.getName()))); 264 } 265 } 266 267 @Override importModelEvaluation( ImportModelEvaluationRequest request, StreamObserver<ModelEvaluation> responseObserver)268 public void importModelEvaluation( 269 ImportModelEvaluationRequest request, StreamObserver<ModelEvaluation> responseObserver) { 270 Object response = responses.poll(); 271 if (response instanceof ModelEvaluation) { 272 requests.add(request); 273 responseObserver.onNext(((ModelEvaluation) 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 ImportModelEvaluation, expected %s or %s", 282 response == null ? "null" : response.getClass().getName(), 283 ModelEvaluation.class.getName(), 284 Exception.class.getName()))); 285 } 286 } 287 288 @Override batchImportModelEvaluationSlices( BatchImportModelEvaluationSlicesRequest request, StreamObserver<BatchImportModelEvaluationSlicesResponse> responseObserver)289 public void batchImportModelEvaluationSlices( 290 BatchImportModelEvaluationSlicesRequest request, 291 StreamObserver<BatchImportModelEvaluationSlicesResponse> responseObserver) { 292 Object response = responses.poll(); 293 if (response instanceof BatchImportModelEvaluationSlicesResponse) { 294 requests.add(request); 295 responseObserver.onNext(((BatchImportModelEvaluationSlicesResponse) response)); 296 responseObserver.onCompleted(); 297 } else if (response instanceof Exception) { 298 responseObserver.onError(((Exception) response)); 299 } else { 300 responseObserver.onError( 301 new IllegalArgumentException( 302 String.format( 303 "Unrecognized response type %s for method BatchImportModelEvaluationSlices, expected %s or %s", 304 response == null ? "null" : response.getClass().getName(), 305 BatchImportModelEvaluationSlicesResponse.class.getName(), 306 Exception.class.getName()))); 307 } 308 } 309 310 @Override batchImportEvaluatedAnnotations( BatchImportEvaluatedAnnotationsRequest request, StreamObserver<BatchImportEvaluatedAnnotationsResponse> responseObserver)311 public void batchImportEvaluatedAnnotations( 312 BatchImportEvaluatedAnnotationsRequest request, 313 StreamObserver<BatchImportEvaluatedAnnotationsResponse> responseObserver) { 314 Object response = responses.poll(); 315 if (response instanceof BatchImportEvaluatedAnnotationsResponse) { 316 requests.add(request); 317 responseObserver.onNext(((BatchImportEvaluatedAnnotationsResponse) 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 BatchImportEvaluatedAnnotations, expected %s or %s", 326 response == null ? "null" : response.getClass().getName(), 327 BatchImportEvaluatedAnnotationsResponse.class.getName(), 328 Exception.class.getName()))); 329 } 330 } 331 332 @Override getModelEvaluation( GetModelEvaluationRequest request, StreamObserver<ModelEvaluation> responseObserver)333 public void getModelEvaluation( 334 GetModelEvaluationRequest request, StreamObserver<ModelEvaluation> responseObserver) { 335 Object response = responses.poll(); 336 if (response instanceof ModelEvaluation) { 337 requests.add(request); 338 responseObserver.onNext(((ModelEvaluation) 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 GetModelEvaluation, expected %s or %s", 347 response == null ? "null" : response.getClass().getName(), 348 ModelEvaluation.class.getName(), 349 Exception.class.getName()))); 350 } 351 } 352 353 @Override listModelEvaluations( ListModelEvaluationsRequest request, StreamObserver<ListModelEvaluationsResponse> responseObserver)354 public void listModelEvaluations( 355 ListModelEvaluationsRequest request, 356 StreamObserver<ListModelEvaluationsResponse> responseObserver) { 357 Object response = responses.poll(); 358 if (response instanceof ListModelEvaluationsResponse) { 359 requests.add(request); 360 responseObserver.onNext(((ListModelEvaluationsResponse) response)); 361 responseObserver.onCompleted(); 362 } else if (response instanceof Exception) { 363 responseObserver.onError(((Exception) response)); 364 } else { 365 responseObserver.onError( 366 new IllegalArgumentException( 367 String.format( 368 "Unrecognized response type %s for method ListModelEvaluations, expected %s or %s", 369 response == null ? "null" : response.getClass().getName(), 370 ListModelEvaluationsResponse.class.getName(), 371 Exception.class.getName()))); 372 } 373 } 374 375 @Override getModelEvaluationSlice( GetModelEvaluationSliceRequest request, StreamObserver<ModelEvaluationSlice> responseObserver)376 public void getModelEvaluationSlice( 377 GetModelEvaluationSliceRequest request, 378 StreamObserver<ModelEvaluationSlice> responseObserver) { 379 Object response = responses.poll(); 380 if (response instanceof ModelEvaluationSlice) { 381 requests.add(request); 382 responseObserver.onNext(((ModelEvaluationSlice) response)); 383 responseObserver.onCompleted(); 384 } else if (response instanceof Exception) { 385 responseObserver.onError(((Exception) response)); 386 } else { 387 responseObserver.onError( 388 new IllegalArgumentException( 389 String.format( 390 "Unrecognized response type %s for method GetModelEvaluationSlice, expected %s or %s", 391 response == null ? "null" : response.getClass().getName(), 392 ModelEvaluationSlice.class.getName(), 393 Exception.class.getName()))); 394 } 395 } 396 397 @Override listModelEvaluationSlices( ListModelEvaluationSlicesRequest request, StreamObserver<ListModelEvaluationSlicesResponse> responseObserver)398 public void listModelEvaluationSlices( 399 ListModelEvaluationSlicesRequest request, 400 StreamObserver<ListModelEvaluationSlicesResponse> responseObserver) { 401 Object response = responses.poll(); 402 if (response instanceof ListModelEvaluationSlicesResponse) { 403 requests.add(request); 404 responseObserver.onNext(((ListModelEvaluationSlicesResponse) response)); 405 responseObserver.onCompleted(); 406 } else if (response instanceof Exception) { 407 responseObserver.onError(((Exception) response)); 408 } else { 409 responseObserver.onError( 410 new IllegalArgumentException( 411 String.format( 412 "Unrecognized response type %s for method ListModelEvaluationSlices, expected %s or %s", 413 response == null ? "null" : response.getClass().getName(), 414 ListModelEvaluationSlicesResponse.class.getName(), 415 Exception.class.getName()))); 416 } 417 } 418 } 419