@@ -81,7 +81,7 @@ void shouldImmediatelyReturnWhenCallingTheProviderMutator() throws Exception {
8181 return true ;
8282 });
8383
84- verify (featureProvider ).initialize (any ());
84+ verify (featureProvider , timeout ( TIMEOUT ) ).initialize (any ());
8585 }
8686
8787 @ Test
@@ -166,7 +166,7 @@ void shouldImmediatelyReturnWhenCallingTheProviderMutator() throws Exception {
166166 return true ;
167167 });
168168
169- verify (newProvider ).initialize (any ());
169+ verify (newProvider , timeout ( TIMEOUT ) ).initialize (any ());
170170 }
171171
172172 @ Test
@@ -239,7 +239,7 @@ void shouldNotThrowExceptionIfProviderThrowsOneOnShutdown() {
239239
240240 assertThatCode (() -> setFeatureProvider (new NoOpProvider ())).doesNotThrowAnyException ();
241241
242- verify (provider ).shutdown ();
242+ verify (provider , timeout ( TIMEOUT ) ).shutdown ();
243243 }
244244 }
245245
@@ -260,9 +260,9 @@ void shouldRunLambdasOnSuccessful() {
260260
261261 setFeatureProvider (featureProvider1 , afterSet , afterInit , afterShutdown , afterError );
262262 setFeatureProvider (featureProvider2 );
263- verify (afterSet ).accept (featureProvider1 );
264- verify (afterInit ).accept (featureProvider1 );
265- verify (afterShutdown ).accept (oldProvider );
263+ verify (afterSet , timeout ( TIMEOUT ) ).accept (featureProvider1 );
264+ verify (afterInit , timeout ( TIMEOUT ) ).accept (featureProvider1 );
265+ verify (afterShutdown , timeout ( TIMEOUT ) ).accept (oldProvider );
266266 verify (afterError , never ()).accept (any (), any ());
267267 }
268268
@@ -278,7 +278,7 @@ void shouldRunLambdasOnError() throws Exception {
278278 FeatureProvider errorFeatureProvider = createMockedErrorProvider ();
279279
280280 setFeatureProvider (errorFeatureProvider , afterSet , afterInit , afterShutdown , afterError );
281- verify (afterSet ).accept (errorFeatureProvider );
281+ verify (afterSet , timeout ( TIMEOUT ) ).accept (errorFeatureProvider );
282282 verify (afterInit , never ()).accept (any ());;
283283 verify (afterError , timeout (TIMEOUT )).accept (eq (errorFeatureProvider ), any ());
284284 }
0 commit comments