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.v2; 18 19 import com.google.api.core.BetaApi; 20 import com.google.cloud.tasks.v2.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.v2.Queue> responseObserver)89 public void getQueue( 90 GetQueueRequest request, StreamObserver<com.google.cloud.tasks.v2.Queue> responseObserver) { 91 Object response = responses.poll(); 92 if (response instanceof com.google.cloud.tasks.v2.Queue) { 93 requests.add(request); 94 responseObserver.onNext(((com.google.cloud.tasks.v2.Queue) response)); 95 responseObserver.onCompleted(); 96 } else if (response instanceof Exception) { 97 responseObserver.onError(((Exception) response)); 98 } else { 99 responseObserver.onError( 100 new IllegalArgumentException( 101 String.format( 102 "Unrecognized response type %s for method GetQueue, expected %s or %s", 103 response == null ? "null" : response.getClass().getName(), 104 com.google.cloud.tasks.v2.Queue.class.getName(), 105 Exception.class.getName()))); 106 } 107 } 108 109 @Override createQueue( CreateQueueRequest request, StreamObserver<com.google.cloud.tasks.v2.Queue> responseObserver)110 public void createQueue( 111 CreateQueueRequest request, 112 StreamObserver<com.google.cloud.tasks.v2.Queue> responseObserver) { 113 Object response = responses.poll(); 114 if (response instanceof com.google.cloud.tasks.v2.Queue) { 115 requests.add(request); 116 responseObserver.onNext(((com.google.cloud.tasks.v2.Queue) response)); 117 responseObserver.onCompleted(); 118 } else if (response instanceof Exception) { 119 responseObserver.onError(((Exception) response)); 120 } else { 121 responseObserver.onError( 122 new IllegalArgumentException( 123 String.format( 124 "Unrecognized response type %s for method CreateQueue, expected %s or %s", 125 response == null ? "null" : response.getClass().getName(), 126 com.google.cloud.tasks.v2.Queue.class.getName(), 127 Exception.class.getName()))); 128 } 129 } 130 131 @Override updateQueue( UpdateQueueRequest request, StreamObserver<com.google.cloud.tasks.v2.Queue> responseObserver)132 public void updateQueue( 133 UpdateQueueRequest request, 134 StreamObserver<com.google.cloud.tasks.v2.Queue> responseObserver) { 135 Object response = responses.poll(); 136 if (response instanceof com.google.cloud.tasks.v2.Queue) { 137 requests.add(request); 138 responseObserver.onNext(((com.google.cloud.tasks.v2.Queue) response)); 139 responseObserver.onCompleted(); 140 } else if (response instanceof Exception) { 141 responseObserver.onError(((Exception) response)); 142 } else { 143 responseObserver.onError( 144 new IllegalArgumentException( 145 String.format( 146 "Unrecognized response type %s for method UpdateQueue, expected %s or %s", 147 response == null ? "null" : response.getClass().getName(), 148 com.google.cloud.tasks.v2.Queue.class.getName(), 149 Exception.class.getName()))); 150 } 151 } 152 153 @Override deleteQueue(DeleteQueueRequest request, StreamObserver<Empty> responseObserver)154 public void deleteQueue(DeleteQueueRequest request, StreamObserver<Empty> responseObserver) { 155 Object response = responses.poll(); 156 if (response instanceof Empty) { 157 requests.add(request); 158 responseObserver.onNext(((Empty) response)); 159 responseObserver.onCompleted(); 160 } else if (response instanceof Exception) { 161 responseObserver.onError(((Exception) response)); 162 } else { 163 responseObserver.onError( 164 new IllegalArgumentException( 165 String.format( 166 "Unrecognized response type %s for method DeleteQueue, expected %s or %s", 167 response == null ? "null" : response.getClass().getName(), 168 Empty.class.getName(), 169 Exception.class.getName()))); 170 } 171 } 172 173 @Override purgeQueue( PurgeQueueRequest request, StreamObserver<com.google.cloud.tasks.v2.Queue> responseObserver)174 public void purgeQueue( 175 PurgeQueueRequest request, StreamObserver<com.google.cloud.tasks.v2.Queue> responseObserver) { 176 Object response = responses.poll(); 177 if (response instanceof com.google.cloud.tasks.v2.Queue) { 178 requests.add(request); 179 responseObserver.onNext(((com.google.cloud.tasks.v2.Queue) response)); 180 responseObserver.onCompleted(); 181 } else if (response instanceof Exception) { 182 responseObserver.onError(((Exception) response)); 183 } else { 184 responseObserver.onError( 185 new IllegalArgumentException( 186 String.format( 187 "Unrecognized response type %s for method PurgeQueue, expected %s or %s", 188 response == null ? "null" : response.getClass().getName(), 189 com.google.cloud.tasks.v2.Queue.class.getName(), 190 Exception.class.getName()))); 191 } 192 } 193 194 @Override pauseQueue( PauseQueueRequest request, StreamObserver<com.google.cloud.tasks.v2.Queue> responseObserver)195 public void pauseQueue( 196 PauseQueueRequest request, StreamObserver<com.google.cloud.tasks.v2.Queue> responseObserver) { 197 Object response = responses.poll(); 198 if (response instanceof com.google.cloud.tasks.v2.Queue) { 199 requests.add(request); 200 responseObserver.onNext(((com.google.cloud.tasks.v2.Queue) response)); 201 responseObserver.onCompleted(); 202 } else if (response instanceof Exception) { 203 responseObserver.onError(((Exception) response)); 204 } else { 205 responseObserver.onError( 206 new IllegalArgumentException( 207 String.format( 208 "Unrecognized response type %s for method PauseQueue, expected %s or %s", 209 response == null ? "null" : response.getClass().getName(), 210 com.google.cloud.tasks.v2.Queue.class.getName(), 211 Exception.class.getName()))); 212 } 213 } 214 215 @Override resumeQueue( ResumeQueueRequest request, StreamObserver<com.google.cloud.tasks.v2.Queue> responseObserver)216 public void resumeQueue( 217 ResumeQueueRequest request, 218 StreamObserver<com.google.cloud.tasks.v2.Queue> responseObserver) { 219 Object response = responses.poll(); 220 if (response instanceof com.google.cloud.tasks.v2.Queue) { 221 requests.add(request); 222 responseObserver.onNext(((com.google.cloud.tasks.v2.Queue) response)); 223 responseObserver.onCompleted(); 224 } else if (response instanceof Exception) { 225 responseObserver.onError(((Exception) response)); 226 } else { 227 responseObserver.onError( 228 new IllegalArgumentException( 229 String.format( 230 "Unrecognized response type %s for method ResumeQueue, expected %s or %s", 231 response == null ? "null" : response.getClass().getName(), 232 com.google.cloud.tasks.v2.Queue.class.getName(), 233 Exception.class.getName()))); 234 } 235 } 236 237 @Override getIamPolicy(GetIamPolicyRequest request, StreamObserver<Policy> responseObserver)238 public void getIamPolicy(GetIamPolicyRequest request, StreamObserver<Policy> responseObserver) { 239 Object response = responses.poll(); 240 if (response instanceof Policy) { 241 requests.add(request); 242 responseObserver.onNext(((Policy) response)); 243 responseObserver.onCompleted(); 244 } else if (response instanceof Exception) { 245 responseObserver.onError(((Exception) response)); 246 } else { 247 responseObserver.onError( 248 new IllegalArgumentException( 249 String.format( 250 "Unrecognized response type %s for method GetIamPolicy, expected %s or %s", 251 response == null ? "null" : response.getClass().getName(), 252 Policy.class.getName(), 253 Exception.class.getName()))); 254 } 255 } 256 257 @Override setIamPolicy(SetIamPolicyRequest request, StreamObserver<Policy> responseObserver)258 public void setIamPolicy(SetIamPolicyRequest request, StreamObserver<Policy> responseObserver) { 259 Object response = responses.poll(); 260 if (response instanceof Policy) { 261 requests.add(request); 262 responseObserver.onNext(((Policy) response)); 263 responseObserver.onCompleted(); 264 } else if (response instanceof Exception) { 265 responseObserver.onError(((Exception) response)); 266 } else { 267 responseObserver.onError( 268 new IllegalArgumentException( 269 String.format( 270 "Unrecognized response type %s for method SetIamPolicy, expected %s or %s", 271 response == null ? "null" : response.getClass().getName(), 272 Policy.class.getName(), 273 Exception.class.getName()))); 274 } 275 } 276 277 @Override testIamPermissions( TestIamPermissionsRequest request, StreamObserver<TestIamPermissionsResponse> responseObserver)278 public void testIamPermissions( 279 TestIamPermissionsRequest request, 280 StreamObserver<TestIamPermissionsResponse> responseObserver) { 281 Object response = responses.poll(); 282 if (response instanceof TestIamPermissionsResponse) { 283 requests.add(request); 284 responseObserver.onNext(((TestIamPermissionsResponse) response)); 285 responseObserver.onCompleted(); 286 } else if (response instanceof Exception) { 287 responseObserver.onError(((Exception) response)); 288 } else { 289 responseObserver.onError( 290 new IllegalArgumentException( 291 String.format( 292 "Unrecognized response type %s for method TestIamPermissions, expected %s or %s", 293 response == null ? "null" : response.getClass().getName(), 294 TestIamPermissionsResponse.class.getName(), 295 Exception.class.getName()))); 296 } 297 } 298 299 @Override listTasks( ListTasksRequest request, StreamObserver<ListTasksResponse> responseObserver)300 public void listTasks( 301 ListTasksRequest request, StreamObserver<ListTasksResponse> responseObserver) { 302 Object response = responses.poll(); 303 if (response instanceof ListTasksResponse) { 304 requests.add(request); 305 responseObserver.onNext(((ListTasksResponse) response)); 306 responseObserver.onCompleted(); 307 } else if (response instanceof Exception) { 308 responseObserver.onError(((Exception) response)); 309 } else { 310 responseObserver.onError( 311 new IllegalArgumentException( 312 String.format( 313 "Unrecognized response type %s for method ListTasks, expected %s or %s", 314 response == null ? "null" : response.getClass().getName(), 315 ListTasksResponse.class.getName(), 316 Exception.class.getName()))); 317 } 318 } 319 320 @Override getTask(GetTaskRequest request, StreamObserver<Task> responseObserver)321 public void getTask(GetTaskRequest request, StreamObserver<Task> responseObserver) { 322 Object response = responses.poll(); 323 if (response instanceof Task) { 324 requests.add(request); 325 responseObserver.onNext(((Task) response)); 326 responseObserver.onCompleted(); 327 } else if (response instanceof Exception) { 328 responseObserver.onError(((Exception) response)); 329 } else { 330 responseObserver.onError( 331 new IllegalArgumentException( 332 String.format( 333 "Unrecognized response type %s for method GetTask, expected %s or %s", 334 response == null ? "null" : response.getClass().getName(), 335 Task.class.getName(), 336 Exception.class.getName()))); 337 } 338 } 339 340 @Override createTask(CreateTaskRequest request, StreamObserver<Task> responseObserver)341 public void createTask(CreateTaskRequest request, StreamObserver<Task> responseObserver) { 342 Object response = responses.poll(); 343 if (response instanceof Task) { 344 requests.add(request); 345 responseObserver.onNext(((Task) response)); 346 responseObserver.onCompleted(); 347 } else if (response instanceof Exception) { 348 responseObserver.onError(((Exception) response)); 349 } else { 350 responseObserver.onError( 351 new IllegalArgumentException( 352 String.format( 353 "Unrecognized response type %s for method CreateTask, expected %s or %s", 354 response == null ? "null" : response.getClass().getName(), 355 Task.class.getName(), 356 Exception.class.getName()))); 357 } 358 } 359 360 @Override deleteTask(DeleteTaskRequest request, StreamObserver<Empty> responseObserver)361 public void deleteTask(DeleteTaskRequest request, StreamObserver<Empty> responseObserver) { 362 Object response = responses.poll(); 363 if (response instanceof Empty) { 364 requests.add(request); 365 responseObserver.onNext(((Empty) 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 DeleteTask, expected %s or %s", 374 response == null ? "null" : response.getClass().getName(), 375 Empty.class.getName(), 376 Exception.class.getName()))); 377 } 378 } 379 380 @Override runTask(RunTaskRequest request, StreamObserver<Task> responseObserver)381 public void runTask(RunTaskRequest request, StreamObserver<Task> responseObserver) { 382 Object response = responses.poll(); 383 if (response instanceof Task) { 384 requests.add(request); 385 responseObserver.onNext(((Task) response)); 386 responseObserver.onCompleted(); 387 } else if (response instanceof Exception) { 388 responseObserver.onError(((Exception) response)); 389 } else { 390 responseObserver.onError( 391 new IllegalArgumentException( 392 String.format( 393 "Unrecognized response type %s for method RunTask, expected %s or %s", 394 response == null ? "null" : response.getClass().getName(), 395 Task.class.getName(), 396 Exception.class.getName()))); 397 } 398 } 399 } 400