1 /* 2 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 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 * A copy of the License is located at 7 * 8 * http://aws.amazon.com/apache2.0 9 * 10 * or in the "license" file accompanying this file. This file is distributed 11 * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either 12 * express or implied. See the License for the specific language governing 13 * permissions and limitations under the License. 14 */ 15 16 package utils.resources; 17 18 import java.time.Duration; 19 import software.amazon.awssdk.testutils.Waiter; 20 import software.amazon.awssdk.utils.Logger; 21 import utils.resources.RequiredResources.ResourceCreationPolicy; 22 import utils.resources.TestResource.ResourceStatus; 23 24 25 public class TestResourceUtils { 26 private static final Logger log = Logger.loggerFor(TestResourceUtils.class); 27 createResource(TestResource resource, ResourceCreationPolicy policy)28 public static void createResource(TestResource resource, ResourceCreationPolicy policy) throws InterruptedException { 29 TestResource.ResourceStatus finalizedStatus = waitForFinalizedStatus(resource); 30 if (policy == ResourceCreationPolicy.ALWAYS_RECREATE) { 31 if (finalizedStatus != ResourceStatus.NOT_EXIST) { 32 resource.delete(true); 33 } 34 resource.create(true); 35 } else if (policy == ResourceCreationPolicy.REUSE_EXISTING) { 36 switch (finalizedStatus) { 37 case AVAILABLE: 38 log.info(() -> "Found existing resource " + resource + " that could be reused..."); 39 return; 40 case EXIST_INCOMPATIBLE_RESOURCE: 41 resource.delete(true); 42 resource.create(true); 43 // fallthru 44 case NOT_EXIST: 45 resource.create(true); 46 break; 47 default: 48 break; 49 } 50 } 51 } 52 deleteResource(TestResource resource)53 public static void deleteResource(TestResource resource) throws InterruptedException { 54 ResourceStatus finalizedStatus = waitForFinalizedStatus(resource); 55 if (finalizedStatus != ResourceStatus.NOT_EXIST) { 56 resource.delete(false); 57 } 58 } 59 waitForFinalizedStatus(TestResource resource)60 public static ResourceStatus waitForFinalizedStatus(TestResource resource) throws InterruptedException { 61 return Waiter.run(resource::getResourceStatus) 62 .until(s -> s != ResourceStatus.TRANSIENT) 63 .orFailAfter(Duration.ofMinutes(10)); 64 } 65 } 66