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.tasks.v2beta2; 18 19 import com.google.api.core.BetaApi; 20 import com.google.cloud.tasks.v2beta2.CloudTasksGrpc.CloudTasksImplBase; 21 import com.google.iam.v1.GetIamPolicyRequest; 22 import com.google.iam.v1.Policy; 23 import com.google.iam.v1.SetIamPolicyRequest; 24 import com.google.iam.v1.TestIamPermissionsRequest; 25 import com.google.iam.v1.TestIamPermissionsResponse; 26 import com.google.protobuf.AbstractMessage; 27 import com.google.protobuf.Empty; 28 import io.grpc.stub.StreamObserver; 29 import java.util.ArrayList; 30 import java.util.LinkedList; 31 import java.util.List; 32 import java.util.Queue; 33 import javax.annotation.Generated; 34 35 @BetaApi 36 @Generated("by gapic-generator-java") 37 public class MockCloudTasksImpl extends CloudTasksImplBase { 38 private List<AbstractMessage> requests; 39 private Queue<Object> responses; 40 MockCloudTasksImpl()41 public MockCloudTasksImpl() { 42 requests = new ArrayList<>(); 43 responses = new LinkedList<>(); 44 } 45 getRequests()46 public List<AbstractMessage> getRequests() { 47 return requests; 48 } 49 addResponse(AbstractMessage response)50 public void addResponse(AbstractMessage response) { 51 responses.add(response); 52 } 53 setResponses(List<AbstractMessage> responses)54 public void setResponses(List<AbstractMessage> responses) { 55 this.responses = new LinkedList<Object>(responses); 56 } 57 addException(Exception exception)58 public void addException(Exception exception) { 59 responses.add(exception); 60 } 61 reset()62 public void reset() { 63 requests = new ArrayList<>(); 64 responses = new LinkedList<>(); 65 } 66 67 @Override listQueues( ListQueuesRequest request, StreamObserver<ListQueuesResponse> responseObserver)68 public void listQueues( 69 ListQueuesRequest request, StreamObserver<ListQueuesResponse> responseObserver) { 70 Object response = responses.poll(); 71 if (response instanceof ListQueuesResponse) { 72 requests.add(request); 73 responseObserver.onNext(((ListQueuesResponse) response)); 74 responseObserver.onCompleted(); 75 } else if (response instanceof Exception) { 76 responseObserver.onError(((Exception) response)); 77 } else { 78 responseObserver.onError( 79 new IllegalArgumentException( 80 String.format( 81 "Unrecognized response type %s for method ListQueues, expected %s or %s", 82 response == null ? "null" : response.getClass().getName(), 83 ListQueuesResponse.class.getName(), 84 Exception.class.getName()))); 85 } 86 } 87 88 @Override getQueue( GetQueueRequest request, StreamObserver<com.google.cloud.tasks.v2beta2.Queue> responseObserver)89 public void getQueue( 90 GetQueueRequest request, 91 StreamObserver<com.google.cloud.tasks.v2beta2.Queue> responseObserver) { 92 Object response = responses.poll(); 93 if (response instanceof com.google.cloud.tasks.v2beta2.Queue) { 94 requests.add(request); 95 responseObserver.onNext(((com.google.cloud.tasks.v2beta2.Queue) response)); 96 responseObserver.onCompleted(); 97 } else if (response instanceof Exception) { 98 responseObserver.onError(((Exception) response)); 99 } else { 100 responseObserver.onError( 101 new IllegalArgumentException( 102 String.format( 103 "Unrecognized response type %s for method GetQueue, expected %s or %s", 104 response == null ? "null" : response.getClass().getName(), 105 com.google.cloud.tasks.v2beta2.Queue.class.getName(), 106 Exception.class.getName()))); 107 } 108 } 109 110 @Override createQueue( CreateQueueRequest request, StreamObserver<com.google.cloud.tasks.v2beta2.Queue> responseObserver)111 public void createQueue( 112 CreateQueueRequest request, 113 StreamObserver<com.google.cloud.tasks.v2beta2.Queue> responseObserver) { 114 Object response = responses.poll(); 115 if (response instanceof com.google.cloud.tasks.v2beta2.Queue) { 116 requests.add(request); 117 responseObserver.onNext(((com.google.cloud.tasks.v2beta2.Queue) response)); 118 responseObserver.onCompleted(); 119 } else if (response instanceof Exception) { 120 responseObserver.onError(((Exception) response)); 121 } else { 122 responseObserver.onError( 123 new IllegalArgumentException( 124 String.format( 125 "Unrecognized response type %s for method CreateQueue, expected %s or %s", 126 response == null ? "null" : response.getClass().getName(), 127 com.google.cloud.tasks.v2beta2.Queue.class.getName(), 128 Exception.class.getName()))); 129 } 130 } 131 132 @Override updateQueue( UpdateQueueRequest request, StreamObserver<com.google.cloud.tasks.v2beta2.Queue> responseObserver)133 public void updateQueue( 134 UpdateQueueRequest request, 135 StreamObserver<com.google.cloud.tasks.v2beta2.Queue> responseObserver) { 136 Object response = responses.poll(); 137 if (response instanceof com.google.cloud.tasks.v2beta2.Queue) { 138 requests.add(request); 139 responseObserver.onNext(((com.google.cloud.tasks.v2beta2.Queue) response)); 140 responseObserver.onCompleted(); 141 } else if (response instanceof Exception) { 142 responseObserver.onError(((Exception) response)); 143 } else { 144 responseObserver.onError( 145 new IllegalArgumentException( 146 String.format( 147 "Unrecognized response type %s for method UpdateQueue, expected %s or %s", 148 response == null ? "null" : response.getClass().getName(), 149 com.google.cloud.tasks.v2beta2.Queue.class.getName(), 150 Exception.class.getName()))); 151 } 152 } 153 154 @Override deleteQueue(DeleteQueueRequest request, StreamObserver<Empty> responseObserver)155 public void deleteQueue(DeleteQueueRequest request, StreamObserver<Empty> responseObserver) { 156 Object response = responses.poll(); 157 if (response instanceof Empty) { 158 requests.add(request); 159 responseObserver.onNext(((Empty) response)); 160 responseObserver.onCompleted(); 161 } else if (response instanceof Exception) { 162 responseObserver.onError(((Exception) response)); 163 } else { 164 responseObserver.onError( 165 new IllegalArgumentException( 166 String.format( 167 "Unrecognized response type %s for method DeleteQueue, expected %s or %s", 168 response == null ? "null" : response.getClass().getName(), 169 Empty.class.getName(), 170 Exception.class.getName()))); 171 } 172 } 173 174 @Override purgeQueue( PurgeQueueRequest request, StreamObserver<com.google.cloud.tasks.v2beta2.Queue> responseObserver)175 public void purgeQueue( 176 PurgeQueueRequest request, 177 StreamObserver<com.google.cloud.tasks.v2beta2.Queue> responseObserver) { 178 Object response = responses.poll(); 179 if (response instanceof com.google.cloud.tasks.v2beta2.Queue) { 180 requests.add(request); 181 responseObserver.onNext(((com.google.cloud.tasks.v2beta2.Queue) response)); 182 responseObserver.onCompleted(); 183 } else if (response instanceof Exception) { 184 responseObserver.onError(((Exception) response)); 185 } else { 186 responseObserver.onError( 187 new IllegalArgumentException( 188 String.format( 189 "Unrecognized response type %s for method PurgeQueue, expected %s or %s", 190 response == null ? "null" : response.getClass().getName(), 191 com.google.cloud.tasks.v2beta2.Queue.class.getName(), 192 Exception.class.getName()))); 193 } 194 } 195 196 @Override pauseQueue( PauseQueueRequest request, StreamObserver<com.google.cloud.tasks.v2beta2.Queue> responseObserver)197 public void pauseQueue( 198 PauseQueueRequest request, 199 StreamObserver<com.google.cloud.tasks.v2beta2.Queue> responseObserver) { 200 Object response = responses.poll(); 201 if (response instanceof com.google.cloud.tasks.v2beta2.Queue) { 202 requests.add(request); 203 responseObserver.onNext(((com.google.cloud.tasks.v2beta2.Queue) response)); 204 responseObserver.onCompleted(); 205 } else if (response instanceof Exception) { 206 responseObserver.onError(((Exception) response)); 207 } else { 208 responseObserver.onError( 209 new IllegalArgumentException( 210 String.format( 211 "Unrecognized response type %s for method PauseQueue, expected %s or %s", 212 response == null ? "null" : response.getClass().getName(), 213 com.google.cloud.tasks.v2beta2.Queue.class.getName(), 214 Exception.class.getName()))); 215 } 216 } 217 218 @Override resumeQueue( ResumeQueueRequest request, StreamObserver<com.google.cloud.tasks.v2beta2.Queue> responseObserver)219 public void resumeQueue( 220 ResumeQueueRequest request, 221 StreamObserver<com.google.cloud.tasks.v2beta2.Queue> responseObserver) { 222 Object response = responses.poll(); 223 if (response instanceof com.google.cloud.tasks.v2beta2.Queue) { 224 requests.add(request); 225 responseObserver.onNext(((com.google.cloud.tasks.v2beta2.Queue) response)); 226 responseObserver.onCompleted(); 227 } else if (response instanceof Exception) { 228 responseObserver.onError(((Exception) response)); 229 } else { 230 responseObserver.onError( 231 new IllegalArgumentException( 232 String.format( 233 "Unrecognized response type %s for method ResumeQueue, expected %s or %s", 234 response == null ? "null" : response.getClass().getName(), 235 com.google.cloud.tasks.v2beta2.Queue.class.getName(), 236 Exception.class.getName()))); 237 } 238 } 239 240 @Override getIamPolicy(GetIamPolicyRequest request, StreamObserver<Policy> responseObserver)241 public void getIamPolicy(GetIamPolicyRequest request, StreamObserver<Policy> responseObserver) { 242 Object response = responses.poll(); 243 if (response instanceof Policy) { 244 requests.add(request); 245 responseObserver.onNext(((Policy) response)); 246 responseObserver.onCompleted(); 247 } else if (response instanceof Exception) { 248 responseObserver.onError(((Exception) response)); 249 } else { 250 responseObserver.onError( 251 new IllegalArgumentException( 252 String.format( 253 "Unrecognized response type %s for method GetIamPolicy, expected %s or %s", 254 response == null ? "null" : response.getClass().getName(), 255 Policy.class.getName(), 256 Exception.class.getName()))); 257 } 258 } 259 260 @Override setIamPolicy(SetIamPolicyRequest request, StreamObserver<Policy> responseObserver)261 public void setIamPolicy(SetIamPolicyRequest request, StreamObserver<Policy> responseObserver) { 262 Object response = responses.poll(); 263 if (response instanceof Policy) { 264 requests.add(request); 265 responseObserver.onNext(((Policy) response)); 266 responseObserver.onCompleted(); 267 } else if (response instanceof Exception) { 268 responseObserver.onError(((Exception) response)); 269 } else { 270 responseObserver.onError( 271 new IllegalArgumentException( 272 String.format( 273 "Unrecognized response type %s for method SetIamPolicy, expected %s or %s", 274 response == null ? "null" : response.getClass().getName(), 275 Policy.class.getName(), 276 Exception.class.getName()))); 277 } 278 } 279 280 @Override testIamPermissions( TestIamPermissionsRequest request, StreamObserver<TestIamPermissionsResponse> responseObserver)281 public void testIamPermissions( 282 TestIamPermissionsRequest request, 283 StreamObserver<TestIamPermissionsResponse> responseObserver) { 284 Object response = responses.poll(); 285 if (response instanceof TestIamPermissionsResponse) { 286 requests.add(request); 287 responseObserver.onNext(((TestIamPermissionsResponse) response)); 288 responseObserver.onCompleted(); 289 } else if (response instanceof Exception) { 290 responseObserver.onError(((Exception) response)); 291 } else { 292 responseObserver.onError( 293 new IllegalArgumentException( 294 String.format( 295 "Unrecognized response type %s for method TestIamPermissions, expected %s or %s", 296 response == null ? "null" : response.getClass().getName(), 297 TestIamPermissionsResponse.class.getName(), 298 Exception.class.getName()))); 299 } 300 } 301 302 @Override listTasks( ListTasksRequest request, StreamObserver<ListTasksResponse> responseObserver)303 public void listTasks( 304 ListTasksRequest request, StreamObserver<ListTasksResponse> responseObserver) { 305 Object response = responses.poll(); 306 if (response instanceof ListTasksResponse) { 307 requests.add(request); 308 responseObserver.onNext(((ListTasksResponse) response)); 309 responseObserver.onCompleted(); 310 } else if (response instanceof Exception) { 311 responseObserver.onError(((Exception) response)); 312 } else { 313 responseObserver.onError( 314 new IllegalArgumentException( 315 String.format( 316 "Unrecognized response type %s for method ListTasks, expected %s or %s", 317 response == null ? "null" : response.getClass().getName(), 318 ListTasksResponse.class.getName(), 319 Exception.class.getName()))); 320 } 321 } 322 323 @Override getTask(GetTaskRequest request, StreamObserver<Task> responseObserver)324 public void getTask(GetTaskRequest request, StreamObserver<Task> responseObserver) { 325 Object response = responses.poll(); 326 if (response instanceof Task) { 327 requests.add(request); 328 responseObserver.onNext(((Task) response)); 329 responseObserver.onCompleted(); 330 } else if (response instanceof Exception) { 331 responseObserver.onError(((Exception) response)); 332 } else { 333 responseObserver.onError( 334 new IllegalArgumentException( 335 String.format( 336 "Unrecognized response type %s for method GetTask, expected %s or %s", 337 response == null ? "null" : response.getClass().getName(), 338 Task.class.getName(), 339 Exception.class.getName()))); 340 } 341 } 342 343 @Override createTask(CreateTaskRequest request, StreamObserver<Task> responseObserver)344 public void createTask(CreateTaskRequest request, StreamObserver<Task> responseObserver) { 345 Object response = responses.poll(); 346 if (response instanceof Task) { 347 requests.add(request); 348 responseObserver.onNext(((Task) response)); 349 responseObserver.onCompleted(); 350 } else if (response instanceof Exception) { 351 responseObserver.onError(((Exception) response)); 352 } else { 353 responseObserver.onError( 354 new IllegalArgumentException( 355 String.format( 356 "Unrecognized response type %s for method CreateTask, expected %s or %s", 357 response == null ? "null" : response.getClass().getName(), 358 Task.class.getName(), 359 Exception.class.getName()))); 360 } 361 } 362 363 @Override deleteTask(DeleteTaskRequest request, StreamObserver<Empty> responseObserver)364 public void deleteTask(DeleteTaskRequest request, StreamObserver<Empty> responseObserver) { 365 Object response = responses.poll(); 366 if (response instanceof Empty) { 367 requests.add(request); 368 responseObserver.onNext(((Empty) response)); 369 responseObserver.onCompleted(); 370 } else if (response instanceof Exception) { 371 responseObserver.onError(((Exception) response)); 372 } else { 373 responseObserver.onError( 374 new IllegalArgumentException( 375 String.format( 376 "Unrecognized response type %s for method DeleteTask, expected %s or %s", 377 response == null ? "null" : response.getClass().getName(), 378 Empty.class.getName(), 379 Exception.class.getName()))); 380 } 381 } 382 383 @Override leaseTasks( LeaseTasksRequest request, StreamObserver<LeaseTasksResponse> responseObserver)384 public void leaseTasks( 385 LeaseTasksRequest request, StreamObserver<LeaseTasksResponse> responseObserver) { 386 Object response = responses.poll(); 387 if (response instanceof LeaseTasksResponse) { 388 requests.add(request); 389 responseObserver.onNext(((LeaseTasksResponse) response)); 390 responseObserver.onCompleted(); 391 } else if (response instanceof Exception) { 392 responseObserver.onError(((Exception) response)); 393 } else { 394 responseObserver.onError( 395 new IllegalArgumentException( 396 String.format( 397 "Unrecognized response type %s for method LeaseTasks, expected %s or %s", 398 response == null ? "null" : response.getClass().getName(), 399 LeaseTasksResponse.class.getName(), 400 Exception.class.getName()))); 401 } 402 } 403 404 @Override acknowledgeTask( AcknowledgeTaskRequest request, StreamObserver<Empty> responseObserver)405 public void acknowledgeTask( 406 AcknowledgeTaskRequest request, StreamObserver<Empty> responseObserver) { 407 Object response = responses.poll(); 408 if (response instanceof Empty) { 409 requests.add(request); 410 responseObserver.onNext(((Empty) response)); 411 responseObserver.onCompleted(); 412 } else if (response instanceof Exception) { 413 responseObserver.onError(((Exception) response)); 414 } else { 415 responseObserver.onError( 416 new IllegalArgumentException( 417 String.format( 418 "Unrecognized response type %s for method AcknowledgeTask, expected %s or %s", 419 response == null ? "null" : response.getClass().getName(), 420 Empty.class.getName(), 421 Exception.class.getName()))); 422 } 423 } 424 425 @Override renewLease(RenewLeaseRequest request, StreamObserver<Task> responseObserver)426 public void renewLease(RenewLeaseRequest request, StreamObserver<Task> responseObserver) { 427 Object response = responses.poll(); 428 if (response instanceof Task) { 429 requests.add(request); 430 responseObserver.onNext(((Task) response)); 431 responseObserver.onCompleted(); 432 } else if (response instanceof Exception) { 433 responseObserver.onError(((Exception) response)); 434 } else { 435 responseObserver.onError( 436 new IllegalArgumentException( 437 String.format( 438 "Unrecognized response type %s for method RenewLease, expected %s or %s", 439 response == null ? "null" : response.getClass().getName(), 440 Task.class.getName(), 441 Exception.class.getName()))); 442 } 443 } 444 445 @Override cancelLease(CancelLeaseRequest request, StreamObserver<Task> responseObserver)446 public void cancelLease(CancelLeaseRequest request, StreamObserver<Task> responseObserver) { 447 Object response = responses.poll(); 448 if (response instanceof Task) { 449 requests.add(request); 450 responseObserver.onNext(((Task) response)); 451 responseObserver.onCompleted(); 452 } else if (response instanceof Exception) { 453 responseObserver.onError(((Exception) response)); 454 } else { 455 responseObserver.onError( 456 new IllegalArgumentException( 457 String.format( 458 "Unrecognized response type %s for method CancelLease, expected %s or %s", 459 response == null ? "null" : response.getClass().getName(), 460 Task.class.getName(), 461 Exception.class.getName()))); 462 } 463 } 464 465 @Override runTask(RunTaskRequest request, StreamObserver<Task> responseObserver)466 public void runTask(RunTaskRequest request, StreamObserver<Task> responseObserver) { 467 Object response = responses.poll(); 468 if (response instanceof Task) { 469 requests.add(request); 470 responseObserver.onNext(((Task) response)); 471 responseObserver.onCompleted(); 472 } else if (response instanceof Exception) { 473 responseObserver.onError(((Exception) response)); 474 } else { 475 responseObserver.onError( 476 new IllegalArgumentException( 477 String.format( 478 "Unrecognized response type %s for method RunTask, expected %s or %s", 479 response == null ? "null" : response.getClass().getName(), 480 Task.class.getName(), 481 Exception.class.getName()))); 482 } 483 } 484 } 485