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.devtools.cloudbuild.v1; 18 19 import com.google.api.core.BetaApi; 20 import com.google.cloudbuild.v1.ApproveBuildRequest; 21 import com.google.cloudbuild.v1.Build; 22 import com.google.cloudbuild.v1.BuildTrigger; 23 import com.google.cloudbuild.v1.CancelBuildRequest; 24 import com.google.cloudbuild.v1.CloudBuildGrpc.CloudBuildImplBase; 25 import com.google.cloudbuild.v1.CreateBuildRequest; 26 import com.google.cloudbuild.v1.CreateBuildTriggerRequest; 27 import com.google.cloudbuild.v1.CreateWorkerPoolRequest; 28 import com.google.cloudbuild.v1.DeleteBuildTriggerRequest; 29 import com.google.cloudbuild.v1.DeleteWorkerPoolRequest; 30 import com.google.cloudbuild.v1.GetBuildRequest; 31 import com.google.cloudbuild.v1.GetBuildTriggerRequest; 32 import com.google.cloudbuild.v1.GetWorkerPoolRequest; 33 import com.google.cloudbuild.v1.ListBuildTriggersRequest; 34 import com.google.cloudbuild.v1.ListBuildTriggersResponse; 35 import com.google.cloudbuild.v1.ListBuildsRequest; 36 import com.google.cloudbuild.v1.ListBuildsResponse; 37 import com.google.cloudbuild.v1.ListWorkerPoolsRequest; 38 import com.google.cloudbuild.v1.ListWorkerPoolsResponse; 39 import com.google.cloudbuild.v1.ReceiveTriggerWebhookRequest; 40 import com.google.cloudbuild.v1.ReceiveTriggerWebhookResponse; 41 import com.google.cloudbuild.v1.RetryBuildRequest; 42 import com.google.cloudbuild.v1.RunBuildTriggerRequest; 43 import com.google.cloudbuild.v1.UpdateBuildTriggerRequest; 44 import com.google.cloudbuild.v1.UpdateWorkerPoolRequest; 45 import com.google.cloudbuild.v1.WorkerPool; 46 import com.google.longrunning.Operation; 47 import com.google.protobuf.AbstractMessage; 48 import com.google.protobuf.Empty; 49 import io.grpc.stub.StreamObserver; 50 import java.util.ArrayList; 51 import java.util.LinkedList; 52 import java.util.List; 53 import java.util.Queue; 54 import javax.annotation.Generated; 55 56 @BetaApi 57 @Generated("by gapic-generator-java") 58 public class MockCloudBuildImpl extends CloudBuildImplBase { 59 private List<AbstractMessage> requests; 60 private Queue<Object> responses; 61 MockCloudBuildImpl()62 public MockCloudBuildImpl() { 63 requests = new ArrayList<>(); 64 responses = new LinkedList<>(); 65 } 66 getRequests()67 public List<AbstractMessage> getRequests() { 68 return requests; 69 } 70 addResponse(AbstractMessage response)71 public void addResponse(AbstractMessage response) { 72 responses.add(response); 73 } 74 setResponses(List<AbstractMessage> responses)75 public void setResponses(List<AbstractMessage> responses) { 76 this.responses = new LinkedList<Object>(responses); 77 } 78 addException(Exception exception)79 public void addException(Exception exception) { 80 responses.add(exception); 81 } 82 reset()83 public void reset() { 84 requests = new ArrayList<>(); 85 responses = new LinkedList<>(); 86 } 87 88 @Override createBuild(CreateBuildRequest request, StreamObserver<Operation> responseObserver)89 public void createBuild(CreateBuildRequest request, StreamObserver<Operation> responseObserver) { 90 Object response = responses.poll(); 91 if (response instanceof Operation) { 92 requests.add(request); 93 responseObserver.onNext(((Operation) response)); 94 responseObserver.onCompleted(); 95 } else if (response instanceof Exception) { 96 responseObserver.onError(((Exception) response)); 97 } else { 98 responseObserver.onError( 99 new IllegalArgumentException( 100 String.format( 101 "Unrecognized response type %s for method CreateBuild, expected %s or %s", 102 response == null ? "null" : response.getClass().getName(), 103 Operation.class.getName(), 104 Exception.class.getName()))); 105 } 106 } 107 108 @Override getBuild(GetBuildRequest request, StreamObserver<Build> responseObserver)109 public void getBuild(GetBuildRequest request, StreamObserver<Build> responseObserver) { 110 Object response = responses.poll(); 111 if (response instanceof Build) { 112 requests.add(request); 113 responseObserver.onNext(((Build) response)); 114 responseObserver.onCompleted(); 115 } else if (response instanceof Exception) { 116 responseObserver.onError(((Exception) response)); 117 } else { 118 responseObserver.onError( 119 new IllegalArgumentException( 120 String.format( 121 "Unrecognized response type %s for method GetBuild, expected %s or %s", 122 response == null ? "null" : response.getClass().getName(), 123 Build.class.getName(), 124 Exception.class.getName()))); 125 } 126 } 127 128 @Override listBuilds( ListBuildsRequest request, StreamObserver<ListBuildsResponse> responseObserver)129 public void listBuilds( 130 ListBuildsRequest request, StreamObserver<ListBuildsResponse> responseObserver) { 131 Object response = responses.poll(); 132 if (response instanceof ListBuildsResponse) { 133 requests.add(request); 134 responseObserver.onNext(((ListBuildsResponse) response)); 135 responseObserver.onCompleted(); 136 } else if (response instanceof Exception) { 137 responseObserver.onError(((Exception) response)); 138 } else { 139 responseObserver.onError( 140 new IllegalArgumentException( 141 String.format( 142 "Unrecognized response type %s for method ListBuilds, expected %s or %s", 143 response == null ? "null" : response.getClass().getName(), 144 ListBuildsResponse.class.getName(), 145 Exception.class.getName()))); 146 } 147 } 148 149 @Override cancelBuild(CancelBuildRequest request, StreamObserver<Build> responseObserver)150 public void cancelBuild(CancelBuildRequest request, StreamObserver<Build> responseObserver) { 151 Object response = responses.poll(); 152 if (response instanceof Build) { 153 requests.add(request); 154 responseObserver.onNext(((Build) response)); 155 responseObserver.onCompleted(); 156 } else if (response instanceof Exception) { 157 responseObserver.onError(((Exception) response)); 158 } else { 159 responseObserver.onError( 160 new IllegalArgumentException( 161 String.format( 162 "Unrecognized response type %s for method CancelBuild, expected %s or %s", 163 response == null ? "null" : response.getClass().getName(), 164 Build.class.getName(), 165 Exception.class.getName()))); 166 } 167 } 168 169 @Override retryBuild(RetryBuildRequest request, StreamObserver<Operation> responseObserver)170 public void retryBuild(RetryBuildRequest request, StreamObserver<Operation> responseObserver) { 171 Object response = responses.poll(); 172 if (response instanceof Operation) { 173 requests.add(request); 174 responseObserver.onNext(((Operation) 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 RetryBuild, expected %s or %s", 183 response == null ? "null" : response.getClass().getName(), 184 Operation.class.getName(), 185 Exception.class.getName()))); 186 } 187 } 188 189 @Override approveBuild( ApproveBuildRequest request, StreamObserver<Operation> responseObserver)190 public void approveBuild( 191 ApproveBuildRequest request, StreamObserver<Operation> responseObserver) { 192 Object response = responses.poll(); 193 if (response instanceof Operation) { 194 requests.add(request); 195 responseObserver.onNext(((Operation) 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 ApproveBuild, expected %s or %s", 204 response == null ? "null" : response.getClass().getName(), 205 Operation.class.getName(), 206 Exception.class.getName()))); 207 } 208 } 209 210 @Override createBuildTrigger( CreateBuildTriggerRequest request, StreamObserver<BuildTrigger> responseObserver)211 public void createBuildTrigger( 212 CreateBuildTriggerRequest request, StreamObserver<BuildTrigger> responseObserver) { 213 Object response = responses.poll(); 214 if (response instanceof BuildTrigger) { 215 requests.add(request); 216 responseObserver.onNext(((BuildTrigger) response)); 217 responseObserver.onCompleted(); 218 } else if (response instanceof Exception) { 219 responseObserver.onError(((Exception) response)); 220 } else { 221 responseObserver.onError( 222 new IllegalArgumentException( 223 String.format( 224 "Unrecognized response type %s for method CreateBuildTrigger, expected %s or %s", 225 response == null ? "null" : response.getClass().getName(), 226 BuildTrigger.class.getName(), 227 Exception.class.getName()))); 228 } 229 } 230 231 @Override getBuildTrigger( GetBuildTriggerRequest request, StreamObserver<BuildTrigger> responseObserver)232 public void getBuildTrigger( 233 GetBuildTriggerRequest request, StreamObserver<BuildTrigger> responseObserver) { 234 Object response = responses.poll(); 235 if (response instanceof BuildTrigger) { 236 requests.add(request); 237 responseObserver.onNext(((BuildTrigger) response)); 238 responseObserver.onCompleted(); 239 } else if (response instanceof Exception) { 240 responseObserver.onError(((Exception) response)); 241 } else { 242 responseObserver.onError( 243 new IllegalArgumentException( 244 String.format( 245 "Unrecognized response type %s for method GetBuildTrigger, expected %s or %s", 246 response == null ? "null" : response.getClass().getName(), 247 BuildTrigger.class.getName(), 248 Exception.class.getName()))); 249 } 250 } 251 252 @Override listBuildTriggers( ListBuildTriggersRequest request, StreamObserver<ListBuildTriggersResponse> responseObserver)253 public void listBuildTriggers( 254 ListBuildTriggersRequest request, 255 StreamObserver<ListBuildTriggersResponse> responseObserver) { 256 Object response = responses.poll(); 257 if (response instanceof ListBuildTriggersResponse) { 258 requests.add(request); 259 responseObserver.onNext(((ListBuildTriggersResponse) response)); 260 responseObserver.onCompleted(); 261 } else if (response instanceof Exception) { 262 responseObserver.onError(((Exception) response)); 263 } else { 264 responseObserver.onError( 265 new IllegalArgumentException( 266 String.format( 267 "Unrecognized response type %s for method ListBuildTriggers, expected %s or %s", 268 response == null ? "null" : response.getClass().getName(), 269 ListBuildTriggersResponse.class.getName(), 270 Exception.class.getName()))); 271 } 272 } 273 274 @Override deleteBuildTrigger( DeleteBuildTriggerRequest request, StreamObserver<Empty> responseObserver)275 public void deleteBuildTrigger( 276 DeleteBuildTriggerRequest request, StreamObserver<Empty> responseObserver) { 277 Object response = responses.poll(); 278 if (response instanceof Empty) { 279 requests.add(request); 280 responseObserver.onNext(((Empty) response)); 281 responseObserver.onCompleted(); 282 } else if (response instanceof Exception) { 283 responseObserver.onError(((Exception) response)); 284 } else { 285 responseObserver.onError( 286 new IllegalArgumentException( 287 String.format( 288 "Unrecognized response type %s for method DeleteBuildTrigger, expected %s or %s", 289 response == null ? "null" : response.getClass().getName(), 290 Empty.class.getName(), 291 Exception.class.getName()))); 292 } 293 } 294 295 @Override updateBuildTrigger( UpdateBuildTriggerRequest request, StreamObserver<BuildTrigger> responseObserver)296 public void updateBuildTrigger( 297 UpdateBuildTriggerRequest request, StreamObserver<BuildTrigger> responseObserver) { 298 Object response = responses.poll(); 299 if (response instanceof BuildTrigger) { 300 requests.add(request); 301 responseObserver.onNext(((BuildTrigger) response)); 302 responseObserver.onCompleted(); 303 } else if (response instanceof Exception) { 304 responseObserver.onError(((Exception) response)); 305 } else { 306 responseObserver.onError( 307 new IllegalArgumentException( 308 String.format( 309 "Unrecognized response type %s for method UpdateBuildTrigger, expected %s or %s", 310 response == null ? "null" : response.getClass().getName(), 311 BuildTrigger.class.getName(), 312 Exception.class.getName()))); 313 } 314 } 315 316 @Override runBuildTrigger( RunBuildTriggerRequest request, StreamObserver<Operation> responseObserver)317 public void runBuildTrigger( 318 RunBuildTriggerRequest request, StreamObserver<Operation> responseObserver) { 319 Object response = responses.poll(); 320 if (response instanceof Operation) { 321 requests.add(request); 322 responseObserver.onNext(((Operation) response)); 323 responseObserver.onCompleted(); 324 } else if (response instanceof Exception) { 325 responseObserver.onError(((Exception) response)); 326 } else { 327 responseObserver.onError( 328 new IllegalArgumentException( 329 String.format( 330 "Unrecognized response type %s for method RunBuildTrigger, expected %s or %s", 331 response == null ? "null" : response.getClass().getName(), 332 Operation.class.getName(), 333 Exception.class.getName()))); 334 } 335 } 336 337 @Override receiveTriggerWebhook( ReceiveTriggerWebhookRequest request, StreamObserver<ReceiveTriggerWebhookResponse> responseObserver)338 public void receiveTriggerWebhook( 339 ReceiveTriggerWebhookRequest request, 340 StreamObserver<ReceiveTriggerWebhookResponse> responseObserver) { 341 Object response = responses.poll(); 342 if (response instanceof ReceiveTriggerWebhookResponse) { 343 requests.add(request); 344 responseObserver.onNext(((ReceiveTriggerWebhookResponse) response)); 345 responseObserver.onCompleted(); 346 } else if (response instanceof Exception) { 347 responseObserver.onError(((Exception) response)); 348 } else { 349 responseObserver.onError( 350 new IllegalArgumentException( 351 String.format( 352 "Unrecognized response type %s for method ReceiveTriggerWebhook, expected %s or %s", 353 response == null ? "null" : response.getClass().getName(), 354 ReceiveTriggerWebhookResponse.class.getName(), 355 Exception.class.getName()))); 356 } 357 } 358 359 @Override createWorkerPool( CreateWorkerPoolRequest request, StreamObserver<Operation> responseObserver)360 public void createWorkerPool( 361 CreateWorkerPoolRequest request, StreamObserver<Operation> responseObserver) { 362 Object response = responses.poll(); 363 if (response instanceof Operation) { 364 requests.add(request); 365 responseObserver.onNext(((Operation) response)); 366 responseObserver.onCompleted(); 367 } else if (response instanceof Exception) { 368 responseObserver.onError(((Exception) response)); 369 } else { 370 responseObserver.onError( 371 new IllegalArgumentException( 372 String.format( 373 "Unrecognized response type %s for method CreateWorkerPool, expected %s or %s", 374 response == null ? "null" : response.getClass().getName(), 375 Operation.class.getName(), 376 Exception.class.getName()))); 377 } 378 } 379 380 @Override getWorkerPool( GetWorkerPoolRequest request, StreamObserver<WorkerPool> responseObserver)381 public void getWorkerPool( 382 GetWorkerPoolRequest request, StreamObserver<WorkerPool> responseObserver) { 383 Object response = responses.poll(); 384 if (response instanceof WorkerPool) { 385 requests.add(request); 386 responseObserver.onNext(((WorkerPool) response)); 387 responseObserver.onCompleted(); 388 } else if (response instanceof Exception) { 389 responseObserver.onError(((Exception) response)); 390 } else { 391 responseObserver.onError( 392 new IllegalArgumentException( 393 String.format( 394 "Unrecognized response type %s for method GetWorkerPool, expected %s or %s", 395 response == null ? "null" : response.getClass().getName(), 396 WorkerPool.class.getName(), 397 Exception.class.getName()))); 398 } 399 } 400 401 @Override deleteWorkerPool( DeleteWorkerPoolRequest request, StreamObserver<Operation> responseObserver)402 public void deleteWorkerPool( 403 DeleteWorkerPoolRequest request, StreamObserver<Operation> responseObserver) { 404 Object response = responses.poll(); 405 if (response instanceof Operation) { 406 requests.add(request); 407 responseObserver.onNext(((Operation) response)); 408 responseObserver.onCompleted(); 409 } else if (response instanceof Exception) { 410 responseObserver.onError(((Exception) response)); 411 } else { 412 responseObserver.onError( 413 new IllegalArgumentException( 414 String.format( 415 "Unrecognized response type %s for method DeleteWorkerPool, expected %s or %s", 416 response == null ? "null" : response.getClass().getName(), 417 Operation.class.getName(), 418 Exception.class.getName()))); 419 } 420 } 421 422 @Override updateWorkerPool( UpdateWorkerPoolRequest request, StreamObserver<Operation> responseObserver)423 public void updateWorkerPool( 424 UpdateWorkerPoolRequest request, StreamObserver<Operation> responseObserver) { 425 Object response = responses.poll(); 426 if (response instanceof Operation) { 427 requests.add(request); 428 responseObserver.onNext(((Operation) response)); 429 responseObserver.onCompleted(); 430 } else if (response instanceof Exception) { 431 responseObserver.onError(((Exception) response)); 432 } else { 433 responseObserver.onError( 434 new IllegalArgumentException( 435 String.format( 436 "Unrecognized response type %s for method UpdateWorkerPool, expected %s or %s", 437 response == null ? "null" : response.getClass().getName(), 438 Operation.class.getName(), 439 Exception.class.getName()))); 440 } 441 } 442 443 @Override listWorkerPools( ListWorkerPoolsRequest request, StreamObserver<ListWorkerPoolsResponse> responseObserver)444 public void listWorkerPools( 445 ListWorkerPoolsRequest request, StreamObserver<ListWorkerPoolsResponse> responseObserver) { 446 Object response = responses.poll(); 447 if (response instanceof ListWorkerPoolsResponse) { 448 requests.add(request); 449 responseObserver.onNext(((ListWorkerPoolsResponse) response)); 450 responseObserver.onCompleted(); 451 } else if (response instanceof Exception) { 452 responseObserver.onError(((Exception) response)); 453 } else { 454 responseObserver.onError( 455 new IllegalArgumentException( 456 String.format( 457 "Unrecognized response type %s for method ListWorkerPools, expected %s or %s", 458 response == null ? "null" : response.getClass().getName(), 459 ListWorkerPoolsResponse.class.getName(), 460 Exception.class.getName()))); 461 } 462 } 463 } 464