package software.amazon.rds.customdbengineversion; import static org.mockito.ArgumentMatchers.any; import static org.mockito.Mockito.atLeastOnce; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.times; import static org.mockito.Mockito.verify; import static org.mockito.Mockito.verifyNoMoreInteractions; import static org.mockito.Mockito.when; import java.time.Duration; 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 lombok.Getter; import software.amazon.awssdk.services.rds.RdsClient; import software.amazon.awssdk.services.rds.model.CustomDbEngineVersionNotFoundException; import software.amazon.awssdk.services.rds.model.DescribeDbEngineVersionsRequest; import software.amazon.cloudformation.proxy.AmazonWebServicesClientProxy; import software.amazon.cloudformation.proxy.HandlerErrorCode; import software.amazon.cloudformation.proxy.ProxyClient; import software.amazon.rds.common.handler.HandlerConfig; import software.amazon.rds.test.common.core.HandlerName; @ExtendWith(MockitoExtension.class) public class ReadHandlerTest extends AbstractHandlerTest { @Mock @Getter private AmazonWebServicesClientProxy proxy; @Mock @Getter private ProxyClient rdsProxy; @Mock RdsClient rdsClient; @Getter private ReadHandler handler; @Override public HandlerName getHandlerName() { return HandlerName.READ; } @BeforeEach public void setup() { handler = new ReadHandler(HandlerConfig.builder().backoff(TEST_BACKOFF_DELAY).build()); proxy = new AmazonWebServicesClientProxy(logger, MOCK_CREDENTIALS, () -> Duration.ofSeconds(600).toMillis()); rdsClient = mock(RdsClient.class); rdsProxy = mockProxy(proxy, rdsClient); } @AfterEach public void tear_down() { verify(rdsClient, atLeastOnce()).serviceName(); verifyNoMoreInteractions(rdsClient); verifyAccessPermissions(rdsClient); } @Test public void handleRequest_ReadSuccess() { final software.amazon.cloudformation.proxy.ProgressEvent progress = test_handleRequest_base( new CallbackContext(), () -> DB_ENGINE_VERSION_AVAILABLE, () -> RESOURCE_MODEL_BUILDER().tags(TAG_LIST).build(), expectSuccess() ); progress.getResourceModel(); verify(rdsProxy.client(), times(1)).describeDBEngineVersions(any(DescribeDbEngineVersionsRequest.class)); } @Test public void handleRequest_NotFound() { when(rdsProxy.client().describeDBEngineVersions(any(DescribeDbEngineVersionsRequest.class))) .thenThrow(CustomDbEngineVersionNotFoundException.builder().message("Not Found").build()); test_handleRequest_base( new CallbackContext(), null, () -> RESOURCE_MODEL, expectFailed(HandlerErrorCode.NotFound) ); verify(rdsProxy.client(), times(1)).describeDBEngineVersions(any(DescribeDbEngineVersionsRequest.class)); } @Test public void handleRequest_RuntimeException() { when(rdsProxy.client().describeDBEngineVersions(any(DescribeDbEngineVersionsRequest.class))) .thenThrow(new RuntimeException("test exception")); test_handleRequest_base( new CallbackContext(), null, () -> RESOURCE_MODEL, expectFailed(HandlerErrorCode.InternalFailure) ); verify(rdsProxy.client(), times(1)).describeDBEngineVersions(any(DescribeDbEngineVersionsRequest.class)); } }