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.container.v1; 18 19 import com.google.api.core.BetaApi; 20 import com.google.container.v1.CancelOperationRequest; 21 import com.google.container.v1.Cluster; 22 import com.google.container.v1.ClusterManagerGrpc.ClusterManagerImplBase; 23 import com.google.container.v1.CompleteIPRotationRequest; 24 import com.google.container.v1.CompleteNodePoolUpgradeRequest; 25 import com.google.container.v1.CreateClusterRequest; 26 import com.google.container.v1.CreateNodePoolRequest; 27 import com.google.container.v1.DeleteClusterRequest; 28 import com.google.container.v1.DeleteNodePoolRequest; 29 import com.google.container.v1.GetClusterRequest; 30 import com.google.container.v1.GetJSONWebKeysRequest; 31 import com.google.container.v1.GetJSONWebKeysResponse; 32 import com.google.container.v1.GetNodePoolRequest; 33 import com.google.container.v1.GetOperationRequest; 34 import com.google.container.v1.GetServerConfigRequest; 35 import com.google.container.v1.ListClustersRequest; 36 import com.google.container.v1.ListClustersResponse; 37 import com.google.container.v1.ListNodePoolsRequest; 38 import com.google.container.v1.ListNodePoolsResponse; 39 import com.google.container.v1.ListOperationsRequest; 40 import com.google.container.v1.ListOperationsResponse; 41 import com.google.container.v1.ListUsableSubnetworksRequest; 42 import com.google.container.v1.ListUsableSubnetworksResponse; 43 import com.google.container.v1.NodePool; 44 import com.google.container.v1.Operation; 45 import com.google.container.v1.RollbackNodePoolUpgradeRequest; 46 import com.google.container.v1.ServerConfig; 47 import com.google.container.v1.SetAddonsConfigRequest; 48 import com.google.container.v1.SetLabelsRequest; 49 import com.google.container.v1.SetLegacyAbacRequest; 50 import com.google.container.v1.SetLocationsRequest; 51 import com.google.container.v1.SetLoggingServiceRequest; 52 import com.google.container.v1.SetMaintenancePolicyRequest; 53 import com.google.container.v1.SetMasterAuthRequest; 54 import com.google.container.v1.SetMonitoringServiceRequest; 55 import com.google.container.v1.SetNetworkPolicyRequest; 56 import com.google.container.v1.SetNodePoolAutoscalingRequest; 57 import com.google.container.v1.SetNodePoolManagementRequest; 58 import com.google.container.v1.SetNodePoolSizeRequest; 59 import com.google.container.v1.StartIPRotationRequest; 60 import com.google.container.v1.UpdateClusterRequest; 61 import com.google.container.v1.UpdateMasterRequest; 62 import com.google.container.v1.UpdateNodePoolRequest; 63 import com.google.protobuf.AbstractMessage; 64 import com.google.protobuf.Empty; 65 import io.grpc.stub.StreamObserver; 66 import java.util.ArrayList; 67 import java.util.LinkedList; 68 import java.util.List; 69 import java.util.Queue; 70 import javax.annotation.Generated; 71 72 @BetaApi 73 @Generated("by gapic-generator-java") 74 public class MockClusterManagerImpl extends ClusterManagerImplBase { 75 private List<AbstractMessage> requests; 76 private Queue<Object> responses; 77 MockClusterManagerImpl()78 public MockClusterManagerImpl() { 79 requests = new ArrayList<>(); 80 responses = new LinkedList<>(); 81 } 82 getRequests()83 public List<AbstractMessage> getRequests() { 84 return requests; 85 } 86 addResponse(AbstractMessage response)87 public void addResponse(AbstractMessage response) { 88 responses.add(response); 89 } 90 setResponses(List<AbstractMessage> responses)91 public void setResponses(List<AbstractMessage> responses) { 92 this.responses = new LinkedList<Object>(responses); 93 } 94 addException(Exception exception)95 public void addException(Exception exception) { 96 responses.add(exception); 97 } 98 reset()99 public void reset() { 100 requests = new ArrayList<>(); 101 responses = new LinkedList<>(); 102 } 103 104 @Override listClusters( ListClustersRequest request, StreamObserver<ListClustersResponse> responseObserver)105 public void listClusters( 106 ListClustersRequest request, StreamObserver<ListClustersResponse> responseObserver) { 107 Object response = responses.poll(); 108 if (response instanceof ListClustersResponse) { 109 requests.add(request); 110 responseObserver.onNext(((ListClustersResponse) 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 ListClusters, expected %s or %s", 119 response == null ? "null" : response.getClass().getName(), 120 ListClustersResponse.class.getName(), 121 Exception.class.getName()))); 122 } 123 } 124 125 @Override getCluster(GetClusterRequest request, StreamObserver<Cluster> responseObserver)126 public void getCluster(GetClusterRequest request, StreamObserver<Cluster> responseObserver) { 127 Object response = responses.poll(); 128 if (response instanceof Cluster) { 129 requests.add(request); 130 responseObserver.onNext(((Cluster) 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 GetCluster, expected %s or %s", 139 response == null ? "null" : response.getClass().getName(), 140 Cluster.class.getName(), 141 Exception.class.getName()))); 142 } 143 } 144 145 @Override createCluster( CreateClusterRequest request, StreamObserver<Operation> responseObserver)146 public void createCluster( 147 CreateClusterRequest request, StreamObserver<Operation> responseObserver) { 148 Object response = responses.poll(); 149 if (response instanceof Operation) { 150 requests.add(request); 151 responseObserver.onNext(((Operation) response)); 152 responseObserver.onCompleted(); 153 } else if (response instanceof Exception) { 154 responseObserver.onError(((Exception) response)); 155 } else { 156 responseObserver.onError( 157 new IllegalArgumentException( 158 String.format( 159 "Unrecognized response type %s for method CreateCluster, expected %s or %s", 160 response == null ? "null" : response.getClass().getName(), 161 Operation.class.getName(), 162 Exception.class.getName()))); 163 } 164 } 165 166 @Override updateCluster( UpdateClusterRequest request, StreamObserver<Operation> responseObserver)167 public void updateCluster( 168 UpdateClusterRequest request, StreamObserver<Operation> responseObserver) { 169 Object response = responses.poll(); 170 if (response instanceof Operation) { 171 requests.add(request); 172 responseObserver.onNext(((Operation) response)); 173 responseObserver.onCompleted(); 174 } else if (response instanceof Exception) { 175 responseObserver.onError(((Exception) response)); 176 } else { 177 responseObserver.onError( 178 new IllegalArgumentException( 179 String.format( 180 "Unrecognized response type %s for method UpdateCluster, expected %s or %s", 181 response == null ? "null" : response.getClass().getName(), 182 Operation.class.getName(), 183 Exception.class.getName()))); 184 } 185 } 186 187 @Override updateNodePool( UpdateNodePoolRequest request, StreamObserver<Operation> responseObserver)188 public void updateNodePool( 189 UpdateNodePoolRequest request, StreamObserver<Operation> responseObserver) { 190 Object response = responses.poll(); 191 if (response instanceof Operation) { 192 requests.add(request); 193 responseObserver.onNext(((Operation) response)); 194 responseObserver.onCompleted(); 195 } else if (response instanceof Exception) { 196 responseObserver.onError(((Exception) response)); 197 } else { 198 responseObserver.onError( 199 new IllegalArgumentException( 200 String.format( 201 "Unrecognized response type %s for method UpdateNodePool, expected %s or %s", 202 response == null ? "null" : response.getClass().getName(), 203 Operation.class.getName(), 204 Exception.class.getName()))); 205 } 206 } 207 208 @Override setNodePoolAutoscaling( SetNodePoolAutoscalingRequest request, StreamObserver<Operation> responseObserver)209 public void setNodePoolAutoscaling( 210 SetNodePoolAutoscalingRequest request, StreamObserver<Operation> responseObserver) { 211 Object response = responses.poll(); 212 if (response instanceof Operation) { 213 requests.add(request); 214 responseObserver.onNext(((Operation) response)); 215 responseObserver.onCompleted(); 216 } else if (response instanceof Exception) { 217 responseObserver.onError(((Exception) response)); 218 } else { 219 responseObserver.onError( 220 new IllegalArgumentException( 221 String.format( 222 "Unrecognized response type %s for method SetNodePoolAutoscaling, expected %s or %s", 223 response == null ? "null" : response.getClass().getName(), 224 Operation.class.getName(), 225 Exception.class.getName()))); 226 } 227 } 228 229 @Override setLoggingService( SetLoggingServiceRequest request, StreamObserver<Operation> responseObserver)230 public void setLoggingService( 231 SetLoggingServiceRequest request, StreamObserver<Operation> responseObserver) { 232 Object response = responses.poll(); 233 if (response instanceof Operation) { 234 requests.add(request); 235 responseObserver.onNext(((Operation) response)); 236 responseObserver.onCompleted(); 237 } else if (response instanceof Exception) { 238 responseObserver.onError(((Exception) response)); 239 } else { 240 responseObserver.onError( 241 new IllegalArgumentException( 242 String.format( 243 "Unrecognized response type %s for method SetLoggingService, expected %s or %s", 244 response == null ? "null" : response.getClass().getName(), 245 Operation.class.getName(), 246 Exception.class.getName()))); 247 } 248 } 249 250 @Override setMonitoringService( SetMonitoringServiceRequest request, StreamObserver<Operation> responseObserver)251 public void setMonitoringService( 252 SetMonitoringServiceRequest request, StreamObserver<Operation> responseObserver) { 253 Object response = responses.poll(); 254 if (response instanceof Operation) { 255 requests.add(request); 256 responseObserver.onNext(((Operation) 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 SetMonitoringService, expected %s or %s", 265 response == null ? "null" : response.getClass().getName(), 266 Operation.class.getName(), 267 Exception.class.getName()))); 268 } 269 } 270 271 @Override setAddonsConfig( SetAddonsConfigRequest request, StreamObserver<Operation> responseObserver)272 public void setAddonsConfig( 273 SetAddonsConfigRequest request, StreamObserver<Operation> responseObserver) { 274 Object response = responses.poll(); 275 if (response instanceof Operation) { 276 requests.add(request); 277 responseObserver.onNext(((Operation) response)); 278 responseObserver.onCompleted(); 279 } else if (response instanceof Exception) { 280 responseObserver.onError(((Exception) response)); 281 } else { 282 responseObserver.onError( 283 new IllegalArgumentException( 284 String.format( 285 "Unrecognized response type %s for method SetAddonsConfig, expected %s or %s", 286 response == null ? "null" : response.getClass().getName(), 287 Operation.class.getName(), 288 Exception.class.getName()))); 289 } 290 } 291 292 @Override setLocations( SetLocationsRequest request, StreamObserver<Operation> responseObserver)293 public void setLocations( 294 SetLocationsRequest request, StreamObserver<Operation> responseObserver) { 295 Object response = responses.poll(); 296 if (response instanceof Operation) { 297 requests.add(request); 298 responseObserver.onNext(((Operation) response)); 299 responseObserver.onCompleted(); 300 } else if (response instanceof Exception) { 301 responseObserver.onError(((Exception) response)); 302 } else { 303 responseObserver.onError( 304 new IllegalArgumentException( 305 String.format( 306 "Unrecognized response type %s for method SetLocations, expected %s or %s", 307 response == null ? "null" : response.getClass().getName(), 308 Operation.class.getName(), 309 Exception.class.getName()))); 310 } 311 } 312 313 @Override updateMaster( UpdateMasterRequest request, StreamObserver<Operation> responseObserver)314 public void updateMaster( 315 UpdateMasterRequest request, StreamObserver<Operation> responseObserver) { 316 Object response = responses.poll(); 317 if (response instanceof Operation) { 318 requests.add(request); 319 responseObserver.onNext(((Operation) response)); 320 responseObserver.onCompleted(); 321 } else if (response instanceof Exception) { 322 responseObserver.onError(((Exception) response)); 323 } else { 324 responseObserver.onError( 325 new IllegalArgumentException( 326 String.format( 327 "Unrecognized response type %s for method UpdateMaster, expected %s or %s", 328 response == null ? "null" : response.getClass().getName(), 329 Operation.class.getName(), 330 Exception.class.getName()))); 331 } 332 } 333 334 @Override setMasterAuth( SetMasterAuthRequest request, StreamObserver<Operation> responseObserver)335 public void setMasterAuth( 336 SetMasterAuthRequest request, StreamObserver<Operation> responseObserver) { 337 Object response = responses.poll(); 338 if (response instanceof Operation) { 339 requests.add(request); 340 responseObserver.onNext(((Operation) response)); 341 responseObserver.onCompleted(); 342 } else if (response instanceof Exception) { 343 responseObserver.onError(((Exception) response)); 344 } else { 345 responseObserver.onError( 346 new IllegalArgumentException( 347 String.format( 348 "Unrecognized response type %s for method SetMasterAuth, expected %s or %s", 349 response == null ? "null" : response.getClass().getName(), 350 Operation.class.getName(), 351 Exception.class.getName()))); 352 } 353 } 354 355 @Override deleteCluster( DeleteClusterRequest request, StreamObserver<Operation> responseObserver)356 public void deleteCluster( 357 DeleteClusterRequest request, StreamObserver<Operation> responseObserver) { 358 Object response = responses.poll(); 359 if (response instanceof Operation) { 360 requests.add(request); 361 responseObserver.onNext(((Operation) response)); 362 responseObserver.onCompleted(); 363 } else if (response instanceof Exception) { 364 responseObserver.onError(((Exception) response)); 365 } else { 366 responseObserver.onError( 367 new IllegalArgumentException( 368 String.format( 369 "Unrecognized response type %s for method DeleteCluster, expected %s or %s", 370 response == null ? "null" : response.getClass().getName(), 371 Operation.class.getName(), 372 Exception.class.getName()))); 373 } 374 } 375 376 @Override listOperations( ListOperationsRequest request, StreamObserver<ListOperationsResponse> responseObserver)377 public void listOperations( 378 ListOperationsRequest request, StreamObserver<ListOperationsResponse> responseObserver) { 379 Object response = responses.poll(); 380 if (response instanceof ListOperationsResponse) { 381 requests.add(request); 382 responseObserver.onNext(((ListOperationsResponse) 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 ListOperations, expected %s or %s", 391 response == null ? "null" : response.getClass().getName(), 392 ListOperationsResponse.class.getName(), 393 Exception.class.getName()))); 394 } 395 } 396 397 @Override getOperation( GetOperationRequest request, StreamObserver<Operation> responseObserver)398 public void getOperation( 399 GetOperationRequest request, StreamObserver<Operation> responseObserver) { 400 Object response = responses.poll(); 401 if (response instanceof Operation) { 402 requests.add(request); 403 responseObserver.onNext(((Operation) 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 GetOperation, expected %s or %s", 412 response == null ? "null" : response.getClass().getName(), 413 Operation.class.getName(), 414 Exception.class.getName()))); 415 } 416 } 417 418 @Override cancelOperation( CancelOperationRequest request, StreamObserver<Empty> responseObserver)419 public void cancelOperation( 420 CancelOperationRequest request, StreamObserver<Empty> responseObserver) { 421 Object response = responses.poll(); 422 if (response instanceof Empty) { 423 requests.add(request); 424 responseObserver.onNext(((Empty) response)); 425 responseObserver.onCompleted(); 426 } else if (response instanceof Exception) { 427 responseObserver.onError(((Exception) response)); 428 } else { 429 responseObserver.onError( 430 new IllegalArgumentException( 431 String.format( 432 "Unrecognized response type %s for method CancelOperation, expected %s or %s", 433 response == null ? "null" : response.getClass().getName(), 434 Empty.class.getName(), 435 Exception.class.getName()))); 436 } 437 } 438 439 @Override getServerConfig( GetServerConfigRequest request, StreamObserver<ServerConfig> responseObserver)440 public void getServerConfig( 441 GetServerConfigRequest request, StreamObserver<ServerConfig> responseObserver) { 442 Object response = responses.poll(); 443 if (response instanceof ServerConfig) { 444 requests.add(request); 445 responseObserver.onNext(((ServerConfig) response)); 446 responseObserver.onCompleted(); 447 } else if (response instanceof Exception) { 448 responseObserver.onError(((Exception) response)); 449 } else { 450 responseObserver.onError( 451 new IllegalArgumentException( 452 String.format( 453 "Unrecognized response type %s for method GetServerConfig, expected %s or %s", 454 response == null ? "null" : response.getClass().getName(), 455 ServerConfig.class.getName(), 456 Exception.class.getName()))); 457 } 458 } 459 460 @Override getJSONWebKeys( GetJSONWebKeysRequest request, StreamObserver<GetJSONWebKeysResponse> responseObserver)461 public void getJSONWebKeys( 462 GetJSONWebKeysRequest request, StreamObserver<GetJSONWebKeysResponse> responseObserver) { 463 Object response = responses.poll(); 464 if (response instanceof GetJSONWebKeysResponse) { 465 requests.add(request); 466 responseObserver.onNext(((GetJSONWebKeysResponse) response)); 467 responseObserver.onCompleted(); 468 } else if (response instanceof Exception) { 469 responseObserver.onError(((Exception) response)); 470 } else { 471 responseObserver.onError( 472 new IllegalArgumentException( 473 String.format( 474 "Unrecognized response type %s for method GetJSONWebKeys, expected %s or %s", 475 response == null ? "null" : response.getClass().getName(), 476 GetJSONWebKeysResponse.class.getName(), 477 Exception.class.getName()))); 478 } 479 } 480 481 @Override listNodePools( ListNodePoolsRequest request, StreamObserver<ListNodePoolsResponse> responseObserver)482 public void listNodePools( 483 ListNodePoolsRequest request, StreamObserver<ListNodePoolsResponse> responseObserver) { 484 Object response = responses.poll(); 485 if (response instanceof ListNodePoolsResponse) { 486 requests.add(request); 487 responseObserver.onNext(((ListNodePoolsResponse) response)); 488 responseObserver.onCompleted(); 489 } else if (response instanceof Exception) { 490 responseObserver.onError(((Exception) response)); 491 } else { 492 responseObserver.onError( 493 new IllegalArgumentException( 494 String.format( 495 "Unrecognized response type %s for method ListNodePools, expected %s or %s", 496 response == null ? "null" : response.getClass().getName(), 497 ListNodePoolsResponse.class.getName(), 498 Exception.class.getName()))); 499 } 500 } 501 502 @Override getNodePool(GetNodePoolRequest request, StreamObserver<NodePool> responseObserver)503 public void getNodePool(GetNodePoolRequest request, StreamObserver<NodePool> responseObserver) { 504 Object response = responses.poll(); 505 if (response instanceof NodePool) { 506 requests.add(request); 507 responseObserver.onNext(((NodePool) response)); 508 responseObserver.onCompleted(); 509 } else if (response instanceof Exception) { 510 responseObserver.onError(((Exception) response)); 511 } else { 512 responseObserver.onError( 513 new IllegalArgumentException( 514 String.format( 515 "Unrecognized response type %s for method GetNodePool, expected %s or %s", 516 response == null ? "null" : response.getClass().getName(), 517 NodePool.class.getName(), 518 Exception.class.getName()))); 519 } 520 } 521 522 @Override createNodePool( CreateNodePoolRequest request, StreamObserver<Operation> responseObserver)523 public void createNodePool( 524 CreateNodePoolRequest request, StreamObserver<Operation> responseObserver) { 525 Object response = responses.poll(); 526 if (response instanceof Operation) { 527 requests.add(request); 528 responseObserver.onNext(((Operation) response)); 529 responseObserver.onCompleted(); 530 } else if (response instanceof Exception) { 531 responseObserver.onError(((Exception) response)); 532 } else { 533 responseObserver.onError( 534 new IllegalArgumentException( 535 String.format( 536 "Unrecognized response type %s for method CreateNodePool, expected %s or %s", 537 response == null ? "null" : response.getClass().getName(), 538 Operation.class.getName(), 539 Exception.class.getName()))); 540 } 541 } 542 543 @Override deleteNodePool( DeleteNodePoolRequest request, StreamObserver<Operation> responseObserver)544 public void deleteNodePool( 545 DeleteNodePoolRequest request, StreamObserver<Operation> responseObserver) { 546 Object response = responses.poll(); 547 if (response instanceof Operation) { 548 requests.add(request); 549 responseObserver.onNext(((Operation) response)); 550 responseObserver.onCompleted(); 551 } else if (response instanceof Exception) { 552 responseObserver.onError(((Exception) response)); 553 } else { 554 responseObserver.onError( 555 new IllegalArgumentException( 556 String.format( 557 "Unrecognized response type %s for method DeleteNodePool, expected %s or %s", 558 response == null ? "null" : response.getClass().getName(), 559 Operation.class.getName(), 560 Exception.class.getName()))); 561 } 562 } 563 564 @Override completeNodePoolUpgrade( CompleteNodePoolUpgradeRequest request, StreamObserver<Empty> responseObserver)565 public void completeNodePoolUpgrade( 566 CompleteNodePoolUpgradeRequest request, StreamObserver<Empty> responseObserver) { 567 Object response = responses.poll(); 568 if (response instanceof Empty) { 569 requests.add(request); 570 responseObserver.onNext(((Empty) response)); 571 responseObserver.onCompleted(); 572 } else if (response instanceof Exception) { 573 responseObserver.onError(((Exception) response)); 574 } else { 575 responseObserver.onError( 576 new IllegalArgumentException( 577 String.format( 578 "Unrecognized response type %s for method CompleteNodePoolUpgrade, expected %s or %s", 579 response == null ? "null" : response.getClass().getName(), 580 Empty.class.getName(), 581 Exception.class.getName()))); 582 } 583 } 584 585 @Override rollbackNodePoolUpgrade( RollbackNodePoolUpgradeRequest request, StreamObserver<Operation> responseObserver)586 public void rollbackNodePoolUpgrade( 587 RollbackNodePoolUpgradeRequest request, StreamObserver<Operation> responseObserver) { 588 Object response = responses.poll(); 589 if (response instanceof Operation) { 590 requests.add(request); 591 responseObserver.onNext(((Operation) response)); 592 responseObserver.onCompleted(); 593 } else if (response instanceof Exception) { 594 responseObserver.onError(((Exception) response)); 595 } else { 596 responseObserver.onError( 597 new IllegalArgumentException( 598 String.format( 599 "Unrecognized response type %s for method RollbackNodePoolUpgrade, expected %s or %s", 600 response == null ? "null" : response.getClass().getName(), 601 Operation.class.getName(), 602 Exception.class.getName()))); 603 } 604 } 605 606 @Override setNodePoolManagement( SetNodePoolManagementRequest request, StreamObserver<Operation> responseObserver)607 public void setNodePoolManagement( 608 SetNodePoolManagementRequest request, StreamObserver<Operation> responseObserver) { 609 Object response = responses.poll(); 610 if (response instanceof Operation) { 611 requests.add(request); 612 responseObserver.onNext(((Operation) response)); 613 responseObserver.onCompleted(); 614 } else if (response instanceof Exception) { 615 responseObserver.onError(((Exception) response)); 616 } else { 617 responseObserver.onError( 618 new IllegalArgumentException( 619 String.format( 620 "Unrecognized response type %s for method SetNodePoolManagement, expected %s or %s", 621 response == null ? "null" : response.getClass().getName(), 622 Operation.class.getName(), 623 Exception.class.getName()))); 624 } 625 } 626 627 @Override setLabels(SetLabelsRequest request, StreamObserver<Operation> responseObserver)628 public void setLabels(SetLabelsRequest request, StreamObserver<Operation> responseObserver) { 629 Object response = responses.poll(); 630 if (response instanceof Operation) { 631 requests.add(request); 632 responseObserver.onNext(((Operation) response)); 633 responseObserver.onCompleted(); 634 } else if (response instanceof Exception) { 635 responseObserver.onError(((Exception) response)); 636 } else { 637 responseObserver.onError( 638 new IllegalArgumentException( 639 String.format( 640 "Unrecognized response type %s for method SetLabels, expected %s or %s", 641 response == null ? "null" : response.getClass().getName(), 642 Operation.class.getName(), 643 Exception.class.getName()))); 644 } 645 } 646 647 @Override setLegacyAbac( SetLegacyAbacRequest request, StreamObserver<Operation> responseObserver)648 public void setLegacyAbac( 649 SetLegacyAbacRequest request, StreamObserver<Operation> responseObserver) { 650 Object response = responses.poll(); 651 if (response instanceof Operation) { 652 requests.add(request); 653 responseObserver.onNext(((Operation) response)); 654 responseObserver.onCompleted(); 655 } else if (response instanceof Exception) { 656 responseObserver.onError(((Exception) response)); 657 } else { 658 responseObserver.onError( 659 new IllegalArgumentException( 660 String.format( 661 "Unrecognized response type %s for method SetLegacyAbac, expected %s or %s", 662 response == null ? "null" : response.getClass().getName(), 663 Operation.class.getName(), 664 Exception.class.getName()))); 665 } 666 } 667 668 @Override startIPRotation( StartIPRotationRequest request, StreamObserver<Operation> responseObserver)669 public void startIPRotation( 670 StartIPRotationRequest request, StreamObserver<Operation> responseObserver) { 671 Object response = responses.poll(); 672 if (response instanceof Operation) { 673 requests.add(request); 674 responseObserver.onNext(((Operation) response)); 675 responseObserver.onCompleted(); 676 } else if (response instanceof Exception) { 677 responseObserver.onError(((Exception) response)); 678 } else { 679 responseObserver.onError( 680 new IllegalArgumentException( 681 String.format( 682 "Unrecognized response type %s for method StartIPRotation, expected %s or %s", 683 response == null ? "null" : response.getClass().getName(), 684 Operation.class.getName(), 685 Exception.class.getName()))); 686 } 687 } 688 689 @Override completeIPRotation( CompleteIPRotationRequest request, StreamObserver<Operation> responseObserver)690 public void completeIPRotation( 691 CompleteIPRotationRequest request, StreamObserver<Operation> responseObserver) { 692 Object response = responses.poll(); 693 if (response instanceof Operation) { 694 requests.add(request); 695 responseObserver.onNext(((Operation) response)); 696 responseObserver.onCompleted(); 697 } else if (response instanceof Exception) { 698 responseObserver.onError(((Exception) response)); 699 } else { 700 responseObserver.onError( 701 new IllegalArgumentException( 702 String.format( 703 "Unrecognized response type %s for method CompleteIPRotation, expected %s or %s", 704 response == null ? "null" : response.getClass().getName(), 705 Operation.class.getName(), 706 Exception.class.getName()))); 707 } 708 } 709 710 @Override setNodePoolSize( SetNodePoolSizeRequest request, StreamObserver<Operation> responseObserver)711 public void setNodePoolSize( 712 SetNodePoolSizeRequest request, StreamObserver<Operation> responseObserver) { 713 Object response = responses.poll(); 714 if (response instanceof Operation) { 715 requests.add(request); 716 responseObserver.onNext(((Operation) response)); 717 responseObserver.onCompleted(); 718 } else if (response instanceof Exception) { 719 responseObserver.onError(((Exception) response)); 720 } else { 721 responseObserver.onError( 722 new IllegalArgumentException( 723 String.format( 724 "Unrecognized response type %s for method SetNodePoolSize, expected %s or %s", 725 response == null ? "null" : response.getClass().getName(), 726 Operation.class.getName(), 727 Exception.class.getName()))); 728 } 729 } 730 731 @Override setNetworkPolicy( SetNetworkPolicyRequest request, StreamObserver<Operation> responseObserver)732 public void setNetworkPolicy( 733 SetNetworkPolicyRequest request, StreamObserver<Operation> responseObserver) { 734 Object response = responses.poll(); 735 if (response instanceof Operation) { 736 requests.add(request); 737 responseObserver.onNext(((Operation) response)); 738 responseObserver.onCompleted(); 739 } else if (response instanceof Exception) { 740 responseObserver.onError(((Exception) response)); 741 } else { 742 responseObserver.onError( 743 new IllegalArgumentException( 744 String.format( 745 "Unrecognized response type %s for method SetNetworkPolicy, expected %s or %s", 746 response == null ? "null" : response.getClass().getName(), 747 Operation.class.getName(), 748 Exception.class.getName()))); 749 } 750 } 751 752 @Override setMaintenancePolicy( SetMaintenancePolicyRequest request, StreamObserver<Operation> responseObserver)753 public void setMaintenancePolicy( 754 SetMaintenancePolicyRequest request, StreamObserver<Operation> responseObserver) { 755 Object response = responses.poll(); 756 if (response instanceof Operation) { 757 requests.add(request); 758 responseObserver.onNext(((Operation) response)); 759 responseObserver.onCompleted(); 760 } else if (response instanceof Exception) { 761 responseObserver.onError(((Exception) response)); 762 } else { 763 responseObserver.onError( 764 new IllegalArgumentException( 765 String.format( 766 "Unrecognized response type %s for method SetMaintenancePolicy, expected %s or %s", 767 response == null ? "null" : response.getClass().getName(), 768 Operation.class.getName(), 769 Exception.class.getName()))); 770 } 771 } 772 773 @Override listUsableSubnetworks( ListUsableSubnetworksRequest request, StreamObserver<ListUsableSubnetworksResponse> responseObserver)774 public void listUsableSubnetworks( 775 ListUsableSubnetworksRequest request, 776 StreamObserver<ListUsableSubnetworksResponse> responseObserver) { 777 Object response = responses.poll(); 778 if (response instanceof ListUsableSubnetworksResponse) { 779 requests.add(request); 780 responseObserver.onNext(((ListUsableSubnetworksResponse) response)); 781 responseObserver.onCompleted(); 782 } else if (response instanceof Exception) { 783 responseObserver.onError(((Exception) response)); 784 } else { 785 responseObserver.onError( 786 new IllegalArgumentException( 787 String.format( 788 "Unrecognized response type %s for method ListUsableSubnetworks, expected %s or %s", 789 response == null ? "null" : response.getClass().getName(), 790 ListUsableSubnetworksResponse.class.getName(), 791 Exception.class.getName()))); 792 } 793 } 794 } 795