def get_mock_register_response(status=get_mock_status(), pplan=get_mock_pplan()): """Returns a mock protobuf RegisterInstanceResponse object from stmgr_pb2""" mock_response = stmgr_pb2.RegisterInstanceResponse() mock_response.status.MergeFrom(status) mock_response.pplan.MergeFrom(pplan) return mock_response
def _send_register_req(self): request = stmgr_pb2.RegisterInstanceRequest() request.instance.MergeFrom(self.instance) request.topology_name = self.topology_name request.topology_id = self.topology_id timeout_sec = float(self.sys_config[constants.INSTANCE_RECONNECT_STREAMMGR_INTERVAL_SEC]) self.send_request(request, "Context", stmgr_pb2.RegisterInstanceResponse(), timeout_sec)