11// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
22// SPDX-License-Identifier: Apache-2.0
33
4- import aws.sdk.kotlin.runtime.auth.credentials.EnvironmentCredentialsProvider
54import aws.sdk.kotlin.services.kendra.KendraClient
65import aws.sdk.kotlin.services.secretsmanager.SecretsManagerClient
76import aws.sdk.kotlin.services.secretsmanager.model.GetSecretValueRequest
@@ -23,11 +22,14 @@ 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
2627import java.util.UUID
2728
2829@TestInstance(TestInstance .Lifecycle .PER_CLASS )
2930@TestMethodOrder(MethodOrderer .OrderAnnotation ::class )
3031class KendraTest {
32+ private val logger: Logger = LoggerFactory .getLogger(KendraTest ::class .java)
3133 private var kendra: KendraClient ? = null
3234 private var indexName = " "
3335 private var indexDescription = " "
@@ -55,29 +57,6 @@ class KendraTest {
5557 dataSourceDescription = values.dataSourceDescription.toString()
5658 dataSourceRoleArn = values.dataSourceRoleArn.toString()
5759 text = values.text.toString()
58-
59- /*
60- try {
61- KendraTest::class.java.classLoader.getResourceAsStream("config.properties").use { input ->
62- val prop = Properties()
63- if (input == null) {
64- println("Sorry, unable to find config.properties")
65- return
66- }
67- prop.load(input)
68- indexName = prop.getProperty("indexName")
69- indexRoleArn = prop.getProperty("indexRoleArn")
70- indexDescription = prop.getProperty("indexDescription")
71- s3BucketName = prop.getProperty("s3BucketName")
72- dataSourceName = prop.getProperty("dataSourceName")
73- dataSourceDescription = prop.getProperty("dataSourceDescription")
74- dataSourceRoleArn = prop.getProperty("dataSourceRoleArn")
75- text = prop.getProperty("text")
76- }
77- } catch (ex: IOException) {
78- ex.printStackTrace()
79- }
80- */
8160 }
8261
8362 @Test
@@ -86,7 +65,7 @@ class KendraTest {
8665 runBlocking {
8766 indexId = createIndex(indexDescription, indexName, indexRoleArn)
8867 assertTrue(! indexId.isEmpty())
89- println (" Test 1 passed" )
68+ logger.info (" Test 1 passed" )
9069 }
9170
9271 @Test
@@ -95,47 +74,47 @@ class KendraTest {
9574 runBlocking {
9675 dataSourceId = createDataSource(s3BucketName, dataSourceName, dataSourceDescription, indexId, dataSourceRoleArn)
9776 assertTrue(! dataSourceId.isEmpty())
98- println (" Test 2 passed" )
77+ logger.info (" Test 2 passed" )
9978 }
10079
10180 @Test
10281 @Order(3 )
10382 fun syncDataSource () =
10483 runBlocking {
10584 startDataSource(indexId, dataSourceId)
106- println (" Test 3 passed" )
85+ logger.info (" Test 3 passed" )
10786 }
10887
10988 @Test
11089 @Order(4 )
11190 fun listSyncJobs () =
11291 runBlocking {
11392 listSyncJobs(indexId, dataSourceId)
114- println (" Test 4 passed" )
93+ logger.info (" Test 4 passed" )
11594 }
11695
11796 @Test
11897 @Order(5 )
11998 fun queryIndex () =
12099 runBlocking {
121100 querySpecificIndex(indexId, text)
122- println (" Test 5 passed" )
101+ logger.info (" Test 5 passed" )
123102 }
124103
125104 @Test
126105 @Order(6 )
127106 fun deleteDataSource () =
128107 runBlocking {
129108 deleteSpecificDataSource(indexId, dataSourceId)
130- println (" Test 6 passed" )
109+ logger.info (" Test 6 passed" )
131110 }
132111
133112 @Test
134113 @Order(7 )
135114 fun deleteIndex () =
136115 runBlocking {
137116 deleteSpecificIndex(indexId)
138- println (" Test 7 passed" )
117+ logger.info (" Test 7 passed" )
139118 }
140119
141120 private suspend fun getSecretValues (): String {
@@ -146,7 +125,6 @@ class KendraTest {
146125 }
147126 SecretsManagerClient {
148127 region = " us-east-1"
149- credentialsProvider = EnvironmentCredentialsProvider ()
150128 }.use { secretClient ->
151129 val valueResponse = secretClient.getSecretValue(valueRequest)
152130 return valueResponse.secretString.toString()
0 commit comments