Skip to content

Commit

Permalink
refactor: rename emr plugins tasks (#578)
Browse files Browse the repository at this point in the history
  • Loading branch information
mgabelle committed Feb 3, 2025
1 parent b847a58 commit 5e6ebf8
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@
tasks:
- id: create_cluster
type: io.kestra.plugin.aws.emr.CreateCluster
type: io.kestra.plugin.aws.emr.CreateClusterAndSubmitSteps
accessKeyId: <access-key>
secretKeyId: <secret-key>
region: eu-west-3
Expand All @@ -64,7 +64,7 @@
)
}
)
public class CreateCluster extends AbstractEmrTask implements RunnableTask<CreateCluster.Output> {
public class CreateClusterAndSubmitSteps extends AbstractEmrTask implements RunnableTask<CreateClusterAndSubmitSteps.Output> {

@Schema(title = "Cluster Name.")
@NotNull
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@
tasks:
- id: add_steps_emr
type: io.kestra.plugin.aws.emr.AddJobFlowsSteps
type: io.kestra.plugin.aws.emr.SubmitSteps
accessKeyId: "<access-key>"
secretKeyId: "<secret-key>"
region: "eu-west-3"
Expand All @@ -55,7 +55,7 @@
)
}
)
public class AddJobFlowsSteps extends AbstractEmrTask implements RunnableTask<VoidOutput> {
public class SubmitSteps extends AbstractEmrTask implements RunnableTask<VoidOutput> {
@Schema(title = "Cluster ID.")
@NotNull
private Property<String> clusterId;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ class EmrIntegrationTest {

@Test
void createCluster() throws Exception {
CreateCluster createCluster = CreateCluster.builder()
CreateClusterAndSubmitSteps createCluster = CreateClusterAndSubmitSteps.builder()
.accessKeyId(Property.of(accessKeyId))
.secretKeyId(Property.of(secretKeyId))
.sessionToken(Property.of(sessionToken))
Expand All @@ -50,7 +50,7 @@ void createCluster() throws Exception {
.keepJobFlowAliveWhenNoSteps(Property.of(true))
.build();

CreateCluster.Output output = createCluster.run(runContextFactory.of());
CreateClusterAndSubmitSteps.Output output = createCluster.run(runContextFactory.of());
assertNotNull(output.getJobFlowId());
}

Expand All @@ -69,7 +69,7 @@ void deleteCluster() throws Exception {

@Test
void addStepsToCluster() throws Exception {
AddJobFlowsSteps addJobFlowsSteps = AddJobFlowsSteps.builder()
SubmitSteps addJobFlowsSteps = SubmitSteps.builder()
.accessKeyId(Property.of(accessKeyId))
.secretKeyId(Property.of(secretKeyId))
.sessionToken(Property.of(sessionToken))
Expand Down

0 comments on commit 5e6ebf8

Please sign in to comment.