Added tests for legacy aws patterns

sksamuel · May 27, 2023 · 8493468 · 8493468
1 parent 6d8a490
commit 8493468
Showing 1 changed file with 34 additions and 1 deletion.
diff --git a/...lite/aws/AwsSecretsManagerResolverTest.kt → ...s/AwsSecretsManagerContextResolverTest.kt b/...lite/aws/AwsSecretsManagerResolverTest.kt → ...s/AwsSecretsManagerContextResolverTest.kt
@@ -21,7 +21,7 @@ import org.testcontainers.utility.DockerImageName
 import java.util.Properties
 
 @OptIn(ExperimentalHoplite::class)
-class AwsSecretsManagerResolverTest : FunSpec() {
+class AwsSecretsManagerContextResolverTest : FunSpec() {
 
   private val localstack = LocalStackContainer(DockerImageName.parse("localstack/localstack:1.3.1"))
     .withServices(LocalStackContainer.Service.SECRETSMANAGER)
@@ -61,6 +61,39 @@ class AwsSecretsManagerResolverTest : FunSpec() {
         .a.shouldBe("secret!")
     }
 
+    test("legacy1 prefix pattern should be detected and used") {
+      val props = Properties()
+      props["a"] = "awssm://foo"
+      ConfigLoaderBuilder.newBuilder()
+        .addResolver(AwsSecretsManagerContextResolvers { client })
+        .addPropertySource(PropsPropertySource(props))
+        .build()
+        .loadConfigOrThrow<ConfigHolder>()
+        .a.shouldBe("secret!")
+    }
+
+    test("legacy2 prefix pattern should be detected and used") {
+      val props = Properties()
+      props["a"] = "awssecret://foo"
+      ConfigLoaderBuilder.newBuilder()
+        .addResolver(AwsSecretsManagerContextResolvers { client })
+        .addPropertySource(PropsPropertySource(props))
+        .build()
+        .loadConfigOrThrow<ConfigHolder>()
+        .a.shouldBe("secret!")
+    }
+
+    test("legacy3 prefix pattern should be detected and used") {
+      val props = Properties()
+      props["a"] = "secretsmanager://foo"
+      ConfigLoaderBuilder.newBuilder()
+        .addResolver(AwsSecretsManagerContextResolvers { client })
+        .addPropertySource(PropsPropertySource(props))
+        .build()
+        .loadConfigOrThrow<ConfigHolder>()
+        .a.shouldBe("secret!")
+    }
+
     test("unknown secret should return error and include key") {
       val props = Properties()
       props["a"] = "\${{ aws-secrets-manager:qwerty }}"