chore: blocking set-provider in tests

Signed-off-by: Todd Baert <[email protected]>
open-feature · Oct 18, 2024 · 48cd5f4 · 48cd5f4
1 parent 3f1cfed
commit 48cd5f4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/test/java/dev/openfeature/sdk/OpenFeatureClientTest.java b/src/test/java/dev/openfeature/sdk/OpenFeatureClientTest.java
@@ -42,7 +42,7 @@ void reset_logs() {
     @DisplayName("should not throw exception if hook has different type argument than hookContext")
     void shouldNotThrowExceptionIfHookHasDifferentTypeArgumentThanHookContext() {
         OpenFeatureAPI api = OpenFeatureAPI.getInstance();
-        api.setProvider("shouldNotThrowExceptionIfHookHasDifferentTypeArgumentThanHookContext", new DoSomethingProvider());
+        api.setProviderAndWait("shouldNotThrowExceptionIfHookHasDifferentTypeArgumentThanHookContext", new DoSomethingProvider());
         Client client = api.getClient("shouldNotThrowExceptionIfHookHasDifferentTypeArgumentThanHookContext");
         client.addHooks(mockBooleanHook(), mockStringHook());
         FlagEvaluationDetails<Boolean> actual = client.getBooleanDetails("feature key", Boolean.FALSE);
@@ -96,7 +96,7 @@ void shouldNotCallEvaluationMethodsWhenProviderIsInFatalErrorState() {
     void shouldNotCallEvaluationMethodsWhenProviderIsInNotReadyState() {
         FeatureProvider provider = new TestEventsProvider(5000);
         OpenFeatureAPI api = OpenFeatureAPI.getInstance();
-        api.setProvider("shouldNotCallEvaluationMethodsWhenProviderIsInNotReadyState", provider);
+        api.setProviderAndWait("shouldNotCallEvaluationMethodsWhenProviderIsInNotReadyState", provider);
         Client client = api.getClient("shouldNotCallEvaluationMethodsWhenProviderIsInNotReadyState");
         FlagEvaluationDetails<Boolean> details = client.getBooleanDetails("key", true);