@@ -260,9 +260,9 @@ void shouldRunLambdasOnSuccessful() {
260
260
261
261
setFeatureProvider (featureProvider1 , afterSet , afterInit , afterShutdown , afterError );
262
262
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 );
266
266
verify (afterError , never ()).accept (any (), any ());
267
267
}
268
268
@@ -278,7 +278,7 @@ void shouldRunLambdasOnError() throws Exception {
278
278
FeatureProvider errorFeatureProvider = createMockedErrorProvider ();
279
279
280
280
setFeatureProvider (errorFeatureProvider , afterSet , afterInit , afterShutdown , afterError );
281
- verify (afterSet ).accept (errorFeatureProvider );
281
+ verify (afterSet , timeout ( TIMEOUT ) ).accept (errorFeatureProvider );
282
282
verify (afterInit , never ()).accept (any ());;
283
283
verify (afterError , timeout (TIMEOUT )).accept (eq (errorFeatureProvider ), any ());
284
284
}
0 commit comments