Skip to content

Commit 385d583

Browse files
committed
Fixing test failures
1 parent d23900d commit 385d583

File tree

1 file changed

+19
-3
lines changed

1 file changed

+19
-3
lines changed

core/aws-core/src/test/java/software/amazon/awssdk/awscore/internal/defaultsmode/AutoDefaultsModeDiscoveryEc2MetadataClientTest.java

Lines changed: 19 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
import static com.github.tomakehurst.wiremock.client.WireMock.putRequestedFor;
2424
import static com.github.tomakehurst.wiremock.client.WireMock.urlEqualTo;
2525
import static com.github.tomakehurst.wiremock.client.WireMock.verify;
26+
import static com.github.tomakehurst.wiremock.core.WireMockConfiguration.wireMockConfig;
2627
import static org.assertj.core.api.Assertions.assertThat;
2728

2829
import com.github.tomakehurst.wiremock.junit.WireMockRule;
@@ -47,12 +48,18 @@ public class AutoDefaultsModeDiscoveryEc2MetadataClientTest {
4748
private static final EnvironmentVariableHelper ENVIRONMENT_VARIABLE_HELPER = new EnvironmentVariableHelper();
4849

4950
@Rule
50-
public WireMockRule wireMock = new WireMockRule(0);
51+
public WireMockRule wireMock = new WireMockRule(wireMockConfig()
52+
.port(0)
53+
.httpsPort(-1));
5154

5255
@Before
5356
public void setup() {
5457
System.setProperty(SdkSystemSetting.AWS_EC2_METADATA_SERVICE_ENDPOINT.property(),
5558
"http://localhost:" + wireMock.port());
59+
60+
clearEnvironmentVariable("AWS_EXECUTION_ENV");
61+
clearEnvironmentVariable("AWS_REGION");
62+
clearEnvironmentVariable("AWS_DEFAULT_REGION");
5663
}
5764

5865
@After
@@ -62,6 +69,15 @@ public void cleanup() {
6269
System.clearProperty(SdkSystemSetting.AWS_EC2_METADATA_SERVICE_ENDPOINT.property());
6370
}
6471

72+
// Clear an environment variable by setting it to null.
73+
private void clearEnvironmentVariable(String name) {
74+
try {
75+
ENVIRONMENT_VARIABLE_HELPER.set(name, null);
76+
} catch (Exception e) {
77+
// Ignore
78+
}
79+
}
80+
6581
@Test
6682
public void autoDefaultsModeDiscovery_shouldUseSharedHttpClient() throws Exception {
6783
// Stub successful IMDS responses
@@ -132,8 +148,8 @@ public void awsEc2MetadataDisabled_shouldSkipImdsAndUseStandardMode() {
132148
assertThat(result).isEqualTo(DefaultsMode.STANDARD);
133149

134150
// Verify no IMDS requests were made
135-
wireMock.verify(0, putRequestedFor(urlEqualTo("/latest/api/token")));
136-
wireMock.verify(0, getRequestedFor(urlEqualTo("/latest/meta-data/placement/region")));
151+
verify(0, putRequestedFor(urlEqualTo("/latest/api/token")));
152+
verify(0, getRequestedFor(urlEqualTo("/latest/meta-data/placement/region")));
137153
}
138154

139155
@Test

0 commit comments

Comments
 (0)