package software.amazon.lightsail.loadbalancertlscertificate; import java.time.Duration; import software.amazon.awssdk.services.lightsail.LightsailClient; import software.amazon.cloudformation.proxy.AmazonWebServicesClientProxy; import software.amazon.cloudformation.proxy.OperationStatus; import software.amazon.cloudformation.proxy.ProgressEvent; import software.amazon.cloudformation.proxy.ProxyClient; import software.amazon.cloudformation.proxy.ResourceHandlerRequest; import org.junit.jupiter.api.AfterEach; 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 software.amazon.lightsail.loadbalancertlscertificate.helpers.handler.LoadBalancerTlsCertificateHandler; import static org.assertj.core.api.Assertions.assertThat; import static org.mockito.Mockito.*; @ExtendWith(MockitoExtension.class) public class DeleteHandlerTest extends AbstractTestBase { @Mock private AmazonWebServicesClientProxy proxy; @Mock private ProxyClient<LightsailClient> proxyClient; @Mock LightsailClient sdkClient; @Mock private LoadBalancerTlsCertificateHandler loadBalancerTlsCertificateHandler; @BeforeEach public void setup() { proxy = new AmazonWebServicesClientProxy(logger, MOCK_CREDENTIALS, () -> Duration.ofSeconds(600).toMillis()); sdkClient = mock(LightsailClient.class); proxyClient = MOCK_PROXY(proxy, sdkClient); loadBalancerTlsCertificateHandler = mock(LoadBalancerTlsCertificateHandler.class); } @AfterEach public void tear_down() { } @Test public void handleRequest_SimpleSuccess() { final DeleteHandler deleteHandler = spy(new DeleteHandler()); final CallbackContext callbackContext = new CallbackContext(); final ResourceModel model = ResourceModel.builder().build(); final ResourceHandlerRequest<ResourceModel> request = ResourceHandlerRequest.<ResourceModel>builder() .desiredResourceState(model) .build(); doReturn(loadBalancerTlsCertificateHandler) .when(deleteHandler).getLoadBalancerTlsCertificateHandler(any(), any(), any(), any(), any()); when(loadBalancerTlsCertificateHandler.handleDelete(any())) .thenReturn(ProgressEvent.progress(model, callbackContext)); final ProgressEvent<ResourceModel, CallbackContext> response = deleteHandler.handleRequest(proxy, request, new CallbackContext(), proxyClient, logger); verify(loadBalancerTlsCertificateHandler, times(1)).handleDelete(ProgressEvent.progress(model, callbackContext)); assertThat(response).isNotNull(); assertThat(response.getStatus()).isEqualTo(OperationStatus.SUCCESS); assertThat(response.getCallbackDelaySeconds()).isEqualTo(0); assertThat(response.getResourceModel()).isNull(); assertThat(response.getResourceModels()).isNull(); assertThat(response.getMessage()).isNull(); assertThat(response.getErrorCode()).isNull(); } }