fixup: another test race condition fix

Signed-off-by: Todd Baert <todd.baert@dynatrace.com>
open-feature · Jul 12, 2023 · be285dd · be285dd
1 parent f0d716e
commit be285dd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/test/java/dev/openfeature/sdk/ProviderRepositoryTest.java b/src/test/java/dev/openfeature/sdk/ProviderRepositoryTest.java
@@ -260,9 +260,9 @@ void shouldRunLambdasOnSuccessful() {
 
                 setFeatureProvider(featureProvider1, afterSet, afterInit, afterShutdown, afterError);
                 setFeatureProvider(featureProvider2);
-                verify(afterSet).accept(featureProvider1);
-                verify(afterInit).accept(featureProvider1);
-                verify(afterShutdown).accept(oldProvider);
+                verify(afterSet, timeout(TIMEOUT)).accept(featureProvider1);
+                verify(afterInit, timeout(TIMEOUT)).accept(featureProvider1);
+                verify(afterShutdown, timeout(TIMEOUT)).accept(oldProvider);
                 verify(afterError, never()).accept(any(), any());
             }
 
@@ -278,7 +278,7 @@ void shouldRunLambdasOnError() throws Exception {
                 FeatureProvider errorFeatureProvider = createMockedErrorProvider();
 
                 setFeatureProvider(errorFeatureProvider, afterSet, afterInit, afterShutdown, afterError);
-                verify(afterSet).accept(errorFeatureProvider);
+                verify(afterSet, timeout(TIMEOUT)).accept(errorFeatureProvider);
                 verify(afterInit, never()).accept(any());;
                 verify(afterError, timeout(TIMEOUT)).accept(eq(errorFeatureProvider), any());
             }