def test_get_handler(self):
osutil.get_osutil()
protocol.get_protocol_util()
dhcp.get_dhcp_handler()
provision.get_provision_handler()
deprovision.get_deprovision_handler()
daemon.get_daemon_handler()
resourcedisk.get_resourcedisk_handler()
scvmm.get_scvmm_handler()
monitor.get_monitor_handler()
update.get_update_handler()
exthandlers.get_exthandlers_handler()
def test_provision_telemetry_fail(self,
mock_util,
distro_name,
distro_version,
distro_full_name):
"""
Assert that the agent issues one telemetry message as part of a
failed provisioning.
1. Provision
"""
ph = get_provision_handler(distro_name, distro_version,
distro_full_name)
ph.report_event = MagicMock()
ph.reg_ssh_host_key = MagicMock(side_effect=ProvisionError(
"--unit-test--"))
mock_osutil = MagicMock()
mock_osutil.decode_customdata = Mock(return_value="")
ph.osutil = mock_osutil
ph.protocol_util.osutil = mock_osutil
ph.protocol_util.get_protocol = MagicMock()
conf.get_dvd_mount_point = Mock(return_value=self.tmp_dir)
ovfenv_file = os.path.join(self.tmp_dir, OVF_FILE_NAME)
ovfenv_data = load_data("ovf-env.xml")
fileutil.write_file(ovfenv_file, ovfenv_data)
ph.run()
positional_args, kw_args = ph.report_event.call_args_list[0]
self.assertTrue(re.match(r'Provisioning failed: \[ProvisionError\] --unit-test-- \(\d+\.\d+s\)', positional_args[0]) is not None)
def provision(self):
"""
Run provision command
"""
from azurelinuxagent.pa.provision import get_provision_handler
provision_handler = get_provision_handler()
provision_handler.run()
def test_provision_telemetry_fail(self, mock_util, distro_name,
distro_version,
distro_full_name):
"""
Assert that the agent issues one telemetry message as part of a
failed provisioning.
1. Provision
"""
ph = get_provision_handler(distro_name, distro_version,
distro_full_name)
ph.report_event = MagicMock()
ph.reg_ssh_host_key = MagicMock(side_effect=ProvisionError(
"--unit-test--"))
mock_osutil = MagicMock()
mock_osutil.decode_customdata = Mock(return_value="")
ph.osutil = mock_osutil
ph.protocol_util.osutil = mock_osutil
ph.protocol_util.get_protocol_by_file = MagicMock()
ph.protocol_util.get_protocol = MagicMock()
conf.get_dvd_mount_point = Mock(return_value=self.tmp_dir)
ovfenv_file = os.path.join(self.tmp_dir, OVF_FILE_NAME)
ovfenv_data = load_data("ovf-env.xml")
fileutil.write_file(ovfenv_file, ovfenv_data)
ph.run()
ph.report_event.assert_called_once_with(
"[ProvisionError] --unit-test--")
def _provision_test(self,
distro_name,
distro_version,
distro_full_name,
ovf_file,
provisionMessage,
expect_success,
patch_write_agent_disabled,
patch_get_instance_id):
"""
Assert that the agent issues two telemetry messages as part of a
successful provisioning.
1. Provision
2. GuestState
"""
ph = get_provision_handler(distro_name,
distro_version,
distro_full_name)
ph.report_event = MagicMock()
ph.reg_ssh_host_key = MagicMock(return_value='--thumprint--')
mock_osutil = MagicMock()
mock_osutil.decode_customdata = Mock(return_value="")
ph.osutil = mock_osutil
ph.protocol_util.osutil = mock_osutil
ph.protocol_util.get_protocol = MagicMock()
conf.get_dvd_mount_point = Mock(return_value=self.tmp_dir)
ovfenv_file = os.path.join(self.tmp_dir, OVF_FILE_NAME)
ovfenv_data = load_data(ovf_file)
fileutil.write_file(ovfenv_file, ovfenv_data)
ph.run()
if expect_success:
self.assertEqual(2, ph.report_event.call_count)
positional_args, kw_args = ph.report_event.call_args_list[0]
# [call('Provisioning succeeded (146473.68s)', duration=65, is_success=True)]
self.assertTrue(re.match(r'Provisioning succeeded \(\d+\.\d+s\)', positional_args[0]) is not None)
self.assertTrue(isinstance(kw_args['duration'], int))
self.assertTrue(kw_args['is_success'])
positional_args, kw_args = ph.report_event.call_args_list[1]
self.assertTrue(kw_args['operation'] == 'ProvisionGuestAgent')
self.assertTrue(kw_args['message'] == provisionMessage)
self.assertTrue(kw_args['is_success'])
expected_disabled = True if provisionMessage == 'false' else False
self.assertTrue(patch_write_agent_disabled.call_count == expected_disabled)
else:
self.assertEqual(1, ph.report_event.call_count)
positional_args, kw_args = ph.report_event.call_args_list[0]
# [call(u'[ProtocolError] Failed to validate OVF: ProvisionGuestAgent not found')]
self.assertTrue('Failed to validate OVF: ProvisionGuestAgent not found' in positional_args[0])
self.assertFalse(kw_args['is_success'])
def daemon(self, child_args=None):
logger.info("Run daemon")
self.protocol_util = get_protocol_util()
self.scvmm_handler = get_scvmm_handler()
self.resourcedisk_handler = get_resourcedisk_handler()
self.rdma_handler = get_rdma_handler()
self.provision_handler = get_provision_handler()
self.update_handler = get_update_handler()
# Create lib dir
if not os.path.isdir(conf.get_lib_dir()):
fileutil.mkdir(conf.get_lib_dir(), mode=0o700)
os.chdir(conf.get_lib_dir())
if conf.get_detect_scvmm_env():
self.scvmm_handler.run()
if conf.get_resourcedisk_format():
self.resourcedisk_handler.run()
# Always redetermine the protocol start (e.g., wireserver vs.
# on-premise) since a VHD can move between environments
self.protocol_util.clear_protocol()
self.provision_handler.run()
# Enable RDMA, continue in errors
if conf.enable_rdma():
self.rdma_handler.install_driver()
logger.info("RDMA capabilities are enabled in configuration")
try:
# Ensure the most recent SharedConfig is available
# - Changes to RDMA state may not increment the goal state
# incarnation number. A forced update ensures the most
# current values.
protocol = self.protocol_util.get_protocol()
client = protocol.client
if client is None or type(client) is not WireClient:
raise Exception("Attempt to setup RDMA without Wireserver")
client.update_goal_state(forced=True)
setup_rdma_device()
except Exception as e:
logger.error("Error setting up rdma device: %s" % e)
else:
logger.info("RDMA capabilities are not enabled, skipping")
while self.running:
self.update_handler.run_latest(child_args=child_args)
def daemon(self, child_args=None):
logger.info("Run daemon")
self.protocol_util = get_protocol_util()
self.scvmm_handler = get_scvmm_handler()
self.resourcedisk_handler = get_resourcedisk_handler()
self.rdma_handler = get_rdma_handler()
self.provision_handler = get_provision_handler()
self.update_handler = get_update_handler()
# Create lib dir
if not os.path.isdir(conf.get_lib_dir()):
fileutil.mkdir(conf.get_lib_dir(), mode=0o700)
os.chdir(conf.get_lib_dir())
if conf.get_detect_scvmm_env():
self.scvmm_handler.run()
if conf.get_resourcedisk_format():
self.resourcedisk_handler.run()
# Always redetermine the protocol start (e.g., wireserver vs.
# on-premise) since a VHD can move between environments
self.protocol_util.clear_protocol()
self.provision_handler.run()
# Enable RDMA, continue in errors
if conf.enable_rdma():
self.rdma_handler.install_driver()
logger.info("RDMA capabilities are enabled in configuration")
try:
# Ensure the most recent SharedConfig is available
# - Changes to RDMA state may not increment the goal state
# incarnation number. A forced update ensures the most
# current values.
protocol = self.protocol_util.get_protocol()
client = protocol.client
if client is None or type(client) is not WireClient:
raise Exception("Attempt to setup RDMA without Wireserver")
client.update_goal_state(forced=True)
setup_rdma_device()
except Exception as e:
logger.error("Error setting up rdma device: %s" % e)
else:
logger.info("RDMA capabilities are not enabled, skipping")
while self.running:
self.update_handler.run_latest(child_args=child_args)
def test_provision(self, mock_util, distro_name, distro_version, distro_full_name):
provision_handler = get_provision_handler(distro_name, distro_version,
distro_full_name)
mock_osutil = MagicMock()
mock_osutil.decode_customdata = Mock(return_value="")
provision_handler.osutil = mock_osutil
provision_handler.protocol_util.osutil = mock_osutil
provision_handler.protocol_util.get_protocol = MagicMock()
conf.get_dvd_mount_point = Mock(return_value=self.tmp_dir)
ovfenv_file = os.path.join(self.tmp_dir, OVF_FILE_NAME)
ovfenv_data = load_data("ovf-env.xml")
fileutil.write_file(ovfenv_file, ovfenv_data)
provision_handler.run()
def test_provision(self, mock_util, distro_name, distro_version, distro_full_name):
provision_handler = get_provision_handler(distro_name, distro_version,
distro_full_name)
mock_osutil = MagicMock()
mock_osutil.decode_customdata = Mock(return_value="")
provision_handler.osutil = mock_osutil
provision_handler.protocol_util.osutil = mock_osutil
provision_handler.protocol_util.get_protocol = MagicMock()
conf.get_dvd_mount_point = Mock(return_value=self.tmp_dir)
ovfenv_file = os.path.join(self.tmp_dir, OVF_FILE_NAME)
ovfenv_data = load_data("ovf-env.xml")
fileutil.write_file(ovfenv_file, ovfenv_data)
provision_handler.run()
def daemon(self):
logger.info("Run daemon")
self.protocol_util = get_protocol_util()
self.scvmm_handler = get_scvmm_handler()
self.resourcedisk_handler = get_resourcedisk_handler()
self.rdma_handler = get_rdma_handler()
self.provision_handler = get_provision_handler()
self.update_handler = get_update_handler()
# Create lib dir
if not os.path.isdir(conf.get_lib_dir()):
fileutil.mkdir(conf.get_lib_dir(), mode=0o700)
os.chdir(conf.get_lib_dir())
if conf.get_detect_scvmm_env():
self.scvmm_handler.run()
if conf.get_resourcedisk_format():
self.resourcedisk_handler.run()
# Always redetermine the protocol start (e.g., wireserver vs.
# on-premise) since a VHD can move between environments
self.protocol_util.clear_protocol()
self.provision_handler.run()
# Enable RDMA, continue in errors
if conf.enable_rdma():
self.rdma_handler.install_driver()
logger.info("RDMA capabilities are enabled in configuration")
try:
setup_rdma_device()
except Exception as e:
logger.error("Error setting up rdma device: %s" % e)
else:
logger.info("RDMA capabilities are not enabled, skipping")
while self.running:
self.update_handler.run_latest()
def daemon(self):
logger.info("Run daemon")
self.protocol_util = get_protocol_util()
self.scvmm_handler = get_scvmm_handler()
self.resourcedisk_handler = get_resourcedisk_handler()
self.rdma_handler = get_rdma_handler()
self.provision_handler = get_provision_handler()
self.update_handler = get_update_handler()
# Create lib dir
if not os.path.isdir(conf.get_lib_dir()):
fileutil.mkdir(conf.get_lib_dir(), mode=0o700)
os.chdir(conf.get_lib_dir())
if conf.get_detect_scvmm_env():
self.scvmm_handler.run()
if conf.get_resourcedisk_format():
self.resourcedisk_handler.run()
# Always redetermine the protocol start (e.g., wireserver vs.
# on-premise) since a VHD can move between environments
self.protocol_util.clear_protocol()
self.provision_handler.run()
# Enable RDMA, continue in errors
if conf.enable_rdma():
self.rdma_handler.install_driver()
logger.info("RDMA capabilities are enabled in configuration")
try:
setup_rdma_device()
except Exception as e:
logger.error("Error setting up rdma device: %s" % e)
else:
logger.info("RDMA capabilities are not enabled, skipping")
while self.running:
self.update_handler.run_latest()
def test_provision_telemetry_success(self, mock_util, distro_name,
distro_version, distro_full_name):
"""
Assert that the agent issues two telemetry messages as part of a
successful provisioning.
1. Provision
2. GuestState
"""
ph = get_provision_handler(distro_name, distro_version,
distro_full_name)
ph.report_event = MagicMock()
ph.reg_ssh_host_key = MagicMock(return_value='--thumprint--')
mock_osutil = MagicMock()
mock_osutil.decode_customdata = Mock(return_value="")
ph.osutil = mock_osutil
ph.protocol_util.osutil = mock_osutil
ph.protocol_util.get_protocol = MagicMock()
conf.get_dvd_mount_point = Mock(return_value=self.tmp_dir)
ovfenv_file = os.path.join(self.tmp_dir, OVF_FILE_NAME)
ovfenv_data = load_data("ovf-env.xml")
fileutil.write_file(ovfenv_file, ovfenv_data)
ph.run()
self.assertEqual(2, ph.report_event.call_count)
positional_args, kw_args = ph.report_event.call_args_list[0]
# [call('Provisioning succeeded (146473.68s)', duration=65, is_success=True)]
self.assertTrue(
re.match(r'Provisioning succeeded \(\d+\.\d+s\)',
positional_args[0]) is not None)
self.assertTrue(isinstance(kw_args['duration'], int))
self.assertTrue(kw_args['is_success'])
positional_args, kw_args = ph.report_event.call_args_list[1]
self.assertTrue(kw_args['operation'] == 'ProvisionGuestAgent')
self.assertTrue(kw_args['message'] == 'false')
self.assertTrue(kw_args['is_success'])
def test_provision_telemetry_success(self, mock_util, distro_name, distro_version,
distro_full_name):
"""
Assert that the agent issues two telemetry messages as part of a
successful provisioning.
1. Provision
2. GuestState
"""
ph = get_provision_handler(distro_name, distro_version,
distro_full_name)
ph.report_event = MagicMock()
ph.reg_ssh_host_key = MagicMock(return_value='--thumprint--')
mock_osutil = MagicMock()
mock_osutil.decode_customdata = Mock(return_value="")
ph.osutil = mock_osutil
ph.protocol_util.osutil = mock_osutil
ph.protocol_util.get_protocol_by_file = MagicMock()
ph.protocol_util.get_protocol = MagicMock()
conf.get_dvd_mount_point = Mock(return_value=self.tmp_dir)
ovfenv_file = os.path.join(self.tmp_dir, OVF_FILE_NAME)
ovfenv_data = load_data("ovf-env.xml")
fileutil.write_file(ovfenv_file, ovfenv_data)
ph.run()
self.assertEqual(2, ph.report_event.call_count)
positional_args, kw_args = ph.report_event.call_args_list[0]
# [call('Provisioning succeeded (146473.68s)', duration=65, is_success=True)]
self.assertTrue(re.match(r'Provisioning succeeded \(\d+\.\d+s\)', positional_args[0]) is not None)
self.assertTrue(isinstance(kw_args['duration'], int))
self.assertTrue(kw_args['is_success'])
positional_args, kw_args = ph.report_event.call_args_list[1]
self.assertTrue(kw_args['operation'] == 'ProvisionGuestAgent')
self.assertTrue(kw_args['message'] == 'false')
self.assertTrue(kw_args['is_success'])
def test_provision_telemetry_fail(
self,
mock_util, # pylint: disable=unused-argument
distro_name,
distro_version,
distro_full_name,
_):
"""
Assert that the agent issues one telemetry message as part of a
failed provisioning.
1. Provision
"""
ph = get_provision_handler(
distro_name,
distro_version, # pylint: disable=invalid-name
distro_full_name)
ph.report_event = MagicMock()
ph.reg_ssh_host_key = MagicMock(
side_effect=ProvisionError("--unit-test--"))
mock_osutil = MagicMock()
mock_osutil.decode_customdata = Mock(return_value="")
ph.osutil = mock_osutil
ph.protocol_util.osutil = mock_osutil
ph.protocol_util.get_protocol = MagicMock()
conf.get_dvd_mount_point = Mock(return_value=self.tmp_dir)
ovfenv_file = os.path.join(self.tmp_dir, OVF_FILE_NAME)
ovfenv_data = load_data("ovf-env.xml")
fileutil.write_file(ovfenv_file, ovfenv_data)
ph.run()
positional_args, kw_args = ph.report_event.call_args_list[0] # pylint: disable=unused-variable
self.assertTrue(
re.match(
r'Provisioning failed: \[ProvisionError\] --unit-test-- \(\d+\.\d+s\)',
positional_args[0]) is not None)
def test_provision_telemetry_success(self, mock_util, distro_name, distro_version,
distro_full_name):
"""
Assert that the agent issues two telemetry messages as part of a
successful provisioning.
1. Provision
2. GuestState
"""
ph = get_provision_handler(distro_name, distro_version,
distro_full_name)
ph.report_event = MagicMock()
ph.reg_ssh_host_key = MagicMock(return_value='--thumprint--')
mock_osutil = MagicMock()
mock_osutil.decode_customdata = Mock(return_value="")
ph.osutil = mock_osutil
ph.protocol_util.osutil = mock_osutil
ph.protocol_util.get_protocol_by_file = MagicMock()
ph.protocol_util.get_protocol = MagicMock()
conf.get_dvd_mount_point = Mock(return_value=self.tmp_dir)
ovfenv_file = os.path.join(self.tmp_dir, OVF_FILE_NAME)
ovfenv_data = load_data("ovf-env.xml")
fileutil.write_file(ovfenv_file, ovfenv_data)
ph.run()
call1 = call("Provisioning succeeded", duration=ANY, is_success=True)
call2 = call(ANY, is_success=True, operation=WALAEventOperation.GuestState)
ph.report_event.assert_has_calls([call1, call2])
args, kwargs = ph.report_event.call_args_list[1]
guest_state_json = json.loads(args[0])
self.assertTrue(1 <= guest_state_json['cpu'])
self.assertTrue(1 <= guest_state_json['mem'])
self.assertEqual(socket.gethostname(), guest_state_json['hostname'])
def test_handle_provision_guest_agent(self,
patch_write_agent_disabled,
distro_name,
distro_version,
distro_full_name):
ph = get_provision_handler(distro_name,
distro_version,
distro_full_name)
patch_write_agent_disabled.call_count = 0
ph.handle_provision_guest_agent(provision_guest_agent='false')
self.assertEqual(1, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent='False')
self.assertEqual(2, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent='FALSE')
self.assertEqual(3, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent='')
self.assertEqual(3, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent=' ')
self.assertEqual(3, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent=None)
self.assertEqual(3, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent='true')
self.assertEqual(3, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent='True')
self.assertEqual(3, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent='TRUE')
self.assertEqual(3, patch_write_agent_disabled.call_count)
def test_handle_provision_guest_agent(self,
patch_write_agent_disabled,
distro_name,
distro_version,
distro_full_name):
ph = get_provision_handler(distro_name,
distro_version,
distro_full_name)
patch_write_agent_disabled.call_count = 0
ph.handle_provision_guest_agent(provision_guest_agent='false')
self.assertEqual(1, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent='False')
self.assertEqual(2, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent='FALSE')
self.assertEqual(3, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent='')
self.assertEqual(3, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent=' ')
self.assertEqual(3, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent=None)
self.assertEqual(3, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent='true')
self.assertEqual(3, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent='True')
self.assertEqual(3, patch_write_agent_disabled.call_count)
ph.handle_provision_guest_agent(provision_guest_agent='TRUE')
self.assertEqual(3, patch_write_agent_disabled.call_count)
def _provision_test(
self, # pylint: disable=invalid-name,too-many-arguments
distro_name,
distro_version,
distro_full_name,
ovf_file,
provisionMessage,
expect_success,
patch_write_agent_disabled,
patch_get_instance_id): # pylint: disable=unused-argument
"""
Assert that the agent issues two telemetry messages as part of a
successful provisioning.
1. Provision
2. GuestState
"""
ph = get_provision_handler(
distro_name, # pylint: disable=invalid-name
distro_version,
distro_full_name)
ph.report_event = MagicMock()
ph.reg_ssh_host_key = MagicMock(return_value='--thumprint--')
mock_osutil = MagicMock()
mock_osutil.decode_customdata = Mock(return_value="")
ph.osutil = mock_osutil
ph.protocol_util.osutil = mock_osutil
ph.protocol_util.get_protocol = MagicMock()
conf.get_dvd_mount_point = Mock(return_value=self.tmp_dir)
ovfenv_file = os.path.join(self.tmp_dir, OVF_FILE_NAME)
ovfenv_data = load_data(ovf_file)
fileutil.write_file(ovfenv_file, ovfenv_data)
ph.run()
if expect_success:
self.assertEqual(2, ph.report_event.call_count)
positional_args, kw_args = ph.report_event.call_args_list[0]
# [call('Provisioning succeeded (146473.68s)', duration=65, is_success=True)]
self.assertTrue(
re.match(r'Provisioning succeeded \(\d+\.\d+s\)',
positional_args[0]) is not None)
self.assertTrue(isinstance(kw_args['duration'], int))
self.assertTrue(kw_args['is_success'])
positional_args, kw_args = ph.report_event.call_args_list[1]
self.assertTrue(kw_args['operation'] == 'ProvisionGuestAgent')
self.assertTrue(kw_args['message'] == provisionMessage)
self.assertTrue(kw_args['is_success'])
expected_disabled = True if provisionMessage == 'false' else False # pylint: disable=simplifiable-if-expression
self.assertTrue(
patch_write_agent_disabled.call_count == expected_disabled)
else:
self.assertEqual(1, ph.report_event.call_count)
positional_args, kw_args = ph.report_event.call_args_list[0]
# [call(u'[ProtocolError] Failed to validate OVF: ProvisionGuestAgent not found')]
self.assertTrue(
'Failed to validate OVF: ProvisionGuestAgent not found' in
positional_args[0])
self.assertFalse(kw_args['is_success'])
def test_get_provision_handler_config_cloudinit(
self,
patch_get_provisioning_agent):
provisioning_handler = get_provision_handler()
self.assertIsInstance(provisioning_handler, CloudInitProvisionHandler, 'Provisioning handler should be cloud-init if agent is set to cloud-init')
def test_get_provision_handler_config_auto_cloudinit(
self,
patch_cloud_init_is_enabled,
patch_get_provisioning_agent):
provisioning_handler = get_provision_handler()
self.assertIsInstance(provisioning_handler, CloudInitProvisionHandler, 'Auto provisioning handler should be cloud-init if cloud-init is enabled')
def test_get_provision_handler_config_waagent(
self,
patch_cloud_init_is_enabled,
patch_get_provisioning_agent):
provisioning_handler = get_provision_handler()
self.assertIsInstance(provisioning_handler, ProvisionHandler, 'Provisioning handler should be waagent if agent is set to waagent')
def daemon(self, child_args=None):
logger.info("Run daemon")
self.protocol_util = get_protocol_util() # pylint: disable=W0201
self.scvmm_handler = get_scvmm_handler() # pylint: disable=W0201
self.resourcedisk_handler = get_resourcedisk_handler() # pylint: disable=W0201
self.rdma_handler = get_rdma_handler() # pylint: disable=W0201
self.provision_handler = get_provision_handler() # pylint: disable=W0201
self.update_handler = get_update_handler() # pylint: disable=W0201
if conf.get_detect_scvmm_env():
self.scvmm_handler.run()
if conf.get_resourcedisk_format():
self.resourcedisk_handler.run()
# Always redetermine the protocol start (e.g., wireserver vs.
# on-premise) since a VHD can move between environments
self.protocol_util.clear_protocol()
self.provision_handler.run()
# Once we have the protocol, complete initialization of the telemetry fields
# that require the goal state and IMDS
self._initialize_telemetry()
# Enable RDMA, continue in errors
if conf.enable_rdma():
nd_version = self.rdma_handler.get_rdma_version()
self.rdma_handler.install_driver_if_needed()
logger.info("RDMA capabilities are enabled in configuration")
try:
# Ensure the most recent SharedConfig is available
# - Changes to RDMA state may not increment the goal state
# incarnation number. A forced update ensures the most
# current values.
protocol = self.protocol_util.get_protocol()
if type(protocol) is not WireProtocol:
raise Exception("Attempt to setup RDMA without Wireserver")
protocol.client.update_goal_state(forced=True)
setup_rdma_device(nd_version,
protocol.client.get_shared_conf())
except Exception as e:
logger.error("Error setting up rdma device: %s" % e)
else:
logger.info("RDMA capabilities are not enabled, skipping")
self.sleep_if_disabled()
# Disable output to /dev/console once provisioning has completed
if logger.console_output_enabled():
logger.info(
"End of log to /dev/console. The agent will now check for updates and then will process extensions."
)
logger.disable_console_output()
while self.running:
self.update_handler.run_latest(child_args=child_args)
