update unit tests to have endpointRegion

This commit is contained in:
Vincent Viloria 2025-02-28 10:10:35 -08:00
parent 58aa4ebae7
commit 37d69fc845
2 changed files with 8 additions and 6 deletions

View file

@ -20,7 +20,7 @@ import org.junit.Test;
import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertEquals;
public class KclSTSAssumeRoleSessionCredentialsProviderTest { public class KclStsAssumeRoleCredentialsProviderTest {
private static final String ARN = "arn"; private static final String ARN = "arn";
private static final String SESSION_NAME = "sessionName"; private static final String SESSION_NAME = "sessionName";
@ -31,22 +31,23 @@ public class KclSTSAssumeRoleSessionCredentialsProviderTest {
*/ */
@Test @Test
public void testConstructorWithoutOptionalParams() { public void testConstructorWithoutOptionalParams() {
new KclStsAssumeRoleCredentialsProvider(new String[] {ARN, SESSION_NAME}); new KclStsAssumeRoleCredentialsProvider(new String[] {ARN, SESSION_NAME, "endpointRegion=us-east-1"});
} }
@Test @Test
public void testAcceptEndpoint() { public void testAcceptEndpoint() {
// discovered exception during e2e testing; therefore, this test is // discovered exception during e2e testing; therefore, this test is
// to simply verify the constructed STS client doesn't go *boom* // to simply verify the constructed STS client doesn't go *boom*
final KclStsAssumeRoleCredentialsProvider provider = new KclStsAssumeRoleCredentialsProvider(ARN, SESSION_NAME); final KclStsAssumeRoleCredentialsProvider provider =
new KclStsAssumeRoleCredentialsProvider(ARN, SESSION_NAME, "endpointRegion=us-east-1");
provider.acceptEndpoint("endpoint", "us-east-1"); provider.acceptEndpoint("endpoint", "us-east-1");
} }
@Test @Test
public void testVarArgs() { public void testVarArgs() {
for (final String[] varargs : Arrays.asList( for (final String[] varargs : Arrays.asList(
new String[] {ARN, SESSION_NAME, "externalId=eid", "foo"}, new String[] {ARN, SESSION_NAME, "externalId=eid", "foo", "endpointRegion=us-east-1"},
new String[] {ARN, SESSION_NAME, "foo", "externalId=eid"})) { new String[] {ARN, SESSION_NAME, "foo", "externalId=eid", "endpointRegion=us-east-1"})) {
final VarArgsSpy provider = new VarArgsSpy(varargs); final VarArgsSpy provider = new VarArgsSpy(varargs);
assertEquals("eid", provider.externalId); assertEquals("eid", provider.externalId);
} }

View file

@ -124,7 +124,8 @@ public class AwsCredentialsProviderPropertyValueDecoderTest {
KclStsAssumeRoleCredentialsProvider.class.getSimpleName(), // name-only; needs prefix KclStsAssumeRoleCredentialsProvider.class.getSimpleName(), // name-only; needs prefix
StsAssumeRoleCredentialsProvider.class.getName(), // user passes full sts package path StsAssumeRoleCredentialsProvider.class.getName(), // user passes full sts package path
StsAssumeRoleCredentialsProvider.class.getSimpleName())) { StsAssumeRoleCredentialsProvider.class.getSimpleName())) {
final AwsCredentialsProvider provider = decoder.decodeValue(className + "|arn|sessionName"); final AwsCredentialsProvider provider =
decoder.decodeValue(className + "|arn|sessionName|endpointRegion=us-east-1");
assertNotNull(className, provider); assertNotNull(className, provider);
} }
} }