def setUp(self):
self.driver = VCenterShellDriver()
self.resource = create_autospec(ResourceInfo)
self.resource.ResourceModelName = 'VMware vCenter'
self.resource.ResourceAttributes = {
'User': '******',
'Password': '******',
'Default dvSwitch': 'switch1',
'Holding Network': 'anetwork',
'Default Port Group Location': 'Quali',
'VM Cluster': 'Quali',
'VM Location': 'Quali',
'VM Resource Pool': 'Quali',
'VM Storage': 'Quali',
'Shutdown Method': 'hard',
'OVF Tool Path': 'C\\program files\ovf',
'Execution Server Selector': '',
'Reserved Networks': 'vlan65',
'Default Datacenter': 'QualiSB'
}
self.context = Mock()
self.context.resource = self.resource
self.driver.command_orchestrator = MagicMock()
self.cancellation_context = Mock()
self.ports = Mock()
def setUp(self):
self.driver = VCenterShellDriver()
self.resource = create_autospec(ResourceInfo)
self.resource.ResourceModelName = 'VMware vCenter'
self.resource.ResourceAttributes = {'User': '******',
'Password': '******',
'Default dvSwitch': 'switch1',
'Holding Network': 'anetwork',
'Default Port Group Location': 'Quali',
'VM Cluster': 'Quali',
'VM Location': 'Quali',
'VM Resource Pool': 'Quali',
'VM Storage': 'Quali',
'Shutdown Method': 'hard',
'OVF Tool Path': 'C\\program files\ovf',
'Execution Server Selector': '',
'Reserved Networks': 'vlan65',
'Default Datacenter': 'QualiSB'}
self.context = Mock()
self.context.resource = self.resource
self.driver.command_orchestrator = MagicMock()
self.cancellation_context = Mock()
self.ports = Mock()
class TestCommandOrchestrator(TestCase):
def setUp(self):
self.driver = VCenterShellDriver()
self.resource = create_autospec(ResourceInfo)
self.resource.ResourceModelName = 'VMware vCenter'
self.resource.ResourceAttributes = {
'User': '******',
'Password': '******',
'Default dvSwitch': 'switch1',
'Holding Network': 'anetwork',
'Default Port Group Location': 'Quali',
'VM Cluster': 'Quali',
'VM Location': 'Quali',
'VM Resource Pool': 'Quali',
'VM Storage': 'Quali',
'Shutdown Method': 'hard',
'OVF Tool Path': 'C\\program files\ovf',
'Execution Server Selector': '',
'Reserved Networks': 'vlan65',
'Default Datacenter': 'QualiSB'
}
self.context = Mock()
self.context.resource = self.resource
self.driver.command_orchestrator = MagicMock()
self.cancellation_context = Mock()
self.ports = Mock()
def test_init(self):
self.driver.initialize()
def test_connect_bulk(self):
self.setUp()
requset = Mock()
res = self.driver.ApplyConnectivityChanges(self.context, requset)
self.assertIsNotNone(res)
self.assertTrue(
self.driver.command_orchestrator.connect_bulk.called_with(
self.context, requset, []))
def test_disconnect_all(self):
self.setUp()
res = self.driver.disconnect_all(self.context, self.ports)
self.assertIsNotNone(res)
self.assertTrue(
self.driver.command_orchestrator.disconnect_all.called_with(
self.context, self.ports))
def test_disconnect(self):
self.setUp()
network_name = Mock()
res = self.driver.disconnect(self.context, self.ports, network_name)
self.assertIsNotNone(res)
self.assertTrue(
self.driver.command_orchestrator.disconnect.called_with(
self.context, self.ports, network_name))
def test_DeleteInstance(self):
self.setUp()
res = self.driver.DeleteInstance(self.context, self.ports)
self.assertIsNotNone(res)
self.assertTrue(
self.driver.command_orchestrator.DeleteInstance.called_with(
self.context, self.ports))
def test_deploy_from_template(self):
self.setUp()
deploy_data = Mock()
res = self.driver.deploy_from_template(self.context, deploy_data,
Mock())
self.assertIsNotNone(res)
self.assertTrue(
self.driver.command_orchestrator.deploy_from_template.called_with(
self.context, deploy_data, Mock()))
def test_deploy_from_image(self):
self.setUp()
deploy_data = Mock()
res = self.driver.deploy_from_image(self.context, deploy_data, Mock())
self.assertIsNotNone(res)
self.assertTrue(
self.driver.command_orchestrator.deploy_from_image.called_with(
self.context, deploy_data))
def test_power_off(self):
self.setUp()
res = self.driver.PowerOff(self.context, self.ports)
self.assertIsNotNone(res)
self.assertTrue(
self.driver.command_orchestrator.power_off.called_with(
self.context, self.ports))
def test_power_on(self):
self.setUp()
res = self.driver.PowerOn(self.context, self.ports)
self.assertIsNotNone(res)
self.assertTrue(
self.driver.command_orchestrator.power_on.called_with(
self.context, self.ports))
def test_power_cycle(self):
self.setUp()
res = self.driver.PowerCycle(self.context, self.ports, 10)
self.assertIsNotNone(res)
def test_refresh_ip(self):
self.setUp()
res = self.driver.remote_refresh_ip(self.context,
self.cancellation_context,
self.ports)
self.assertIsNotNone(res)
self.assertTrue(
self.driver.command_orchestrator._refresh_ip.called_with(
self.context, self.cancellation_context, self.ports))
class TestCommandOrchestrator(TestCase):
def setUp(self):
self.driver = VCenterShellDriver()
self.resource = create_autospec(ResourceInfo)
self.resource.ResourceModelName = 'VMware vCenter'
self.resource.ResourceAttributes = {'User': '******',
'Password': '******',
'Default dvSwitch': 'switch1',
'Holding Network': 'anetwork',
'Default Port Group Location': 'Quali',
'VM Cluster': 'Quali',
'VM Location': 'Quali',
'VM Resource Pool': 'Quali',
'VM Storage': 'Quali',
'Shutdown Method': 'hard',
'OVF Tool Path': 'C\\program files\ovf',
'Execution Server Selector': '',
'Reserved Networks': 'vlan65',
'Default Datacenter': 'QualiSB'}
self.context = Mock()
self.context.resource = self.resource
self.driver.command_orchestrator = MagicMock()
self.cancellation_context = Mock()
self.ports = Mock()
def test_init(self):
self.driver.initialize()
def test_connect_bulk(self):
self.setUp()
requset = Mock()
res = self.driver.ApplyConnectivityChanges(self.context, requset)
self.assertIsNotNone(res)
self.assertTrue(self.driver.command_orchestrator.connect_bulk.called_with(self.context, requset, []))
def test_disconnect_all(self):
self.setUp()
res = self.driver.disconnect_all(self.context, self.ports)
self.assertIsNotNone(res)
self.assertTrue(self.driver.command_orchestrator.disconnect_all.called_with(self.context, self.ports))
def test_disconnect(self):
self.setUp()
network_name = Mock()
res = self.driver.disconnect(self.context, self.ports, network_name)
self.assertIsNotNone(res)
self.assertTrue(self.driver.command_orchestrator.disconnect.called_with(self.context, self.ports, network_name))
def test_destroy_vm_only(self):
self.setUp()
res = self.driver.destroy_vm_only(self.context, self.ports)
self.assertIsNotNone(res)
self.assertTrue(self.driver.command_orchestrator.destroy_vm_only.called_with(self.context, self.ports))
def test_deploy_from_template(self):
self.setUp()
deploy_data = Mock()
res = self.driver.deploy_from_template(self.context, deploy_data)
self.assertIsNotNone(res)
self.assertTrue(self.driver.command_orchestrator.deploy_from_template.called_with(self.context,
deploy_data, Mock()))
def test_deploy_from_image(self):
self.setUp()
deploy_data = Mock()
res = self.driver.deploy_from_image(self.context, deploy_data)
self.assertIsNotNone(res)
self.assertTrue(self.driver.command_orchestrator.deploy_from_image.called_with(self.context,
deploy_data))
def test_power_off(self):
self.setUp()
res = self.driver.PowerOff(self.context, self.ports)
self.assertIsNotNone(res)
self.assertTrue(self.driver.command_orchestrator.power_off.called_with(self.context, self.ports))
def test_power_on(self):
self.setUp()
res = self.driver.PowerOn(self.context, self.ports)
self.assertIsNotNone(res)
self.assertTrue(self.driver.command_orchestrator.power_on.called_with(self.context, self.ports))
def test_power_cycle(self):
self.setUp()
res = self.driver.PowerCycle(self.context, self.ports, 10)
self.assertIsNotNone(res)
def test_refresh_ip(self):
self.setUp()
res = self.driver.remote_refresh_ip(self.context, self.cancellation_context, self.ports)
self.assertIsNotNone(res)
self.assertTrue(self.driver.command_orchestrator._refresh_ip.called_with(self.context, self.cancellation_context, self.ports))
