Skip to content

Commit

Permalink
Fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
charlesbvll committed Aug 15, 2024
1 parent 33a9d08 commit 755d397
Showing 1 changed file with 8 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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 (
Expand All @@ -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),
Expand All @@ -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")
Expand Down

0 comments on commit 755d397

Please sign in to comment.