24
24
import java .util .Collection ;
25
25
import java .util .List ;
26
26
import java .util .Set ;
27
+ import java .util .concurrent .TimeUnit ;
27
28
import java .util .stream .Collectors ;
28
29
29
30
import net .shibboleth .utilities .java .support .xml .SerializeSupport ;
@@ -89,7 +90,7 @@ public void setup() throws Exception {
89
90
@ Test
90
91
public void withMetadataUrlLocationWhenResolvableThenFindByEntityIdReturns () throws Exception {
91
92
try (MockWebServer server = new MockWebServer ()) {
92
- server .setDispatcher (new AlwaysDispatch (new MockResponse (). setBody ( this .metadata ). setResponseCode ( 200 ) ));
93
+ server .setDispatcher (new AlwaysDispatch (this .metadata ));
93
94
AssertingPartyMetadataRepository parties = OpenSaml4AssertingPartyMetadataRepository
94
95
.withTrustedMetadataLocation (server .url ("/" ).toString ())
95
96
.build ();
@@ -106,8 +107,7 @@ public void withMetadataUrlLocationWhenResolvableThenFindByEntityIdReturns() thr
106
107
@ Test
107
108
public void withMetadataUrlLocationnWhenResolvableThenIteratorReturns () throws Exception {
108
109
try (MockWebServer server = new MockWebServer ()) {
109
- server .setDispatcher (
110
- new AlwaysDispatch (new MockResponse ().setBody (this .entitiesDescriptor ).setResponseCode (200 )));
110
+ server .setDispatcher (new AlwaysDispatch (this .entitiesDescriptor ));
111
111
List <AssertingPartyMetadata > parties = new ArrayList <>();
112
112
OpenSaml4AssertingPartyMetadataRepository .withTrustedMetadataLocation (server .url ("/" ).toString ())
113
113
.build ()
@@ -122,7 +122,9 @@ public void withMetadataUrlLocationnWhenResolvableThenIteratorReturns() throws E
122
122
@ Test
123
123
public void withMetadataUrlLocationWhenUnresolvableThenThrowsSaml2Exception () throws Exception {
124
124
try (MockWebServer server = new MockWebServer ()) {
125
- server .enqueue (new MockResponse ().setBody (this .metadata ).setResponseCode (200 ));
125
+ server .enqueue (new MockResponse ().setBody (this .metadata )
126
+ .setResponseCode (200 )
127
+ .setBodyDelay (1 , TimeUnit .MILLISECONDS ));
126
128
String url = server .url ("/" ).toString ();
127
129
server .shutdown ();
128
130
assertThatExceptionOfType (Saml2Exception .class )
@@ -360,7 +362,9 @@ private static final class AlwaysDispatch extends Dispatcher {
360
362
private final MockResponse response ;
361
363
362
364
private AlwaysDispatch (String body ) {
363
- this .response = new MockResponse ().setBody (body ).setResponseCode (200 );
365
+ this .response = new MockResponse ().setBody (body )
366
+ .setResponseCode (200 )
367
+ .setBodyDelay (1 , TimeUnit .MILLISECONDS );
364
368
}
365
369
366
370
private AlwaysDispatch (MockResponse response ) {
0 commit comments