@@ -79,16 +79,16 @@ public void getAmiId_queriesAndIncludesToken() {
79
79
WireMock .verify (getRequestedFor (urlPathEqualTo (AMI_ID_RESOURCE )).withHeader (TOKEN_HEADER , equalTo (TOKEN_STUB )));
80
80
}
81
81
82
- // @Test
83
- // public void getAmiId_tokenQueryTimeout_fallsBackToInsecure() {
84
- // stubFor(put(urlPathEqualTo(TOKEN_RESOURCE_PATH)).willReturn(aResponse().withFixedDelay(Integer.MAX_VALUE)));
85
- // stubFor(get(urlPathEqualTo(AMI_ID_RESOURCE)).willReturn(aResponse().withBody("{}")));
86
- //
87
- // EC2MetadataUtils.getAmiId();
88
- //
89
- // WireMock.verify(putRequestedFor(urlPathEqualTo(TOKEN_RESOURCE_PATH)).withHeader(EC2_METADATA_TOKEN_TTL_HEADER, equalTo("21600")));
90
- // WireMock.verify(getRequestedFor(urlPathEqualTo(AMI_ID_RESOURCE)).withoutHeader(TOKEN_HEADER));
91
- // }
82
+ @ Test
83
+ public void getAmiId_tokenQueryTimeout_fallsBackToInsecure () {
84
+ stubFor (put (urlPathEqualTo (TOKEN_RESOURCE_PATH )).willReturn (aResponse ().withFixedDelay (Integer .MAX_VALUE )));
85
+ stubFor (get (urlPathEqualTo (AMI_ID_RESOURCE )).willReturn (aResponse ().withBody ("{}" )));
86
+
87
+ EC2MetadataUtils .getAmiId ();
88
+
89
+ WireMock .verify (putRequestedFor (urlPathEqualTo (TOKEN_RESOURCE_PATH )).withHeader (EC2_METADATA_TOKEN_TTL_HEADER , equalTo ("21600" )));
90
+ WireMock .verify (getRequestedFor (urlPathEqualTo (AMI_ID_RESOURCE )).withoutHeader (TOKEN_HEADER ));
91
+ }
92
92
93
93
@ Test
94
94
public void getAmiId_queriesTokenResource_403Error_fallbackToInsecure () {
@@ -163,16 +163,16 @@ public void getAmiId_queriesTokenResource_400Error_throws() {
163
163
EC2MetadataUtils .getAmiId ();
164
164
}
165
165
166
- // @Test
167
- // public void fetchDataWithAttemptNumber_ioError_shouldHonor() {
168
- // int attempts = 1;
169
- // thrown.expect(SdkClientException.class);
170
- // thrown.expectMessage("Unable to contact EC2 metadata service");
171
- //
172
- // stubFor(put(urlPathEqualTo(TOKEN_RESOURCE_PATH)).willReturn(aResponse().withBody(TOKEN_STUB)));
173
- // stubFor(get(urlPathEqualTo(AMI_ID_RESOURCE)).willReturn(aResponse().withFault(Fault.CONNECTION_RESET_BY_PEER)));
174
- //
175
- // EC2MetadataUtils.fetchData(AMI_ID_RESOURCE, false, attempts);
176
- // WireMock.verify(attempts, getRequestedFor(urlPathEqualTo(AMI_ID_RESOURCE)));
177
- // }
166
+ @ Test
167
+ public void fetchDataWithAttemptNumber_ioError_shouldHonor () {
168
+ int attempts = 1 ;
169
+ thrown .expect (SdkClientException .class );
170
+ thrown .expectMessage ("Unable to contact EC2 metadata service" );
171
+
172
+ stubFor (put (urlPathEqualTo (TOKEN_RESOURCE_PATH )).willReturn (aResponse ().withBody (TOKEN_STUB )));
173
+ stubFor (get (urlPathEqualTo (AMI_ID_RESOURCE )).willReturn (aResponse ().withFault (Fault .CONNECTION_RESET_BY_PEER )));
174
+
175
+ EC2MetadataUtils .fetchData (AMI_ID_RESOURCE , false , attempts );
176
+ WireMock .verify (attempts , getRequestedFor (urlPathEqualTo (AMI_ID_RESOURCE )));
177
+ }
178
178
}
0 commit comments