def setUp(self):
db_adapter_url = 'hackathon.azureformation.service.db_adapter'
with mock.patch(db_adapter_url) as db_adapter:
subscription_id = 'hter45sd-gf34-gfd4-ht34-ds43gree23gr'
pem_url = '/home/azureformation.pem'
management_host = 'das342.fdsf34.com'
db_adapter.get_object.return_value = AzureKey(subscription_id=subscription_id,
pem_url=pem_url,
management_host=management_host)
azure_key_id = 0
self.service = AzureAdapter(azure_key_id)
def __get_available_public_ports(self, expr_id, host_server, host_ports):
self.log.debug("starting to get azure ports")
ep = EndpointService(AzureAdapter(self.load_azure_key_id(expr_id)))
host_server_name = host_server.vm_name
host_server_dns = host_server.public_dns.split('.')[0]
public_endpoints = ep.assign_public_endpoints(host_server_dns, 'Production', host_server_name, host_ports)
if not isinstance(public_endpoints, list):
self.log.debug("failed to get public ports")
return internal_server_error('cannot get public ports')
self.log.debug("public ports : %s" % public_endpoints)
return public_endpoints
def setUp(self):
db_adapter_url = "hackathon.azureformation.service.db_adapter"
with mock.patch(db_adapter_url) as db_adapter:
subscription_id = "hter45sd-gf34-gfd4-ht34-ds43gree23gr"
pem_url = "/home/azureformation.pem"
management_host = "das342.fdsf34.com"
db_adapter.get_object.return_value = AzureKey(
subscription_id=subscription_id, pem_url=pem_url, management_host=management_host
)
azure_key_id = 0
self.service = AzureAdapter(azure_key_id)
def __release_public_ports(self, expr_id, host_server, host_ports):
ep = EndpointService(AzureAdapter(self.load_azure_key_id(expr_id)))
host_server_name = host_server.vm_name
host_server_dns = host_server.public_dns.split('.')[0]
self.log.debug("starting to release ports ... ")
ep.release_public_endpoints(host_server_dns, 'Production', host_server_name, host_ports)
class ServiceTest(unittest.TestCase):
def setUp(self):
db_adapter_url = "hackathon.azureformation.service.db_adapter"
with mock.patch(db_adapter_url) as db_adapter:
subscription_id = "hter45sd-gf34-gfd4-ht34-ds43gree23gr"
pem_url = "/home/azureformation.pem"
management_host = "das342.fdsf34.com"
db_adapter.get_object.return_value = AzureKey(
subscription_id=subscription_id, pem_url=pem_url, management_host=management_host
)
azure_key_id = 0
self.service = AzureAdapter(azure_key_id)
def tearDown(self):
pass
def test_storage_account_exists(self):
name = "djfh434"
self.service.get_storage_account_properties = Mock()
self.service.get_storage_account_properties.return_value = None
self.assertFalse(self.service.storage_account_exists(name))
self.service.get_storage_account_properties.return_value = not None
self.assertTrue(self.service.storage_account_exists(name))
self.service.get_storage_account_properties.side_effect = Exception
self.assertFalse(self.service.storage_account_exists(name))
def test_cloud_service_exists(self):
name = "sdhsj6598"
self.service.get_hosted_service_properties = Mock()
self.service.get_hosted_service_properties.return_value = None
self.assertFalse(self.service.cloud_service_exists(name))
self.service.get_hosted_service_properties.return_value = not None
self.assertTrue(self.service.cloud_service_exists(name))
self.service.get_hosted_service_properties.side_effect = Exception
self.assertFalse(self.service.cloud_service_exists(name))
def test_deployment_exists(self):
name = "fhjd545"
slot = "df8dfkj"
self.service.get_deployment_by_slot = Mock()
self.service.get_deployment_by_slot.return_value = None
self.assertFalse(self.service.deployment_exists(name, slot))
self.service.get_deployment_by_slot.return_value = not None
self.assertTrue(self.service.deployment_exists(name, slot))
self.service.get_deployment_by_slot.side_effect = Exception
self.assertFalse(self.service.deployment_exists(name, slot))
def test_wait_for_deployment(self):
cs_name = "dhsj23"
dm_name = "fdu43f"
sec = 1
loop = 3
self.service.get_deployment_by_name = Mock()
d = Deployment()
self.service.get_deployment_by_name.return_value = d
self.assertFalse(self.service.wait_for_deployment(cs_name, dm_name, sec, loop))
d.status = ADStatus.RUNNING
self.assertTrue(self.service.wait_for_deployment(cs_name, dm_name, sec, loop))
def test_get_virtual_machine_instance_status(self):
d = Deployment()
vm_name = "dghu34"
self.assertIsNone(self.service.get_virtual_machine_instance_status(d, vm_name))
r = RoleInstance()
r.instance_name = vm_name
r.instance_status = AVMStatus.READY_ROLE
d.role_instance_list = [r]
self.assertEqual(self.service.get_virtual_machine_instance_status(d, vm_name), AVMStatus.READY_ROLE)
def test_wait_for_virtual_machine(self):
cs_name = "dsandj2"
dm_name = "dshudu2"
vm_name = "dhfdr3f"
sec = 1
loop = 3
self.service.get_deployment_by_name = Mock()
self.service.get_deployment_by_name.return_value = None
self.service.get_virtual_machine_instance_status = Mock()
self.service.get_virtual_machine_instance_status.return_value = AVMStatus.STOPPED
self.assertFalse(
self.service.wait_for_virtual_machine(cs_name, dm_name, vm_name, sec, loop, AVMStatus.READY_ROLE)
)
self.service.get_virtual_machine_instance_status.return_value = AVMStatus.READY_ROLE
self.assertTrue(
self.service.wait_for_virtual_machine(cs_name, dm_name, vm_name, sec, loop, AVMStatus.READY_ROLE)
)
def test_get_virtual_machine_public_endpoint(self):
cs_name = "ds4"
dm_name = "dh3"
vm_name = "fd4"
ep_name = "fj4"
self.service.get_deployment_by_name = Mock()
d = Deployment()
self.service.get_deployment_by_name.return_value = d
self.assertIsNone(self.service.get_virtual_machine_public_endpoint(cs_name, dm_name, vm_name, ep_name))
r = RoleInstance()
r.role_name = vm_name
ep = InstanceEndpoint()
ep.name = ep_name
ep.public_port = "10086"
r.instance_endpoints = [ep]
d.role_instance_list = [r]
self.assertEqual(
self.service.get_virtual_machine_public_endpoint(cs_name, dm_name, vm_name, ep_name), ep.public_port
)
def test_get_virtual_machine_public_ip(self):
cs_name = "ds65"
dm_name = "fdhj"
vm_name = "sd43"
self.service.get_deployment_by_name = Mock()
d = Deployment()
self.service.get_deployment_by_name.return_value = d
self.assertIsNone(self.service.get_virtual_machine_public_ip(cs_name, dm_name, vm_name))
r = RoleInstance()
r.role_name = vm_name
ep = InstanceEndpoint()
ep.vip = "125.125.125.125"
r.instance_endpoints.instance_endpoints = [ep]
d.role_instance_list = [r]
self.assertEqual(self.service.get_virtual_machine_public_ip(cs_name, dm_name, vm_name), ep.vip)
def test_get_virtual_machine_private_ip(self):
cs_name = "dsfd"
dm_name = "fdh4"
vm_name = "fgr5"
self.service.get_deployment_by_name = Mock()
d = Deployment()
self.service.get_deployment_by_name.return_value = d
self.assertIsNone(self.service.get_virtual_machine_private_ip(cs_name, dm_name, vm_name))
r = RoleInstance()
r.role_name = vm_name
r.ip_address = "125.125.125.125"
d.role_instance_list = [r]
self.assertEqual(self.service.get_virtual_machine_private_ip(cs_name, dm_name, vm_name), r.ip_address)
def test_virtual_machine_exists(self):
cs_name = "fhdjf"
dm_name = "fdij5"
vm_name = "grh5k"
self.service.get_virtual_machine = Mock()
self.service.get_virtual_machine.return_value = None
self.assertFalse(self.service.virtual_machine_exists(cs_name, dm_name, vm_name))
self.service.get_virtual_machine.return_value = not None
self.assertTrue(self.service.virtual_machine_exists(cs_name, dm_name, vm_name))
self.service.get_virtual_machine.side_effect = Exception
self.assertFalse(self.service.virtual_machine_exists(cs_name, dm_name, vm_name))
def test_get_virtual_machine_network_config(self):
cs_name = "grtit3"
dm_name = "fdjhfr"
vm_name = "grt65g"
self.service.get_virtual_machine = Mock()
self.service.get_virtual_machine.return_value = None
self.assertIsNone(self.service.get_virtual_machine_network_config(cs_name, dm_name, vm_name))
c = ConfigurationSet()
p = PersistentVMRole()
p.configuration_sets.configuration_sets = [c]
self.service.get_virtual_machine.return_value = p
self.assertEqual(self.service.get_virtual_machine_network_config(cs_name, dm_name, vm_name), c)
def test_get_assigned_endpoints(self):
cs_name = "dsfsd"
self.service.get_hosted_service_properties = Mock()
i = ConfigurationSetInputEndpoint("http", "tcp", "80", "80")
c = ConfigurationSet()
c.input_endpoints.input_endpoints = [i]
r = Role()
r.configuration_sets.configuration_sets = [c]
d = Deployment()
d.role_list.roles = [r]
h = HostedService()
h.deployments.deployments = [d]
self.service.get_hosted_service_properties.return_value = h
self.assertEqual(len(self.service.get_assigned_endpoints(cs_name)), 1)
def test_wait_for_async(self):
r_id = 1
sec = 1
loop = 3
self.service.get_operation_status = Mock()
o = Operation()
o.status = "InProgress"
self.service.get_operation_status.return_value = o
self.assertFalse(self.service.wait_for_async(r_id, sec, loop))
o.status = "Succeeded"
self.assertTrue(self.service.wait_for_async(r_id, sec, loop))
class ServiceTest(unittest.TestCase):
def setUp(self):
db_adapter_url = 'hackathon.azureformation.service.db_adapter'
with mock.patch(db_adapter_url) as db_adapter:
subscription_id = 'hter45sd-gf34-gfd4-ht34-ds43gree23gr'
pem_url = '/home/azureformation.pem'
management_host = 'das342.fdsf34.com'
db_adapter.get_object.return_value = AzureKey(subscription_id=subscription_id,
pem_url=pem_url,
management_host=management_host)
azure_key_id = 0
self.service = AzureAdapter(azure_key_id)
def tearDown(self):
pass
def test_storage_account_exists(self):
name = 'djfh434'
self.service.get_storage_account_properties = Mock()
self.service.get_storage_account_properties.return_value = None
self.assertFalse(self.service.storage_account_exists(name))
self.service.get_storage_account_properties.return_value = not None
self.assertTrue(self.service.storage_account_exists(name))
self.service.get_storage_account_properties.side_effect = Exception
self.assertFalse(self.service.storage_account_exists(name))
def test_cloud_service_exists(self):
name = 'sdhsj6598'
self.service.get_hosted_service_properties = Mock()
self.service.get_hosted_service_properties.return_value = None
self.assertFalse(self.service.cloud_service_exists(name))
self.service.get_hosted_service_properties.return_value = not None
self.assertTrue(self.service.cloud_service_exists(name))
self.service.get_hosted_service_properties.side_effect = Exception
self.assertFalse(self.service.cloud_service_exists(name))
def test_deployment_exists(self):
name = 'fhjd545'
slot = 'df8dfkj'
self.service.get_deployment_by_slot = Mock()
self.service.get_deployment_by_slot.return_value = None
self.assertFalse(self.service.deployment_exists(name, slot))
self.service.get_deployment_by_slot.return_value = not None
self.assertTrue(self.service.deployment_exists(name, slot))
self.service.get_deployment_by_slot.side_effect = Exception
self.assertFalse(self.service.deployment_exists(name, slot))
def test_wait_for_deployment(self):
cs_name = 'dhsj23'
dm_name = 'fdu43f'
sec = 1
loop = 3
self.service.get_deployment_by_name = Mock()
d = Deployment()
self.service.get_deployment_by_name.return_value = d
self.assertFalse(self.service.wait_for_deployment(cs_name, dm_name, sec, loop))
d.status = ADStatus.RUNNING
self.assertTrue(self.service.wait_for_deployment(cs_name, dm_name, sec, loop))
def test_get_virtual_machine_instance_status(self):
d = Deployment()
vm_name = 'dghu34'
self.assertIsNone(self.service.get_virtual_machine_instance_status(d, vm_name))
r = RoleInstance()
r.instance_name = vm_name
r.instance_status = AVMStatus.READY_ROLE
d.role_instance_list = [r]
self.assertEqual(self.service.get_virtual_machine_instance_status(d, vm_name), AVMStatus.READY_ROLE)
def test_wait_for_virtual_machine(self):
cs_name = 'dsandj2'
dm_name = 'dshudu2'
vm_name = 'dhfdr3f'
sec = 1
loop = 3
self.service.get_deployment_by_name = Mock()
self.service.get_deployment_by_name.return_value = None
self.service.get_virtual_machine_instance_status = Mock()
self.service.get_virtual_machine_instance_status.return_value = AVMStatus.STOPPED
self.assertFalse(
self.service.wait_for_virtual_machine(cs_name, dm_name, vm_name, sec, loop, AVMStatus.READY_ROLE)
)
self.service.get_virtual_machine_instance_status.return_value = AVMStatus.READY_ROLE
self.assertTrue(
self.service.wait_for_virtual_machine(cs_name, dm_name, vm_name, sec, loop, AVMStatus.READY_ROLE)
)
def test_get_virtual_machine_public_endpoint(self):
cs_name = 'ds4'
dm_name = 'dh3'
vm_name = 'fd4'
ep_name = 'fj4'
self.service.get_deployment_by_name = Mock()
d = Deployment()
self.service.get_deployment_by_name.return_value = d
self.assertIsNone(self.service.get_virtual_machine_public_endpoint(cs_name, dm_name, vm_name, ep_name))
r = RoleInstance()
r.role_name = vm_name
ep = InstanceEndpoint()
ep.name = ep_name
ep.public_port = '10086'
r.instance_endpoints = [ep]
d.role_instance_list = [r]
self.assertEqual(
self.service.get_virtual_machine_public_endpoint(cs_name, dm_name, vm_name, ep_name),
ep.public_port
)
def test_get_virtual_machine_public_ip(self):
cs_name = 'ds65'
dm_name = 'fdhj'
vm_name = 'sd43'
self.service.get_deployment_by_name = Mock()
d = Deployment()
self.service.get_deployment_by_name.return_value = d
self.assertIsNone(self.service.get_virtual_machine_public_ip(cs_name, dm_name, vm_name))
r = RoleInstance()
r.role_name = vm_name
ep = InstanceEndpoint()
ep.vip = '125.125.125.125'
r.instance_endpoints.instance_endpoints = [ep]
d.role_instance_list = [r]
self.assertEqual(self.service.get_virtual_machine_public_ip(cs_name, dm_name, vm_name), ep.vip)
def test_get_virtual_machine_private_ip(self):
cs_name = 'dsfd'
dm_name = 'fdh4'
vm_name = 'fgr5'
self.service.get_deployment_by_name = Mock()
d = Deployment()
self.service.get_deployment_by_name.return_value = d
self.assertIsNone(self.service.get_virtual_machine_private_ip(cs_name, dm_name, vm_name))
r = RoleInstance()
r.role_name = vm_name
r.ip_address = '125.125.125.125'
d.role_instance_list = [r]
self.assertEqual(self.service.get_virtual_machine_private_ip(cs_name, dm_name, vm_name), r.ip_address)
def test_virtual_machine_exists(self):
cs_name = 'fhdjf'
dm_name = 'fdij5'
vm_name = 'grh5k'
self.service.get_virtual_machine = Mock()
self.service.get_virtual_machine.return_value = None
self.assertFalse(self.service.virtual_machine_exists(cs_name, dm_name, vm_name))
self.service.get_virtual_machine.return_value = not None
self.assertTrue(self.service.virtual_machine_exists(cs_name, dm_name, vm_name))
self.service.get_virtual_machine.side_effect = Exception
self.assertFalse(self.service.virtual_machine_exists(cs_name, dm_name, vm_name))
def test_get_virtual_machine_network_config(self):
cs_name = 'grtit3'
dm_name = 'fdjhfr'
vm_name = 'grt65g'
self.service.get_virtual_machine = Mock()
self.service.get_virtual_machine.return_value = None
self.assertIsNone(self.service.get_virtual_machine_network_config(cs_name, dm_name, vm_name))
c = ConfigurationSet()
p = PersistentVMRole()
p.configuration_sets.configuration_sets = [c]
self.service.get_virtual_machine.return_value = p
self.assertEqual(self.service.get_virtual_machine_network_config(cs_name, dm_name, vm_name), c)
def test_get_assigned_endpoints(self):
cs_name = 'dsfsd'
self.service.get_hosted_service_properties = Mock()
i = ConfigurationSetInputEndpoint('http', 'tcp', '80', '80')
c = ConfigurationSet()
c.input_endpoints.input_endpoints = [i]
r = Role()
r.configuration_sets.configuration_sets = [c]
d = Deployment()
d.role_list.roles = [r]
h = HostedService()
h.deployments.deployments = [d]
self.service.get_hosted_service_properties.return_value = h
self.assertEqual(len(self.service.get_assigned_endpoints(cs_name)), 1)
def test_wait_for_async(self):
r_id = 1
sec = 1
loop = 3
self.service.get_operation_status = Mock()
o = Operation()
o.status = 'InProgress'
self.service.get_operation_status.return_value = o
self.assertFalse(self.service.wait_for_async(r_id, sec, loop))
o.status = 'Succeeded'
self.assertTrue(self.service.wait_for_async(r_id, sec, loop))
STATUS: HEALTH_STATUS.OK
}
except Exception as e:
self.log.error(e)
return {
STATUS: HEALTH_STATUS.ERROR
}
class AzureHealthCheck(HealthCheck):
"""Check the status of azure to make sure config is right and azure is available"""
def report_health(self):
azure_key = self.db.find_first_object(AzureKey)
<<<<<<< HEAD
azure = AzureAdapter(azure_key.id)
=======
if not azure_key:
return {
STATUS: HEALTH_STATUS.WARNING,
DESCRIPTION: "No Azure key found"
}
azure = Service(azure_key.id)
>>>>>>> msopentech/master
if azure.ping():
return {
STATUS: HEALTH_STATUS.OK
}
else:
return {
STATUS: HEALTH_STATUS.ERROR
