Merge pull request #781 from QualiSystems/gil_773

Gil 773

Author: GilGald

static_vm_package/VCenterAutoloadStaticVMDriver/__init__.py

@@ -24,6 +24,7 @@ def __init__(self):
         self.ip_manager = VMIPManager()
         self.task_waiter = SynchronousTaskWaiter()
         self.logger = get_qs_logger('VM AutoLoad')
+        self.pv_service = pyVmomiService(SmartConnect, Disconnect, self.task_waiter)
 
     def get_inventory(self, context):
         """
@@ -44,18 +45,18 @@ def get_inventory(self, context):
         vcenter_resource = self.model_parser.convert_to_vcenter_model(vcenter_api_res)
 
         si = None
-        pv_service = pyVmomiService(SmartConnect, Disconnect, self.task_waiter)
+
 
         try:
             self.logger.info('connecting to vcenter ({0})'.format(vcenter_api_res.Address))
-            si = self._get_connection_to_vcenter(pv_service, session, vcenter_resource, vcenter_api_res.Address)
+            si = self._get_connection_to_vcenter(self.pv_service, session, vcenter_resource, vcenter_api_res.Address)
 
             self.logger.info('loading vm uuid')
-            vm_loader = VMLoader(pv_service)
+            vm_loader = VMLoader(self.pv_service)
             uuid = vm_loader.load_vm_uuid_by_name(si, vcenter_resource, vcenter_vm_name)
             self.logger.info('vm uuid: {0}'.format(uuid))
             self.logger.info('loading the ip of the vm')
-            ip = self._try_get_ip(pv_service, si, uuid, vcenter_resource)
+            ip = self._try_get_ip(self.pv_service, si, uuid, vcenter_resource)
             if ip:
                 session.UpdateResourceAddress(context.resource.name, ip)
 
@@ -64,7 +65,7 @@ def get_inventory(self, context):
             raise
         finally:
             if si:
-                pv_service.disconnect(si)
+                self.pv_service.disconnect(si)
 
         return self._get_auto_load_response(uuid, vcenter_name, context.resource)
 

static_vm_package/VCenterAutoloadStaticVMDriver/app_discovery/vm_autoload_driver.py

@@ -0,0 +1,58 @@
+import uuid
+from unittest import TestCase
+
+import jsonpickle
+from pyVmomi import vim
+from mock import Mock, create_autospec, patch
+from pyVmomi.VmomiSupport import vmodlNames
+
+from static_vm_package.VCenterAutoloadStaticVMDriver.app_discovery.vm_autoload_driver import \
+    DeployAppOrchestrationDriver
+
+
+class TestGetInventory(TestCase):
+    def setUp(self):
+        self.deploy_app_orchestration_driver = DeployAppOrchestrationDriver()
+        self.context = Mock()
+        self.context.resource.attributes = Mock()
+        self.VMLoader = Mock()
+
+    def test_get_inventory(self):
+        vcenter_data_model = Mock()
+        vcenter_data_model.default_datacenter = 'name'
+        self.deploy_app_orchestration_driver.cs_helper.get_session = Mock(return_value=Mock())
+        self.deploy_app_orchestration_driver.model_parser.convert_to_vcenter_model = Mock(
+            return_value=vcenter_data_model)
+        self.deploy_app_orchestration_driver._get_connection_to_vcenter = Mock(return_value=Mock())
+        self.deploy_app_orchestration_driver._try_get_ip = Mock(return_value=Mock())
+        self.deploy_app_orchestration_driver._get_vm_details = Mock(return_value=Mock())
+        self.deploy_app_orchestration_driver.pv_service.find_vm_by_name = Mock(
+            return_value=vim.VirtualMachine(Mock(), Mock()))
+
+        # deployer.deploy_from_image = Mock(return_value=res)
+        self.context.resource.attributes = {'vCenter VM': 'dd', 'vCenter Name': 'd2'}
+
+        self.logger = Mock()
+        # Act
+
+        self.deploy_app_orchestration_driver.get_inventory(self.context)
+        # Assert
+
+        self.assertTrue(self.deploy_app_orchestration_driver.model_parser.convert_to_vcenter_model.called)
+        self.assertTrue(self.deploy_app_orchestration_driver.cs_helper.get_session.called)
+        self.assertTrue(self.deploy_app_orchestration_driver.model_parser.convert_to_vcenter_model.valled)
+        self.assertTrue(self.deploy_app_orchestration_driver._get_connection_to_vcenter.called)
+        self.assertTrue(self.deploy_app_orchestration_driver._try_get_ip.called)
+        self.assertTrue(self.deploy_app_orchestration_driver._get_vm_details.called)
+        self.assertTrue(self.deploy_app_orchestration_driver.pv_service.find_vm_by_name.called)
+
+    def test_get_vm_details(self):
+        vm_details = self.deploy_app_orchestration_driver._get_vm_details(uuid="Piplup",
+                                                                          vcenter_name="Prinplup",
+                                                                          resource="Empoleon")
+        str_vm_details = jsonpickle.decode(vm_details)
+        self.assertTrue(str_vm_details['CloudProviderName']=="Prinplup")
+        self.assertTrue(str_vm_details['UID']=='Piplup')
+
+
+