* code cleanup

* revised unittest approach
* added stubout and a number of tests

5 files changed, 20 insertions, 79 deletions

diff --git a/nova/virt/xenapi/__init__.py b/nova/virt/xenapi/__init__.py
index 1a2903b98..c7038deae 100644
--- a/nova/virt/xenapi/__init__.py
+++ b/nova/virt/xenapi/__init__.py
@@ -20,31 +20,11 @@
 """
 
 
-def load_sdk(flags):
-    """
-    This method is used for loading the XenAPI SDK (fake or real)
-    """
-    xenapi_module = \
-        flags.xenapi_use_fake_session and 'nova.virt.xenapi.fake' or 'XenAPI'
-    from_list = \
-        flags.xenapi_use_fake_session and ['fake'] or []
-
-    return __import__(xenapi_module, globals(), locals(), from_list, -1)
-
-
 class HelperBase():
     """
-    The class that wraps the helper methods together.
+    The base for helper classes. This adds the XenAPI class attribute
     """
     XenAPI = None
 
     def __init__(self):
         return
-
-    @classmethod
-    def late_import(cls, FLAGS):
-        """
-        Load XenAPI module in for helper class
-        """
-        if cls.XenAPI is None:
-            cls.XenAPI = load_sdk(FLAGS)
diff --git a/nova/virt/xenapi/fake.py b/nova/virt/xenapi/fake.py
index f5fea3cc2..038064e8e 100644
--- a/nova/virt/xenapi/fake.py
+++ b/nova/virt/xenapi/fake.py
@@ -154,7 +154,7 @@ class Failure(Exception):
     def __str__(self):
         try:
             return str(self.details)
-        except Exception, exn:
+        except Exception, exc:
             return "XenAPI Fake Failure: %s" % str(self.details)
 
     def _details_map(self):
@@ -324,8 +324,8 @@ class SessionBase(object):
         try:
             task['result'] = self.xenapi_request(func, params[1:])
             task['status'] = 'success'
-        except Failure, exn:
-            task['error_info'] = exn.details
+        except Failure, exc:
+            task['error_info'] = exc.details
             task['status'] = 'failed'
         task['finished'] = datetime.datetime.now()
         return task_ref
@@ -372,37 +372,3 @@ class _Dispatcher:
 
     def __call__(self, *args):
         return self.__send(self.__name, args)
-
-
-class FakeSession(SessionBase):
-    def __init__(self, uri):
-        super(FakeSession, self).__init__(uri)
-
-    def network_get_all_records_where(self, _1, _2):
-        return self.xenapi.network.get_all_records()
-
-    def host_call_plugin(self, _1, _2, _3, _4, _5):
-        return ''
-
-    def VM_start(self, _1, ref, _2, _3):
-        vm = get_record('VM', ref)
-        if vm['power_state'] != 'Halted':
-            raise Failure(['VM_BAD_POWER_STATE', ref, 'Halted',
-                                  vm['power_state']])
-        vm['power_state'] = 'Running'
-
-    def VBD_plug(self, _1, _2):
-        #FIXME(armando):make proper plug
-        pass
-
-    def VDI_introduce(self, _1, uuid, _2, _3, _4, _5,
-                      _6, _7, _8, _9, _10, _11):
-        #FIXME(armando):make proper introduce
-        valid_vdi = False
-        refs = get_all('VDI')
-        for ref in refs:
-            rec = get_record('VDI', ref)
-            if rec['uuid'] == uuid:
-                valid_vdi = True
-        if not valid_vdi:
-            raise Failure([['INVALID_VDI', 'session', self._session]])
diff --git a/nova/virt/xenapi/vmops.py b/nova/virt/xenapi/vmops.py
index 36b8fecc2..abaa1f5f1 100644
--- a/nova/virt/xenapi/vmops.py
+++ b/nova/virt/xenapi/vmops.py
@@ -28,7 +28,6 @@ from nova import flags
 from nova import exception
 
 from nova.auth.manager import AuthManager
-from nova.virt.xenapi import load_sdk
 from nova.virt.xenapi.network_utils import NetworkHelper
 from nova.virt.xenapi.vm_utils import VMHelper
 
@@ -38,10 +37,9 @@ class VMOps(object):
     Management class for VM-related tasks
     """
     def __init__(self, session):
-        self.XenAPI = load_sdk(flags.FLAGS)
+        self.XenAPI = session.get_imported_xenapi()
         self._session = session
-        # Load XenAPI module in the helper class
-        VMHelper.late_import(flags.FLAGS)
+        VMHelper.XenAPI = self.XenAPI
 
     def list_instances(self):
         """ List VM instances """
diff --git a/nova/virt/xenapi/volumeops.py b/nova/virt/xenapi/volumeops.py
index 1b337a6ed..68806c4c2 100644
--- a/nova/virt/xenapi/volumeops.py
+++ b/nova/virt/xenapi/volumeops.py
@@ -22,7 +22,6 @@ import logging
 from twisted.internet import defer
 
 from nova import flags
-from nova.virt.xenapi import load_sdk
 from nova.virt.xenapi.vm_utils import VMHelper
 from nova.virt.xenapi.volume_utils import VolumeHelper
 from nova.virt.xenapi.volume_utils import StorageError
@@ -33,11 +32,11 @@ class VolumeOps(object):
     Management class for Volume-related tasks
     """
     def __init__(self, session):
-        self.XenAPI = load_sdk(flags.FLAGS)
+        self.XenAPI = session.get_imported_xenapi()
         self._session = session
         # Load XenAPI module in the helper classes respectively
-        VolumeHelper.late_import(flags.FLAGS)
-        VMHelper.late_import(flags.FLAGS)
+        VolumeHelper.XenAPI = self.XenAPI
+        VMHelper.XenAPI = self.XenAPI
 
     @defer.inlineCallbacks
     def attach_volume(self, instance_name, device_path, mountpoint):
diff --git a/nova/virt/xenapi_conn.py b/nova/virt/xenapi_conn.py
index 5f2b9c7c6..a7e3a6723 100644
--- a/nova/virt/xenapi_conn.py
+++ b/nova/virt/xenapi_conn.py
@@ -36,7 +36,6 @@ reactor thread if the VM.get_by_name_label or VM.get_record calls block.
 
 **Related Flags**
 
-:xenapi_use_fake_session:     To be set for unit testing
 :xenapi_connection_url:  URL for connection to XenServer/Xen Cloud Platform.
 :xenapi_connection_username:  Username for connection to XenServer/Xen Cloud
                               Platform (default: root).
@@ -59,15 +58,11 @@ from twisted.internet import reactor
 
 from nova import utils
 from nova import flags
-from nova.virt.xenapi import load_sdk
 from nova.virt.xenapi.vmops import VMOps
 from nova.virt.xenapi.volumeops import VolumeOps
 
 FLAGS = flags.FLAGS
 
-flags.DEFINE_boolean('xenapi_use_fake_session',
-                    False,
-                    'Set to true in order to use the fake XenAPI SDK')
 flags.DEFINE_string('xenapi_connection_url',
                     None,
                     'URL for connection to XenServer/Xen Cloud Platform.'
@@ -159,15 +154,14 @@ class XenAPIConnection(object):
 class XenAPISession(object):
     """ The session to invoke XenAPI SDK calls """
     def __init__(self, url, user, pw):
-        # This is loaded late so that there's no need to install this
-        # library when not using XenAPI.
-        self.XenAPI = load_sdk(FLAGS)
-        if FLAGS.xenapi_use_fake_session:
-            self._session = self.XenAPI.FakeSession(url)
-        else:
-            self._session = self.XenAPI.Session(url)
+        self.XenAPI = self.get_imported_xenapi()
+        self._session = self._create_session(url)
         self._session.login_with_password(user, pw)
 
+    def get_imported_xenapi(self):
+        """Stubout point. This can be replaced with a mock xenapi module."""
+        return __import__('XenAPI')
+
     def get_xenapi(self):
         """ Return the xenapi object """
         return self._session.xenapi
@@ -200,6 +194,10 @@ class XenAPISession(object):
         reactor.callLater(0, self._poll_task, task, d)
         return d
 
+    def _create_session(self, url):
+        """Stubout point. This can be replaced with a mock session."""
+        return self.XenAPI.Session(url)
+
     @utils.deferredToThread
     def _poll_task(self, task, deferred):
         """Poll the given XenAPI task, and fire the given Deferred if we
@@ -220,7 +218,7 @@ class XenAPISession(object):
                              error_info)
                 deferred.errback(self.XenAPI.Failure(error_info))
             #logging.debug('Polling task %s done.', task)
-        except self.XenAPI.Failure, exc:
+        except Exception, exc:
             logging.warn(exc)
             deferred.errback(exc)