11// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
22// SPDX-License-Identifier: Apache-2.0
3- import aws.sdk.kotlin.runtime.auth.credentials.EnvironmentCredentialsProvider
43import aws.sdk.kotlin.services.secretsmanager.SecretsManagerClient
54import aws.sdk.kotlin.services.secretsmanager.model.GetSecretValueRequest
65import com.google.gson.Gson
@@ -22,12 +21,15 @@ import org.junit.jupiter.api.Order
2221import org.junit.jupiter.api.Test
2322import org.junit.jupiter.api.TestInstance
2423import org.junit.jupiter.api.TestMethodOrder
24+ import org.slf4j.Logger
25+ import org.slf4j.LoggerFactory
2526import java.util.Random
2627import java.util.concurrent.TimeUnit
2728
2829@TestInstance(TestInstance .Lifecycle .PER_CLASS )
2930@TestMethodOrder(OrderAnnotation ::class )
3031class ForecastKotlinTest {
32+ private val logger: Logger = LoggerFactory .getLogger(ForecastKotlinTest ::class .java)
3133 private var predictorARN = " "
3234 private var forecastArn = " " // set in test 3
3335 private var forecastName = " "
@@ -46,17 +48,6 @@ class ForecastKotlinTest {
4648 predictorARN = values.predARN.toString()
4749 forecastName = values.forecastName.toString() + randomNum
4850 dataSetName = values.dataSet.toString() + randomNum
49-
50- /*
51- // load the properties file.
52- val input: InputStream = this.javaClass.getClassLoader().getResourceAsStream("config.properties")
53- val prop = Properties()
54- prop.load(input)
55- forecastName = prop.getProperty("forecastName")
56- dataSetName = prop.getProperty("dataSetName")
57- predictorARN = prop.getProperty("predictorARN")
58- existingforecastDelete = prop.getProperty("existingforecastDelete")
59- */
6051 }
6152
6253 @Test
@@ -65,7 +56,7 @@ class ForecastKotlinTest {
6556 runBlocking {
6657 myDataSetARN = createForecastDataSet(dataSetName).toString()
6758 assertTrue(! myDataSetARN.isEmpty())
68- println (" Test 1 passed" )
59+ logger.info (" Test 1 passed" )
6960 }
7061
7162 @Test
@@ -74,47 +65,47 @@ class ForecastKotlinTest {
7465 runBlocking {
7566 forecastArn = createNewForecast(forecastName, predictorARN).toString()
7667 assertTrue(! forecastArn.isEmpty())
77- println (" Test 2 passed" )
68+ logger.info (" Test 2 passed" )
7869 }
7970
8071 @Test
8172 @Order(3 )
8273 fun listDataSets () =
8374 runBlocking {
8475 listForecastDataSets()
85- println (" Test 3 passed" )
76+ logger.info (" Test 3 passed" )
8677 }
8778
8879 @Test
8980 @Order(4 )
9081 fun listDataSetGroups () =
9182 runBlocking {
9283 listDataGroups()
93- println (" Test 4 passed" )
84+ logger.info (" Test 4 passed" )
9485 }
9586
9687 @Test
9788 @Order(5 )
9889 fun listForecasts () =
9990 runBlocking {
10091 listAllForeCasts()
101- println (" Test 5 passed" )
92+ logger.info (" Test 5 passed" )
10293 }
10394
10495 @Test
10596 @Order(6 )
10697 fun describeForecast () =
10798 runBlocking {
10899 describe(forecastArn)
109- println (" Test 6 passed" )
100+ logger.info (" Test 6 passed" )
110101 }
111102
112103 @Test
113104 @Order(7 )
114105 fun deleteDataSet () =
115106 runBlocking {
116107 deleteForecastDataSet(myDataSetARN)
117- println (" Test 7 passed" )
108+ logger.info (" Test 7 passed" )
118109 }
119110
120111 @Test
@@ -124,7 +115,7 @@ class ForecastKotlinTest {
124115 println (" Wait 40 mins for resource to become available." )
125116 TimeUnit .MINUTES .sleep(40 )
126117 delForecast(forecastArn)
127- println (" Test 8 passed" )
118+ logger.info (" Test 8 passed" )
128119 }
129120
130121 private suspend fun getSecretValues (): String {
@@ -135,7 +126,6 @@ class ForecastKotlinTest {
135126 }
136127 SecretsManagerClient {
137128 region = " us-east-1"
138- credentialsProvider = EnvironmentCredentialsProvider ()
139129 }.use { secretClient ->
140130 val valueResponse = secretClient.getSecretValue(valueRequest)
141131 return valueResponse.secretString.toString()
0 commit comments