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.domains.v1beta1; 18 19 import com.google.api.core.BetaApi; 20 import com.google.cloud.domains.v1beta1.DomainsGrpc.DomainsImplBase; 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 MockDomainsImpl extends DomainsImplBase { 33 private List<AbstractMessage> requests; 34 private Queue<Object> responses; 35 MockDomainsImpl()36 public MockDomainsImpl() { 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 searchDomains( SearchDomainsRequest request, StreamObserver<SearchDomainsResponse> responseObserver)63 public void searchDomains( 64 SearchDomainsRequest request, StreamObserver<SearchDomainsResponse> responseObserver) { 65 Object response = responses.poll(); 66 if (response instanceof SearchDomainsResponse) { 67 requests.add(request); 68 responseObserver.onNext(((SearchDomainsResponse) response)); 69 responseObserver.onCompleted(); 70 } else if (response instanceof Exception) { 71 responseObserver.onError(((Exception) response)); 72 } else { 73 responseObserver.onError( 74 new IllegalArgumentException( 75 String.format( 76 "Unrecognized response type %s for method SearchDomains, expected %s or %s", 77 response == null ? "null" : response.getClass().getName(), 78 SearchDomainsResponse.class.getName(), 79 Exception.class.getName()))); 80 } 81 } 82 83 @Override retrieveRegisterParameters( RetrieveRegisterParametersRequest request, StreamObserver<RetrieveRegisterParametersResponse> responseObserver)84 public void retrieveRegisterParameters( 85 RetrieveRegisterParametersRequest request, 86 StreamObserver<RetrieveRegisterParametersResponse> responseObserver) { 87 Object response = responses.poll(); 88 if (response instanceof RetrieveRegisterParametersResponse) { 89 requests.add(request); 90 responseObserver.onNext(((RetrieveRegisterParametersResponse) response)); 91 responseObserver.onCompleted(); 92 } else if (response instanceof Exception) { 93 responseObserver.onError(((Exception) response)); 94 } else { 95 responseObserver.onError( 96 new IllegalArgumentException( 97 String.format( 98 "Unrecognized response type %s for method RetrieveRegisterParameters, expected %s or %s", 99 response == null ? "null" : response.getClass().getName(), 100 RetrieveRegisterParametersResponse.class.getName(), 101 Exception.class.getName()))); 102 } 103 } 104 105 @Override registerDomain( RegisterDomainRequest request, StreamObserver<Operation> responseObserver)106 public void registerDomain( 107 RegisterDomainRequest request, StreamObserver<Operation> responseObserver) { 108 Object response = responses.poll(); 109 if (response instanceof Operation) { 110 requests.add(request); 111 responseObserver.onNext(((Operation) response)); 112 responseObserver.onCompleted(); 113 } else if (response instanceof Exception) { 114 responseObserver.onError(((Exception) response)); 115 } else { 116 responseObserver.onError( 117 new IllegalArgumentException( 118 String.format( 119 "Unrecognized response type %s for method RegisterDomain, expected %s or %s", 120 response == null ? "null" : response.getClass().getName(), 121 Operation.class.getName(), 122 Exception.class.getName()))); 123 } 124 } 125 126 @Override retrieveTransferParameters( RetrieveTransferParametersRequest request, StreamObserver<RetrieveTransferParametersResponse> responseObserver)127 public void retrieveTransferParameters( 128 RetrieveTransferParametersRequest request, 129 StreamObserver<RetrieveTransferParametersResponse> responseObserver) { 130 Object response = responses.poll(); 131 if (response instanceof RetrieveTransferParametersResponse) { 132 requests.add(request); 133 responseObserver.onNext(((RetrieveTransferParametersResponse) response)); 134 responseObserver.onCompleted(); 135 } else if (response instanceof Exception) { 136 responseObserver.onError(((Exception) response)); 137 } else { 138 responseObserver.onError( 139 new IllegalArgumentException( 140 String.format( 141 "Unrecognized response type %s for method RetrieveTransferParameters, expected %s or %s", 142 response == null ? "null" : response.getClass().getName(), 143 RetrieveTransferParametersResponse.class.getName(), 144 Exception.class.getName()))); 145 } 146 } 147 148 @Override transferDomain( TransferDomainRequest request, StreamObserver<Operation> responseObserver)149 public void transferDomain( 150 TransferDomainRequest request, StreamObserver<Operation> responseObserver) { 151 Object response = responses.poll(); 152 if (response instanceof Operation) { 153 requests.add(request); 154 responseObserver.onNext(((Operation) 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 TransferDomain, expected %s or %s", 163 response == null ? "null" : response.getClass().getName(), 164 Operation.class.getName(), 165 Exception.class.getName()))); 166 } 167 } 168 169 @Override listRegistrations( ListRegistrationsRequest request, StreamObserver<ListRegistrationsResponse> responseObserver)170 public void listRegistrations( 171 ListRegistrationsRequest request, 172 StreamObserver<ListRegistrationsResponse> responseObserver) { 173 Object response = responses.poll(); 174 if (response instanceof ListRegistrationsResponse) { 175 requests.add(request); 176 responseObserver.onNext(((ListRegistrationsResponse) response)); 177 responseObserver.onCompleted(); 178 } else if (response instanceof Exception) { 179 responseObserver.onError(((Exception) response)); 180 } else { 181 responseObserver.onError( 182 new IllegalArgumentException( 183 String.format( 184 "Unrecognized response type %s for method ListRegistrations, expected %s or %s", 185 response == null ? "null" : response.getClass().getName(), 186 ListRegistrationsResponse.class.getName(), 187 Exception.class.getName()))); 188 } 189 } 190 191 @Override getRegistration( GetRegistrationRequest request, StreamObserver<Registration> responseObserver)192 public void getRegistration( 193 GetRegistrationRequest request, StreamObserver<Registration> responseObserver) { 194 Object response = responses.poll(); 195 if (response instanceof Registration) { 196 requests.add(request); 197 responseObserver.onNext(((Registration) response)); 198 responseObserver.onCompleted(); 199 } else if (response instanceof Exception) { 200 responseObserver.onError(((Exception) response)); 201 } else { 202 responseObserver.onError( 203 new IllegalArgumentException( 204 String.format( 205 "Unrecognized response type %s for method GetRegistration, expected %s or %s", 206 response == null ? "null" : response.getClass().getName(), 207 Registration.class.getName(), 208 Exception.class.getName()))); 209 } 210 } 211 212 @Override updateRegistration( UpdateRegistrationRequest request, StreamObserver<Operation> responseObserver)213 public void updateRegistration( 214 UpdateRegistrationRequest request, StreamObserver<Operation> responseObserver) { 215 Object response = responses.poll(); 216 if (response instanceof Operation) { 217 requests.add(request); 218 responseObserver.onNext(((Operation) response)); 219 responseObserver.onCompleted(); 220 } else if (response instanceof Exception) { 221 responseObserver.onError(((Exception) response)); 222 } else { 223 responseObserver.onError( 224 new IllegalArgumentException( 225 String.format( 226 "Unrecognized response type %s for method UpdateRegistration, expected %s or %s", 227 response == null ? "null" : response.getClass().getName(), 228 Operation.class.getName(), 229 Exception.class.getName()))); 230 } 231 } 232 233 @Override configureManagementSettings( ConfigureManagementSettingsRequest request, StreamObserver<Operation> responseObserver)234 public void configureManagementSettings( 235 ConfigureManagementSettingsRequest request, StreamObserver<Operation> responseObserver) { 236 Object response = responses.poll(); 237 if (response instanceof Operation) { 238 requests.add(request); 239 responseObserver.onNext(((Operation) response)); 240 responseObserver.onCompleted(); 241 } else if (response instanceof Exception) { 242 responseObserver.onError(((Exception) response)); 243 } else { 244 responseObserver.onError( 245 new IllegalArgumentException( 246 String.format( 247 "Unrecognized response type %s for method ConfigureManagementSettings, expected %s or %s", 248 response == null ? "null" : response.getClass().getName(), 249 Operation.class.getName(), 250 Exception.class.getName()))); 251 } 252 } 253 254 @Override configureDnsSettings( ConfigureDnsSettingsRequest request, StreamObserver<Operation> responseObserver)255 public void configureDnsSettings( 256 ConfigureDnsSettingsRequest request, StreamObserver<Operation> responseObserver) { 257 Object response = responses.poll(); 258 if (response instanceof Operation) { 259 requests.add(request); 260 responseObserver.onNext(((Operation) response)); 261 responseObserver.onCompleted(); 262 } else if (response instanceof Exception) { 263 responseObserver.onError(((Exception) response)); 264 } else { 265 responseObserver.onError( 266 new IllegalArgumentException( 267 String.format( 268 "Unrecognized response type %s for method ConfigureDnsSettings, expected %s or %s", 269 response == null ? "null" : response.getClass().getName(), 270 Operation.class.getName(), 271 Exception.class.getName()))); 272 } 273 } 274 275 @Override configureContactSettings( ConfigureContactSettingsRequest request, StreamObserver<Operation> responseObserver)276 public void configureContactSettings( 277 ConfigureContactSettingsRequest request, StreamObserver<Operation> responseObserver) { 278 Object response = responses.poll(); 279 if (response instanceof Operation) { 280 requests.add(request); 281 responseObserver.onNext(((Operation) response)); 282 responseObserver.onCompleted(); 283 } else if (response instanceof Exception) { 284 responseObserver.onError(((Exception) response)); 285 } else { 286 responseObserver.onError( 287 new IllegalArgumentException( 288 String.format( 289 "Unrecognized response type %s for method ConfigureContactSettings, expected %s or %s", 290 response == null ? "null" : response.getClass().getName(), 291 Operation.class.getName(), 292 Exception.class.getName()))); 293 } 294 } 295 296 @Override exportRegistration( ExportRegistrationRequest request, StreamObserver<Operation> responseObserver)297 public void exportRegistration( 298 ExportRegistrationRequest request, StreamObserver<Operation> responseObserver) { 299 Object response = responses.poll(); 300 if (response instanceof Operation) { 301 requests.add(request); 302 responseObserver.onNext(((Operation) response)); 303 responseObserver.onCompleted(); 304 } else if (response instanceof Exception) { 305 responseObserver.onError(((Exception) response)); 306 } else { 307 responseObserver.onError( 308 new IllegalArgumentException( 309 String.format( 310 "Unrecognized response type %s for method ExportRegistration, expected %s or %s", 311 response == null ? "null" : response.getClass().getName(), 312 Operation.class.getName(), 313 Exception.class.getName()))); 314 } 315 } 316 317 @Override deleteRegistration( DeleteRegistrationRequest request, StreamObserver<Operation> responseObserver)318 public void deleteRegistration( 319 DeleteRegistrationRequest request, StreamObserver<Operation> responseObserver) { 320 Object response = responses.poll(); 321 if (response instanceof Operation) { 322 requests.add(request); 323 responseObserver.onNext(((Operation) response)); 324 responseObserver.onCompleted(); 325 } else if (response instanceof Exception) { 326 responseObserver.onError(((Exception) response)); 327 } else { 328 responseObserver.onError( 329 new IllegalArgumentException( 330 String.format( 331 "Unrecognized response type %s for method DeleteRegistration, expected %s or %s", 332 response == null ? "null" : response.getClass().getName(), 333 Operation.class.getName(), 334 Exception.class.getName()))); 335 } 336 } 337 338 @Override retrieveAuthorizationCode( RetrieveAuthorizationCodeRequest request, StreamObserver<AuthorizationCode> responseObserver)339 public void retrieveAuthorizationCode( 340 RetrieveAuthorizationCodeRequest request, 341 StreamObserver<AuthorizationCode> responseObserver) { 342 Object response = responses.poll(); 343 if (response instanceof AuthorizationCode) { 344 requests.add(request); 345 responseObserver.onNext(((AuthorizationCode) 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 RetrieveAuthorizationCode, expected %s or %s", 354 response == null ? "null" : response.getClass().getName(), 355 AuthorizationCode.class.getName(), 356 Exception.class.getName()))); 357 } 358 } 359 360 @Override resetAuthorizationCode( ResetAuthorizationCodeRequest request, StreamObserver<AuthorizationCode> responseObserver)361 public void resetAuthorizationCode( 362 ResetAuthorizationCodeRequest request, StreamObserver<AuthorizationCode> responseObserver) { 363 Object response = responses.poll(); 364 if (response instanceof AuthorizationCode) { 365 requests.add(request); 366 responseObserver.onNext(((AuthorizationCode) response)); 367 responseObserver.onCompleted(); 368 } else if (response instanceof Exception) { 369 responseObserver.onError(((Exception) response)); 370 } else { 371 responseObserver.onError( 372 new IllegalArgumentException( 373 String.format( 374 "Unrecognized response type %s for method ResetAuthorizationCode, expected %s or %s", 375 response == null ? "null" : response.getClass().getName(), 376 AuthorizationCode.class.getName(), 377 Exception.class.getName()))); 378 } 379 } 380 } 381