/* * Copyright 2022 Google LLC * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * https://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.google.appengine.v1; import static com.google.appengine.v1.InstancesClient.ListInstancesPagedResponse; import com.google.api.gax.core.NoCredentialsProvider; import com.google.api.gax.grpc.GaxGrpcProperties; import com.google.api.gax.grpc.testing.LocalChannelProvider; import com.google.api.gax.grpc.testing.MockGrpcService; import com.google.api.gax.grpc.testing.MockServiceHelper; import com.google.api.gax.rpc.ApiClientHeaderProvider; import com.google.api.gax.rpc.InvalidArgumentException; import com.google.api.gax.rpc.StatusCode; import com.google.common.collect.Lists; import com.google.longrunning.Operation; import com.google.protobuf.AbstractMessage; import com.google.protobuf.Any; import com.google.protobuf.Empty; import com.google.protobuf.Timestamp; import io.grpc.StatusRuntimeException; import java.io.IOException; import java.util.Arrays; import java.util.List; import java.util.UUID; import java.util.concurrent.ExecutionException; import javax.annotation.Generated; import org.junit.After; import org.junit.AfterClass; import org.junit.Assert; import org.junit.Before; import org.junit.BeforeClass; import org.junit.Test; @Generated("by gapic-generator-java") public class InstancesClientTest { private static MockInstances mockInstances; private static MockServiceHelper mockServiceHelper; private LocalChannelProvider channelProvider; private InstancesClient client; @BeforeClass public static void startStaticServer() { mockInstances = new MockInstances(); mockServiceHelper = new MockServiceHelper( UUID.randomUUID().toString(), Arrays.asList(mockInstances)); mockServiceHelper.start(); } @AfterClass public static void stopServer() { mockServiceHelper.stop(); } @Before public void setUp() throws IOException { mockServiceHelper.reset(); channelProvider = mockServiceHelper.createChannelProvider(); InstancesSettings settings = InstancesSettings.newBuilder() .setTransportChannelProvider(channelProvider) .setCredentialsProvider(NoCredentialsProvider.create()) .build(); client = InstancesClient.create(settings); } @After public void tearDown() throws Exception { client.close(); } @Test public void listInstancesTest() throws Exception { Instance responsesElement = Instance.newBuilder().build(); ListInstancesResponse expectedResponse = ListInstancesResponse.newBuilder() .setNextPageToken("") .addAllInstances(Arrays.asList(responsesElement)) .build(); mockInstances.addResponse(expectedResponse); ListInstancesRequest request = ListInstancesRequest.newBuilder() .setParent("parent-995424086") .setPageSize(883849137) .setPageToken("pageToken873572522") .build(); ListInstancesPagedResponse pagedListResponse = client.listInstances(request); List resources = Lists.newArrayList(pagedListResponse.iterateAll()); Assert.assertEquals(1, resources.size()); Assert.assertEquals(expectedResponse.getInstancesList().get(0), resources.get(0)); List actualRequests = mockInstances.getRequests(); Assert.assertEquals(1, actualRequests.size()); ListInstancesRequest actualRequest = ((ListInstancesRequest) actualRequests.get(0)); Assert.assertEquals(request.getParent(), actualRequest.getParent()); Assert.assertEquals(request.getPageSize(), actualRequest.getPageSize()); Assert.assertEquals(request.getPageToken(), actualRequest.getPageToken()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); } @Test public void listInstancesExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT); mockInstances.addException(exception); try { ListInstancesRequest request = ListInstancesRequest.newBuilder() .setParent("parent-995424086") .setPageSize(883849137) .setPageToken("pageToken873572522") .build(); client.listInstances(request); Assert.fail("No exception raised"); } catch (InvalidArgumentException e) { // Expected exception. } } @Test public void getInstanceTest() throws Exception { Instance expectedResponse = Instance.newBuilder() .setName("name3373707") .setId("id3355") .setAppEngineRelease("appEngineRelease-1601369404") .setVmName("vmName-813643294") .setVmZoneName("vmZoneName-245094354") .setVmId("vmId3622450") .setStartTime(Timestamp.newBuilder().build()) .setRequests(-393257020) .setErrors(-1294635157) .setQps(112180) .setAverageLatency(1814586588) .setMemoryUsage(1034557667) .setVmStatus("vmStatus-66804343") .setVmDebugEnabled(true) .setVmIp("vmIp3622462") .build(); mockInstances.addResponse(expectedResponse); GetInstanceRequest request = GetInstanceRequest.newBuilder().setName("name3373707").build(); Instance actualResponse = client.getInstance(request); Assert.assertEquals(expectedResponse, actualResponse); List actualRequests = mockInstances.getRequests(); Assert.assertEquals(1, actualRequests.size()); GetInstanceRequest actualRequest = ((GetInstanceRequest) actualRequests.get(0)); Assert.assertEquals(request.getName(), actualRequest.getName()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); } @Test public void getInstanceExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT); mockInstances.addException(exception); try { GetInstanceRequest request = GetInstanceRequest.newBuilder().setName("name3373707").build(); client.getInstance(request); Assert.fail("No exception raised"); } catch (InvalidArgumentException e) { // Expected exception. } } @Test public void deleteInstanceTest() throws Exception { Empty expectedResponse = Empty.newBuilder().build(); Operation resultOperation = Operation.newBuilder() .setName("deleteInstanceTest") .setDone(true) .setResponse(Any.pack(expectedResponse)) .build(); mockInstances.addResponse(resultOperation); DeleteInstanceRequest request = DeleteInstanceRequest.newBuilder().setName("name3373707").build(); client.deleteInstanceAsync(request).get(); List actualRequests = mockInstances.getRequests(); Assert.assertEquals(1, actualRequests.size()); DeleteInstanceRequest actualRequest = ((DeleteInstanceRequest) actualRequests.get(0)); Assert.assertEquals(request.getName(), actualRequest.getName()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); } @Test public void deleteInstanceExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT); mockInstances.addException(exception); try { DeleteInstanceRequest request = DeleteInstanceRequest.newBuilder().setName("name3373707").build(); client.deleteInstanceAsync(request).get(); Assert.fail("No exception raised"); } catch (ExecutionException e) { Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass()); InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause()); Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode()); } } @Test public void debugInstanceTest() throws Exception { Instance expectedResponse = Instance.newBuilder() .setName("name3373707") .setId("id3355") .setAppEngineRelease("appEngineRelease-1601369404") .setVmName("vmName-813643294") .setVmZoneName("vmZoneName-245094354") .setVmId("vmId3622450") .setStartTime(Timestamp.newBuilder().build()) .setRequests(-393257020) .setErrors(-1294635157) .setQps(112180) .setAverageLatency(1814586588) .setMemoryUsage(1034557667) .setVmStatus("vmStatus-66804343") .setVmDebugEnabled(true) .setVmIp("vmIp3622462") .build(); Operation resultOperation = Operation.newBuilder() .setName("debugInstanceTest") .setDone(true) .setResponse(Any.pack(expectedResponse)) .build(); mockInstances.addResponse(resultOperation); DebugInstanceRequest request = DebugInstanceRequest.newBuilder() .setName("name3373707") .setSshKey("sshKey-893236425") .build(); Instance actualResponse = client.debugInstanceAsync(request).get(); Assert.assertEquals(expectedResponse, actualResponse); List actualRequests = mockInstances.getRequests(); Assert.assertEquals(1, actualRequests.size()); DebugInstanceRequest actualRequest = ((DebugInstanceRequest) actualRequests.get(0)); Assert.assertEquals(request.getName(), actualRequest.getName()); Assert.assertEquals(request.getSshKey(), actualRequest.getSshKey()); Assert.assertTrue( channelProvider.isHeaderSent( ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern())); } @Test public void debugInstanceExceptionTest() throws Exception { StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT); mockInstances.addException(exception); try { DebugInstanceRequest request = DebugInstanceRequest.newBuilder() .setName("name3373707") .setSshKey("sshKey-893236425") .build(); client.debugInstanceAsync(request).get(); Assert.fail("No exception raised"); } catch (ExecutionException e) { Assert.assertEquals(InvalidArgumentException.class, e.getCause().getClass()); InvalidArgumentException apiException = ((InvalidArgumentException) e.getCause()); Assert.assertEquals(StatusCode.Code.INVALID_ARGUMENT, apiException.getStatusCode().getCode()); } } }