22// SPDX-License-Identifier: Apache-2.0
33// SPDX-License-Identifier: Apache-2.0
44
5- import aws.sdk.kotlin.runtime.auth.credentials.EnvironmentCredentialsProvider
65import aws.sdk.kotlin.services.secretsmanager.SecretsManagerClient
76import aws.sdk.kotlin.services.secretsmanager.model.GetSecretValueRequest
87import com.google.gson.Gson
@@ -23,10 +22,13 @@ import org.junit.jupiter.api.Order
2322import org.junit.jupiter.api.Test
2423import org.junit.jupiter.api.TestInstance
2524import org.junit.jupiter.api.TestMethodOrder
25+ import org.slf4j.Logger
26+ import org.slf4j.LoggerFactory
2627
2728@TestInstance(TestInstance .Lifecycle .PER_CLASS )
2829@TestMethodOrder(OrderAnnotation ::class )
2930class AthenaTest {
31+ private val logger: Logger = LoggerFactory .getLogger(AthenaTest ::class .java)
3032 private var nameQuery: String? = null
3133 private var queryString: String? = null
3234 private var database: String? = null
@@ -43,39 +45,28 @@ class AthenaTest {
4345 queryString = values.queryString.toString()
4446 database = values.database.toString()
4547 outputLocation = values.outputLocation.toString()
46-
47- // Uncomment this code block if you prefer using a config.properties file to retrieve AWS values required for these tests.
48- /*
49- val input: InputStream = this.javaClass.getClassLoader().getResourceAsStream("config.properties")
50- val prop = Properties()
51- prop.load(input)
52- nameQuery = prop.getProperty("nameQuery")
53- queryString = prop.getProperty("queryString")
54- database = prop.getProperty("database")
55- outputLocation = prop.getProperty("outputLocation")
56- */
5748 }
5849
5950 @Test
6051 @Order(1 )
6152 fun createNamedQueryTest () = runBlocking {
6253 queryId = createNamedQuery(queryString.toString(), nameQuery.toString(), database.toString())
6354 queryId?.let { assertTrue(it.isNotEmpty()) }
64- println (" Test 1 passed" )
55+ logger.info (" Test 1 passed" )
6556 }
6657
6758 @Test
6859 @Order(2 )
6960 fun listNamedQueryTest () = runBlocking {
7061 listNamedQueries()
71- println (" Test 2 passed" )
62+ logger.info (" Test 2 passed" )
7263 }
7364
7465 @Test
7566 @Order(3 )
7667 fun listQueryExecutionsTest () = runBlocking {
7768 listQueryIds()
78- println (" Test 3 passed" )
69+ logger.info (" Test 3 passed" )
7970 }
8071
8172 @Test
@@ -84,14 +75,14 @@ class AthenaTest {
8475 val queryExecutionId = submitAthenaQuery(queryString.toString(), database.toString(), outputLocation.toString())
8576 waitForQueryToComplete(queryExecutionId)
8677 processResultRows(queryExecutionId)
87- println (" Test 4 passed" )
78+ logger.info (" Test 4 passed" )
8879 }
8980
9081 @Test
9182 @Order(5 )
9283 fun deleteNamedQueryTest () = runBlocking {
9384 deleteQueryName(queryId)
94- println (" Test 5 passed" )
85+ logger.info (" Test 5 passed" )
9586 }
9687
9788 private suspend fun getSecretValues (): String {
@@ -101,7 +92,6 @@ class AthenaTest {
10192 }
10293 SecretsManagerClient {
10394 region = " us-east-1"
104- credentialsProvider = EnvironmentCredentialsProvider ()
10595 }.use { secretClient ->
10696 val valueResponse = secretClient.getSecretValue(valueRequest)
10797 return valueResponse.secretString.toString()
0 commit comments