class TestClient:
def setup(self):
self.env_config = get_env_config()
self.trainable_push_groups = self.env_config.trainable_push_groups
self.serve_config = get_serve_config()
self.action_space = ActionSpace(self.trainable_push_groups)
self.action_space.seed(2048)
self.mock_agent = mock_agent_with_action_space(self.action_space)
self.saved_model = SavedModel(self.mock_agent, Environment,
"/tmp/model_location", {})
def test_get_policy(self):
server = Server(self.serve_config)
policy_service = PolicyService(self.saved_model)
server.set_policy_service(policy_service)
try:
server.start()
time.sleep(0.5)
# create the client
address = "{}:{}".format(self.serve_config.host,
self.serve_config.port)
channel = grpc.insecure_channel(address)
client_stub = Client(channel)
policy = client_stub.get_policy(
url="https://www.example.com",
client_env=client.get_random_client_environment(),
manifest=self.env_config,
)
assert policy is not None
assert len(list(policy.push)) + len(list(policy.preload)) > 0
finally:
server.stop()
def get_action_space():
action_space = ActionSpace(get_push_groups())
action_space.seed(4096)
return action_space