Update providers/openfeature-provider-flagd/src/openfeature/contrib/provider/flagd/resolvers/grpc.py

Co-authored-by: Todd Baert <[email protected]>
Signed-off-by: Simon Schrottner <[email protected]>

Author: aepfli

providers/openfeature-provider-flagd/src/openfeature/contrib/provider/flagd/resolvers/grpc.py

@@ -127,6 +127,7 @@ def listen(self) -> None:
                         return
             except grpc.RpcError as e:
                 logger.error(f"SyncFlags stream error, {e.code()=} {e.details()=}")
+                # re-create the stub if there's a connection issue - otherwise reconnect does not work as expected
                 self.stub, self.channel = self._create_stub()
             except ParseError:
                 logger.exception(

providers/openfeature-provider-flagd/tests/e2e/steps.py

@@ -28,10 +28,7 @@ def evaluation_context() -> EvaluationContext:
 @given("a provider is registered", target_fixture="client")
 def setup_provider(setup, resolver_type, client_name) -> OpenFeatureClient:
     api.set_provider(
-        FlagdProvider(
-            resolver_type=resolver_type,
-            port=setup,
-        ),
+        FlagdProvider(resolver_type=resolver_type, port=setup, timeout=1),
         client_name,
     )
     client = api.get_client(client_name)