package com.amazonaws.iot.cacertificate; import org.junit.jupiter.api.Assertions; import org.mockito.ArgumentCaptor; import software.amazon.awssdk.services.iot.model.AutoRegistrationStatus; import software.amazon.awssdk.services.iot.model.CACertificateDescription; import software.amazon.awssdk.services.iot.model.DescribeCaCertificateRequest; import software.amazon.awssdk.services.iot.model.DescribeCaCertificateResponse; import software.amazon.awssdk.services.iot.model.InternalFailureException; import software.amazon.awssdk.services.iot.model.InvalidRequestException; import software.amazon.awssdk.services.iot.model.ResourceNotFoundException; import software.amazon.awssdk.services.iot.model.ThrottlingException; import software.amazon.cloudformation.exceptions.CfnInvalidRequestException; import software.amazon.cloudformation.exceptions.CfnNotFoundException; import software.amazon.cloudformation.exceptions.CfnServiceInternalErrorException; import software.amazon.cloudformation.exceptions.CfnThrottlingException; import software.amazon.cloudformation.proxy.AmazonWebServicesClientProxy; import software.amazon.cloudformation.proxy.Logger; import software.amazon.cloudformation.proxy.OperationStatus; import software.amazon.cloudformation.proxy.ProgressEvent; import software.amazon.cloudformation.proxy.ResourceHandlerRequest; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; import org.junit.jupiter.api.extension.ExtendWith; import org.mockito.Mock; import org.mockito.junit.jupiter.MockitoExtension; import static org.assertj.core.api.Assertions.assertThat; import static org.mockito.ArgumentMatchers.any; import static org.mockito.Mockito.doThrow; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.verify; import static org.mockito.Mockito.when; @ExtendWith(MockitoExtension.class) public class ReadHandlerTest extends CACertificateTestBase{ private final ReadHandler handler = new ReadHandler(); @Test public void handleRequest_SimpleSuccess() { final ResourceModel model = ResourceModel.builder().id(CA_CERT_ID).build(); final ResourceHandlerRequest request = defaultRequestBuilder(model) .previousResourceState(null) .build(); final ArgumentCaptor describeCaCertificateRequest = ArgumentCaptor.forClass(DescribeCaCertificateRequest.class); when(iotClient.describeCACertificate(any(DescribeCaCertificateRequest.class))) .thenReturn(DescribeCaCertificateResponse.builder() .certificateDescription( CACertificateDescription.builder() .status(CA_CERT_STATUS_ACTIVE) .autoRegistrationStatus(AutoRegistrationStatus.DISABLE.toString()) .certificatePem(CA_CERT_PEM) .certificateId(CA_CERT_ID) .certificateArn(CA_CERT_ARN) .build()) .build()); final ProgressEvent response = handler.handleRequest(proxy, request, new CallbackContext(), proxyClient, LOGGER); verify(iotClient).describeCACertificate(describeCaCertificateRequest.capture()); assertThat(describeCaCertificateRequest.getValue().certificateId()).isEqualTo(CA_CERT_ID); assertThat(response).isNotNull(); assertThat(response.getStatus()).isEqualTo(OperationStatus.SUCCESS); assertThat(response.getCallbackContext()).isNull(); assertThat(response.getCallbackDelaySeconds()).isEqualTo(0); assertThat(response.getResourceModels()).isNull(); assertThat(response.getMessage()).isNull(); assertThat(response.getErrorCode()).isNull(); assertThat(response.getResourceModel().getId()).isEqualTo(CA_CERT_ID); assertThat(response.getResourceModel().getArn()).isEqualTo(CA_CERT_ARN); assertThat(response.getResourceModel().getCACertificatePem()).isEqualTo(CA_CERT_PEM); assertThat(response.getResourceModel().getStatus()).isEqualTo(CA_CERT_STATUS_ACTIVE); assertThat(response.getResourceModel().getAutoRegistrationStatus()).isEqualTo(CA_CERT_AUTO_REGISTRATION_DISABLE); } @Test public void handleRequest_InternalFailure() { final ResourceModel model = ResourceModel.builder().id(CA_CERT_ID).build(); final ResourceHandlerRequest request = defaultRequestBuilder(model).build(); doThrow(InternalFailureException.builder().build()) .when(iotClient) .describeCACertificate(any(DescribeCaCertificateRequest.class)); Assertions.assertThrows(CfnServiceInternalErrorException.class, () -> handler.handleRequest(proxy, request, new CallbackContext(), proxyClient, LOGGER)); } @Test public void handleRequest_InvalidRequest() { final ResourceModel model = ResourceModel.builder().id(CA_CERT_ID).build(); final ResourceHandlerRequest request = defaultRequestBuilder(model).build(); doThrow(InvalidRequestException.builder().build()) .when(iotClient) .describeCACertificate(any(DescribeCaCertificateRequest.class)); Assertions.assertThrows(CfnInvalidRequestException.class, () -> handler.handleRequest(proxy, request, new CallbackContext(), proxyClient, LOGGER)); } @Test public void handleRequest_ResourceNotFound() { final ResourceModel model = ResourceModel.builder().id(CA_CERT_ID).build(); final ResourceHandlerRequest request = defaultRequestBuilder(model).build(); doThrow(ResourceNotFoundException.builder().build()) .when(iotClient) .describeCACertificate(any(DescribeCaCertificateRequest.class)); Assertions.assertThrows(CfnNotFoundException.class, () -> handler.handleRequest(proxy, request, new CallbackContext(), proxyClient, LOGGER)); } @Test public void handleRequest_ThrottlingException() { final ResourceModel model = ResourceModel.builder().id(CA_CERT_ID).build(); final ResourceHandlerRequest request = defaultRequestBuilder(model).build(); doThrow(ThrottlingException.builder().build()) .when(iotClient) .describeCACertificate(any(DescribeCaCertificateRequest.class)); Assertions.assertThrows(CfnThrottlingException.class, () -> handler.handleRequest(proxy, request, new CallbackContext(), proxyClient, LOGGER)); } }