diff --git a/src/py/flwr/server/superlink/fleet/grpc_rere/server_interceptor_test.py b/src/py/flwr/server/superlink/fleet/grpc_rere/server_interceptor_test.py index b87a4293a775..ece443a816cb 100644 --- a/src/py/flwr/server/superlink/fleet/grpc_rere/server_interceptor_test.py +++ b/src/py/flwr/server/superlink/fleet/grpc_rere/server_interceptor_test.py @@ -43,6 +43,7 @@ from flwr.proto.run_pb2 import GetRunRequest, GetRunResponse # pylint: disable=E0611 from flwr.proto.task_pb2 import Task, TaskRes # pylint: disable=E0611 from flwr.server.app import ADDRESS_FLEET_API_GRPC_RERE, _run_fleet_api_grpc_rere +from flwr.server.superlink.ffs.ffs_factory import FfsFactory from flwr.server.superlink.state.state_factory import StateFactory from .server_interceptor import ( @@ -62,6 +63,8 @@ def setUp(self) -> None: state_factory = StateFactory(":flwr-in-memory-state:") self.state = state_factory.state() + ffs_factory = FfsFactory(".") + self.ffs = ffs_factory.ffs() self.state.store_server_private_public_key( private_key_to_bytes(self._server_private_key), public_key_to_bytes(self._server_public_key), @@ -72,7 +75,11 @@ def setUp(self) -> None: self._server_interceptor = AuthenticateServerInterceptor(self.state) self._server: grpc.Server = _run_fleet_api_grpc_rere( - ADDRESS_FLEET_API_GRPC_RERE, state_factory, None, [self._server_interceptor] + ADDRESS_FLEET_API_GRPC_RERE, + state_factory, + ffs_factory, + None, + [self._server_interceptor], ) self._channel = grpc.insecure_channel("localhost:9092")