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_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_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_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_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_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_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_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_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_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_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)