def test_apply_mof(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration('mof/dscnode.nxFile.meta.mof')
self.assertTrue('ReturnValue=0' in config)
def test_register(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
exit_code, output = dsc.register_automation('somekey','http://dummy','','','','')
self.assertEqual(exit_code, 0)
def test_register_without_registration_info(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
exit_code, output = dsc.register_automation('','','','','','')
self.assertEqual(exit_code, 51)
def test_is_statusfile_update_idempotent(self):
status_file = 'status/0.status'
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.update_statusfile(status_file, '123','345')
dsc.update_statusfile(status_file, '123','345')
self.verify_nodeid_vmuuid(status_file)
def test_is_statusfile_update_idempotent(self):
status_file = 'status/0.status'
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.update_statusfile(status_file, '123', '345')
dsc.update_statusfile(status_file, '123', '345')
self.verify_nodeid_vmuuid(status_file)
def test_nodeid_with_dsc(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration('mof/dscnode.nxFile.meta.push.mof')
nodeid = dsc.get_nodeid('/etc/opt/omi/conf/omsconfig/agentid')
self.assertTrue(nodeid is not None, "nodeid is none")
def test_register(self):
print "Register test case with valid parameters"
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
exit_code, output = dsc.register_automation('somekey','http://dummy','test.localhost.mof','15','30','applyandmonitor')
self.assertEqual(exit_code, 0)
def test_is_statusfile_update_register(self):
status_file = 'status/0.status'
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
exit_code, output = dsc.register_automation('somekey','http://dummy','','','','')
self.verify_nodeid_vmuuid(status_file)
def test_apply_mof(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = Dummy()
dsc.hutil.log = waagent.Log
dsc.install_dsc_packages()
dsc.start_omiservice()
dsc.apply_dsc_configuration('mof/localhost.nxFile.mof')
self.assertTrue(os.path.exists('/tmp/dsctest'))
def test_is_statusfile_update_register(self):
status_file = 'status/0.status'
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
exit_code, output = dsc.register_automation('somekey','http://dummy','','','','')
self.verify_nodeid_vmuuid(status_file)
def test_register_invalid_configuration_mode(self):
print "Register test case with invalid configuration mode"
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
exit_code, output = dsc.register_automation('somekey','http://dummy','','','','some')
self.assertEqual(exit_code, 51)
def test_apply_mof(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration('mof/dscnode.nxFile.meta.push.mof')
dsc.apply_dsc_configuration('mof/localhost.nxFile.mof')
self.assertTrue(os.path.exists('/tmp/dsctest'))
def test_register_without_registration_info(self):
print "Register test case with invalid Registration url and Registration key"
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
exit_code, output = dsc.register_automation('','','','','','')
self.assertEqual(exit_code, 51)
def test_nodeid_with_dsc(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration('mof/dscnode.nxFile.meta.push.mof')
nodeid = dsc.get_nodeid('/etc/opt/omi/conf/omsconfig/agentid')
self.assertTrue(nodeid is not None, "nodeid is none")
def test_apply_mof(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = Dummy()
dsc.hutil.log = waagent.Log
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration('mof/dscnode.nxFile.meta.mof')
self.assertTrue('ReturnValue=0' in config)
def test_apply_mof(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration(
'mof/dscnode.nxFile.meta.push.mof')
dsc.apply_dsc_configuration('mof/localhost.nxFile.mof')
self.assertTrue(os.path.exists('/tmp/dsctest'))
def test_statusfile_update(self):
status_file = 'status/0.status'
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
class cresponse: deserialized_data= { "AccountName" : "test", "ResourceGroupName" : "rgName", "SubscriptionId" : "testsubid" }
dsc.update_statusfile(status_file, '123','345', cresponse)
self.verify_nodeid_vmuuid(status_file)
def test_is_statusfile_update_pull(self):
status_file = 'status/0.status'
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration('mof/dscnode.nxFile.meta.mof')
self.assertTrue('ReturnValue=0' in config)
self.verify_nodeid_vmuuid(status_file)
def test_is_statusfile_update_pull(self):
status_file = 'status/0.status'
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration('mof/dscnode.nxFile.meta.mof')
self.assertTrue('ReturnValue=0' in config)
self.verify_nodeid_vmuuid(status_file)
def test_send_request_to_pullserver(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration('mof/azureautomation.df.meta.mof')
self.assertTrue('ReturnValue=0' in config)
response = dsc.send_heart_beat_msg_to_agent_service("install")
self.assertEqual(response.status_code, 200)
def test_push_request_properties(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration('mof/dscnode.nxFile.meta.push.mof')
self.assertTrue('ReturnValue=0' in config)
response = dsc.send_heart_beat_msg_to_agent_service("install")
self.assertIsNone(response)
def test_is_statusfile_update_push(self):
status_file = 'status/0.status'
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration('mof/dscnode.nxFile.meta.push.mof')
dsc.apply_dsc_configuration('mof/localhost.nxFile.mof')
self.assertTrue(os.path.exists('/tmp/dsctest'))
self.verify_nodeid_vmuuid(status_file)
def test_is_statusfile_update_push(self):
status_file = 'status/0.status'
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration('mof/dscnode.nxFile.meta.push.mof')
dsc.apply_dsc_configuration('mof/localhost.nxFile.mof')
self.assertTrue(os.path.exists('/tmp/dsctest'))
self.verify_nodeid_vmuuid(status_file)
def test_push_request_properties(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration(
'mof/dscnode.nxFile.meta.push.mof')
self.assertTrue('ReturnValue=0' in config)
response = dsc.send_heart_beat_msg_to_agent_service()
self.assertIsNone(response)
def test_send_request_to_pullserver(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration(
'mof/azureautomation.df.meta.mof')
self.assertTrue('ReturnValue=0' in config)
response = dsc.send_heart_beat_msg_to_agent_service()
self.assertEqual(response.status_code, 200)
def test_properties_for_pull(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration('mof/dscnode.nxFile.meta.mof')
self.assertTrue('ReturnValue=0' in config)
content = dsc.construct_node_extension_properties(config, "upgrade")
data = json.dumps(content)
self.assertTrue('OMSCloudId' in data, "OMSCLoudID doesn't exist")
def test_properties_for_pull(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
dsc.install_dsc_packages()
dsc.start_omiservice()
config = dsc.apply_dsc_meta_configuration(
'mof/dscnode.nxFile.meta.mof')
self.assertTrue('ReturnValue=0' in config)
content = dsc.construct_node_extension_properties(config)
data = json.dumps(content)
self.assertTrue('OMSCloudId' in data, "OMSCLoudID doesn't exist")
def test_nodeid_without_dsc(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
nodeid = dsc.get_nodeid('/etc/opt/omi/conf/omsconfig/agentid1')
self.assertTrue(nodeid is None, "nodeid is not none")
def test_nodeid_without_dsc(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
nodeid = dsc.get_nodeid('/etc/opt/omi/conf/omsconfig/agentid1')
self.assertTrue(nodeid is None, "nodeid is not none")
def test_with_lower_version(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = Dummy()
dsc.hutil.log = waagent.Log
output = dsc.compare_pkg_version('1.0.4.35', 1, 1, 1, 294)
self.assertEqual(0, output)
def test_update_node_properties(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
response = dsc.update()
self.assertIsNone(response)
def test_with_higher_version(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = Dummy()
dsc.hutil.log = waagent.Log
output = dsc.compare_pkg_version('1.2.0.35', 1, 1, 1, 294)
self.assertEqual(1, output)
def test_update_node_properties(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = MockUtil(self)
response = dsc.update()
self.assertIsNone(response)
def test_with_lower_version(self):
dsc.distro_category = dsc.get_distro_category()
dsc.hutil = Dummy()
dsc.hutil.log = waagent.Log
output = dsc.compare_pkg_version('1.0.4', 1, 0, 8)
self.assertEqual(0, output)
