1010import software .amazon .awssdk .services .secretsmanager .SecretsManagerClient ;
1111import software .amazon .awssdk .services .secretsmanager .model .GetSecretValueRequest ;
1212import software .amazon .awssdk .services .secretsmanager .model .GetSecretValueResponse ;
13+
1314import java .io .*;
15+
1416import static org .junit .jupiter .api .Assertions .*;
1517
1618/**
@@ -32,9 +34,9 @@ public class EMRTest {
3234 @ BeforeAll
3335 public static void setUp () throws IOException {
3436 emrClient = EmrClient .builder ()
35- .region (Region .US_WEST_2 )
36- .credentialsProvider (EnvironmentVariableCredentialsProvider .create ())
37- .build ();
37+ .region (Region .US_WEST_2 )
38+ .credentialsProvider (EnvironmentVariableCredentialsProvider .create ())
39+ .build ();
3840
3941 // Get the values to run these tests from AWS Secrets Manager.
4042 Gson gson = new Gson ();
@@ -50,7 +52,7 @@ public static void setUp() throws IOException {
5052 // Uncomment this code block if you prefer using a config.properties file to
5153 // retrieve AWS values required for these tests.
5254 /*
53- *
55+ *
5456 * try (InputStream input =
5557 * EMRTest.class.getClassLoader().getResourceAsStream("config.properties")) {
5658 * Properties prop = new Properties();
@@ -66,7 +68,7 @@ public static void setUp() throws IOException {
6668 * logUri = prop.getProperty("logUri");
6769 * name = prop.getProperty("name");
6870 * existingClusterId= prop.getProperty("existingClusterId");
69- *
71+ *
7072 * } catch (IOException ex) {
7173 * ex.printStackTrace();
7274 * }
@@ -83,71 +85,51 @@ public void createClusterTest() {
8385
8486 @ Test
8587 @ Order (2 )
86- public void describeClusterTest () {
87- assertDoesNotThrow (() -> DescribeCluster .describeMyCluster (emrClient , existingClusterId ));
88- System .out .println ("Test 2 passed" );
89- }
90-
91- @ Test
92- @ Order (3 )
9388 public void listClusterTest () {
9489 assertDoesNotThrow (() -> ListClusters .listAllClusters (emrClient ));
9590 System .out .println ("Test 3 passed" );
9691 }
9792
9893 @ Test
99- @ Order (4 )
94+ @ Order (3 )
10095 public void createEmrFleetTest () {
10196 assertDoesNotThrow (() -> CreateEmrFleet .createFleet (emrClient ));
10297 System .out .println ("Test 4 passed" );
10398 }
10499
105- @ Test
106- @ Order (5 )
107- public void addStepsTest () {
108- assertDoesNotThrow (() -> AddSteps .addNewStep (emrClient , jobFlowId , jar , myClass ));
109- System .out .println ("Test 5 passed" );
110- }
111100
112101 @ Test
113- @ Order (6 )
102+ @ Order (4 )
114103 public void createSparkClusterTest () {
115104 assertDoesNotThrow (() -> CreateSparkCluster .createCluster (emrClient , jar , myClass , keys , logUri , name ));
116105 System .out .println ("Test 6 passed" );
117106 }
118107
119108 @ Test
120- @ Order (7 )
109+ @ Order (5 )
121110 public void createHiveClusterTest () {
122111 assertDoesNotThrow (() -> CreateHiveCluster .createCluster (emrClient , jar , myClass , keys , logUri , name ));
123112 System .out .println ("Test 7 passed" );
124113
125114 }
126115
127116 @ Test
128- @ Order (8 )
117+ @ Order (6 )
129118 public void customEmrfsMaterialsTest () {
130119 assertDoesNotThrow (() -> CustomEmrfsMaterials .createEmrfsCluster (emrClient , jar , myClass , keys , logUri , name ));
131120 System .out .println ("Test 8 passed" );
132121 }
133122
134- @ Test
135- @ Order (9 )
136- public void terminateJobFlowTest () {
137- assertDoesNotThrow (() -> TerminateJobFlow .terminateFlow (emrClient , existingClusterId ));
138- System .out .println ("Test 9 passed" );
139- }
140-
141123 private static String getSecretValues () {
142124 SecretsManagerClient secretClient = SecretsManagerClient .builder ()
143- .region (Region .US_EAST_1 )
144- .credentialsProvider (EnvironmentVariableCredentialsProvider .create ())
145- .build ();
125+ .region (Region .US_EAST_1 )
126+ .credentialsProvider (EnvironmentVariableCredentialsProvider .create ())
127+ .build ();
146128 String secretName = "text/emr" ;
147129
148130 GetSecretValueRequest valueRequest = GetSecretValueRequest .builder ()
149- .secretId (secretName )
150- .build ();
131+ .secretId (secretName )
132+ .build ();
151133
152134 GetSecretValueResponse valueResponse = secretClient .getSecretValue (valueRequest );
153135 return valueResponse .secretString ();
0 commit comments