tests(): enable tests + flaky (#679)

Author: Malaydewangan09

src/main/java/io/kestra/plugin/aws/athena/Query.java

@@ -186,7 +186,7 @@ public QueryOutput run(RunContext runContext) throws Exception {
             .resultConfiguration(resultConfiguration)
             .build();
 
-        try (var client = client(runContext)) {
+        try (var client = athenaClient(runContext)) {
             var startQueryExecution = client.startQueryExecution(startQueryExecutionRequest);
             var fetchType = runContext.render(this.fetchType).as(FetchType.class).orElseThrow();
             runContext.logger().info("Query created with Athena execution identifier {}", startQueryExecution.queryExecutionId());
@@ -257,6 +257,10 @@ else if (fetchType == FetchType.STORE) {
         }
     }
 
+    protected AthenaClient athenaClient(RunContext runContext) throws IllegalVariableEvaluationException {
+        return client(runContext);
+    }
+
     private AthenaClient client(final RunContext runContext) throws IllegalVariableEvaluationException {
         AwsClientConfig clientConfig = awsClientConfig(runContext);
         return ConnectionUtils.configureSyncClient(clientConfig, AthenaClient.builder()).build();

src/main/java/io/kestra/plugin/aws/ecr/GetAuthToken.java

@@ -66,7 +66,7 @@ public TokenOutput run(RunContext runContext) throws Exception {
         }
     }
 
-    private EcrClient client(final RunContext runContext) throws IllegalVariableEvaluationException {
+    protected EcrClient client(final RunContext runContext) throws IllegalVariableEvaluationException {
         final AwsClientConfig clientConfig = awsClientConfig(runContext);
         return ConnectionUtils.configureSyncClient(clientConfig, EcrClient.builder()).build();
     }

src/main/java/io/kestra/plugin/aws/emr/AbstractEmrTask.java

@@ -21,4 +21,8 @@ protected EmrClient client(final RunContext runContext) throws IllegalVariableEv
         final AwsClientConfig clientConfig = awsClientConfig(runContext);
         return ConnectionUtils.configureSyncClient(clientConfig, EmrClient.builder()).build();
     }
+
+    protected EmrClient emrClient(RunContext runContext) throws IllegalVariableEvaluationException {
+        return client(runContext);
+    }
 }

src/main/java/io/kestra/plugin/aws/emr/CreateClusterAndSubmitSteps.java

@@ -159,7 +159,7 @@ To launch the cluster in Amazon Virtual Private Cloud (Amazon VPC), set this par
 
     @Override
     public Output run(RunContext runContext) throws IllegalVariableEvaluationException {
-        try(var emrClient = this.client(runContext)) {
+        try (var emrClient = this.emrClient(runContext)) {
             //Create Applications to be loaded
             List<Application> applicationsList = runContext.render(this.applications).asList(String.class)
                 .stream()

src/main/java/io/kestra/plugin/aws/emr/DeleteCluster.java

@@ -57,7 +57,7 @@ public class DeleteCluster extends AbstractEmrTask implements RunnableTask<VoidO
 
     @Override
     public VoidOutput run(RunContext runContext) throws IllegalVariableEvaluationException {
-        try (EmrClient emrClient = this.client(runContext)) {
+        try (var emrClient = this.emrClient(runContext)) {
             emrClient.terminateJobFlows(throwConsumer(request -> request.jobFlowIds(runContext.render(this.clusterIds).asList(String.class))));
             return null;
         }

src/main/java/io/kestra/plugin/aws/emr/SubmitSteps.java

@@ -78,7 +78,7 @@ The Amazon Resource Name (ARN) of the runtime role for a step on the cluster. Th
 
     @Override
     public VoidOutput run(RunContext runContext) throws IllegalVariableEvaluationException {
-        try(var emrClient = this.client(runContext)) {
+        try (var emrClient = this.emrClient(runContext)) {
             List<software.amazon.awssdk.services.emr.model.StepConfig> jobSteps = steps.stream()
                 .map(throwFunction(stepConfig -> stepConfig.toStep(runContext)))
                 .toList();

src/main/java/io/kestra/plugin/aws/glue/AbstractGlueTask.java

@@ -21,4 +21,8 @@ protected GlueClient client(final RunContext runContext) throws IllegalVariableE
         final AwsClientConfig clientConfig = awsClientConfig(runContext);
         return ConnectionUtils.configureSyncClient(clientConfig, GlueClient.builder()).build();
     }
+
+    protected GlueClient glueClient(RunContext runContext) throws IllegalVariableEvaluationException {
+        return client(runContext);
+    }
 }

src/main/java/io/kestra/plugin/aws/glue/GetJobRun.java

@@ -82,7 +82,7 @@ public class GetJobRun extends AbstractGlueTask implements RunnableTask<Output>
 
     @Override
     public Output run(RunContext runContext) throws IllegalVariableEvaluationException {
-        try (GlueClient glueClient = this.client(runContext)) {
+        try (GlueClient glueClient = this.glueClient(runContext)) {
             String jobNameValue = runContext.render(this.jobName).as(String.class).orElseThrow();
             String runIdValue = null;
 

src/main/java/io/kestra/plugin/aws/glue/StartJobRun.java

@@ -90,7 +90,7 @@ public class StartJobRun extends AbstractGlueTask implements RunnableTask<Output
 
     @Override
     public Output run(RunContext runContext) throws IllegalVariableEvaluationException {
-        try (GlueClient glueClient = this.client(runContext)) {
+        try (GlueClient glueClient = this.glueClient(runContext)) {
             String jobNameValue = runContext.render(this.jobName).as(String.class).orElseThrow();
             String jobRunId = startJob(runContext, glueClient, jobNameValue);
 

src/main/java/io/kestra/plugin/aws/glue/StopJobRun.java

@@ -88,7 +88,7 @@ public Output run(RunContext runContext) throws Exception {
         boolean rwaitValue = runContext.render(this.wait).as(Boolean.class).orElse(true);
         Duration rintervalValue = runContext.render(this.interval).as(Duration.class).orElse(Duration.ofSeconds(1));
 
-        try (GlueClient glueClient = this.client(runContext)) {
+        try (GlueClient glueClient = this.glueClient(runContext)) {
             // Stop the job
             BatchStopJobRunResponse stopResponse = glueClient.batchStopJobRun(
                 BatchStopJobRunRequest.builder()