Create AWSAthenaServiceIntegrationTest.java

Author: scmacdon

javav2/example_code/athena/src/test/java/AWSAthenaServiceIntegrationTest.java

@@ -0,0 +1,110 @@
+import aws.example.athena.*;
+import org.junit.jupiter.api.*;
+import software.amazon.awssdk.regions.Region;
+import software.amazon.awssdk.services.athena.AthenaClient;
+import software.amazon.awssdk.services.athena.model.AthenaException;
+import java.io.*;
+import java.util.Properties;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+@TestInstance(TestInstance.Lifecycle.PER_METHOD)
+@TestMethodOrder(MethodOrderer.OrderAnnotation.class)
+public class AWSAthenaServiceIntegrationTest {
+
+    private static AthenaClient athenaClient;
+    private static String nameQuery;
+
+    @BeforeAll
+    public static void setUp() throws IOException {
+        athenaClient = AthenaClient.builder()
+                .region(Region.US_WEST_2)
+                .build();
+
+        try (InputStream input = AWSAthenaServiceIntegrationTest.class.getClassLoader().getResourceAsStream("config.properties")) {
+
+            Properties prop = new Properties();
+
+            if (input == null) {
+                System.out.println("Sorry, unable to find config.properties");
+                return;
+            }
+
+            //load a properties file from class path, inside static method
+            prop.load(input);
+
+            // Populate the data members required for all tests
+            nameQuery = prop.getProperty("nameQuery");
+
+
+        } catch (IOException ex) {
+            ex.printStackTrace();
+        }
+    }
+    @Test
+    @Order(1)
+    public void whenInitializingAWSAthenaService_thenNotNull() {
+        assertNotNull(athenaClient);
+        System.out.println("Test 1 passed");
+    }
+
+    @Test
+    @Order(2)
+    public void CreateNamedQueryExample() {
+
+       CreateNamedQueryExample.createNamedQuery(athenaClient, nameQuery);
+       System.out.println("Test 2 passed");
+    }
+
+    @Test
+    @Order(3)
+    public void ListNamedQueryExample() {
+
+       ListNamedQueryExample.listNamedQueries(athenaClient);
+       System.out.println("Test 3 passed");
+    }
+
+    @Test
+    @Order(4)
+    public void ListQueryExecutionsExample() {
+
+        ListQueryExecutionsExample.listQueryIds(athenaClient);
+        System.out.println("Test 4 passed");
+    }
+
+    @Test
+    @Order(5)
+    public void DeleteNamedQueryExample()
+    {
+        String sampleNamedQueryId = DeleteNamedQueryExample.getNamedQueryId(athenaClient, nameQuery);
+        DeleteNamedQueryExample.deleteQueryName(athenaClient, sampleNamedQueryId);
+        System.out.println("Test 5 passed");
+
+    }
+
+
+    @Test
+    @Order(6)
+    public void StartQueryExample() {
+
+        try {
+            String queryExecutionId = StartQueryExample.submitAthenaQuery(athenaClient);
+            StartQueryExample.waitForQueryToComplete(athenaClient, queryExecutionId);
+            StartQueryExample.processResultRows(athenaClient, queryExecutionId);
+             System.out.println("Test 6 passed");
+
+        }catch (InterruptedException e) {
+         e.getMessage();
+        }
+    }
+
+    @Test
+    @Order(7)
+    public void StopQueryExecutionExample() {
+
+        String sampleQueryExecutionId = StopQueryExecutionExample.submitAthenaQuery(athenaClient);
+        StopQueryExecutionExample.stopAthenaQuery(athenaClient, sampleQueryExecutionId);
+        System.out.println("Test 7 passed");
+    }
+}
+