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)
