def make_internal_kv_calls(): response = task_servicer.KVPut( ray_client_pb2.KVPutRequest(key=b"key", value=b"val") ) assert isinstance(response, ray_client_pb2.KVPutResponse) assert not response.already_exists response = task_servicer.KVPut( ray_client_pb2.KVPutRequest(key=b"key", value=b"val2") ) assert isinstance(response, ray_client_pb2.KVPutResponse) assert response.already_exists response = task_servicer.KVGet(ray_client_pb2.KVGetRequest(key=b"key")) assert isinstance(response, ray_client_pb2.KVGetResponse) assert response.value == b"val" response = task_servicer.KVPut( ray_client_pb2.KVPutRequest(key=b"key", value=b"val2", overwrite=True) ) assert isinstance(response, ray_client_pb2.KVPutResponse) assert response.already_exists response = task_servicer.KVGet(ray_client_pb2.KVGetRequest(key=b"key")) assert isinstance(response, ray_client_pb2.KVGetResponse) assert response.value == b"val2"
def internal_kv_put(self, key: bytes, value: bytes, overwrite: bool) -> bool: req = ray_client_pb2.KVPutRequest(key=key, value=value, overwrite=overwrite) resp = self.server.KVPut(req, metadata=self.metadata) return resp.already_exists
def internal_kv_put(self, key: bytes, value: bytes, overwrite: bool) -> bool: req = ray_client_pb2.KVPutRequest( key=key, value=value, overwrite=overwrite) metadata = self._add_ids_to_metadata(self.metadata) resp = self._call_stub("KVPut", req, metadata=metadata) return resp.already_exists