1 files changed, 131 insertions, 0 deletions
diff --git a/nova/tests/unit/virt/vmwareapi/stubs.py b/nova/tests/unit/virt/vmwareapi/stubs.py
new file mode 100644
index 0000000000..d126b36e0f
--- /dev/null
+++ b/nova/tests/unit/virt/vmwareapi/stubs.py
@@ -0,0 +1,131 @@
+# Copyright (c) 2011 Citrix Systems, Inc.
+# Copyright 2011 OpenStack Foundation
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+
+"""
+Stubouts for the test suite
+"""
+
+import contextlib
+
+import mock
+from oslo.vmware import exceptions as vexc
+
+from nova import db
+from nova.tests.unit import test_flavors
+from nova.tests.unit.virt.vmwareapi import fake
+from nova.virt.vmwareapi import driver
+from nova.virt.vmwareapi import images
+from nova.virt.vmwareapi import network_util
+
+
+def fake_get_vim_object(arg):
+    """Stubs out the VMwareAPISession's get_vim_object method."""
+    return fake.FakeVim()
+
+
+@property
+def fake_vim_prop(arg):
+    """Stubs out the VMwareAPISession's vim property access method."""
+    return fake.get_fake_vim_object(arg)
+
+
+def fake_is_vim_object(arg, module):
+    """Stubs out the VMwareAPISession's is_vim_object method."""
+    return isinstance(module, fake.FakeVim)
+
+
+def fake_temp_method_exception():
+    raise vexc.VimFaultException(
+            [vexc.NOT_AUTHENTICATED],
+            "Session Empty/Not Authenticated")
+
+
+def fake_temp_session_exception():
+    raise vexc.VimConnectionException("it's a fake!",
+            "Session Exception")
+
+
+def fake_session_file_exception():
+    fault_list = [vexc.FILE_ALREADY_EXISTS]
+    raise vexc.VimFaultException(fault_list,
+                                 Exception('fake'))
+
+
+def fake_session_permission_exception():
+    fault_list = [vexc.NO_PERMISSION]
+    fault_string = 'Permission to perform this operation was denied.'
+    details = {'privilegeId': 'Resource.AssignVMToPool', 'object': 'domain-c7'}
+    raise vexc.VimFaultException(fault_list, fault_string, details=details)
+
+
+def _fake_flavor_get(context, id):
+    for instance_type in test_flavors.DEFAULT_FLAVORS:
+        if instance_type['id'] == id:
+            return instance_type
+    return {'memory_mb': 128, 'root_gb': 0, 'deleted_at': None,
+            'name': 'm1.micro', 'deleted': 0, 'created_at': None,
+            'ephemeral_gb': 0, 'updated_at': None,
+            'disabled': False, 'vcpus': 1, 'extra_specs': {},
+            'swap': 0, 'rxtx_factor': 1.0, 'is_public': True,
+            'flavorid': '1', 'vcpu_weight': None, 'id': 2}
+
+
+def set_stubs(stubs):
+    """Set the stubs."""
+    stubs.Set(network_util, 'get_network_with_the_name',
+              fake.fake_get_network)
+    stubs.Set(images, 'upload_image', fake.fake_upload_image)
+    stubs.Set(images, 'fetch_image', fake.fake_fetch_image)
+    stubs.Set(driver.VMwareAPISession, "vim", fake_vim_prop)
+    stubs.Set(driver.VMwareAPISession, "_is_vim_object",
+              fake_is_vim_object)
+    stubs.Set(db, 'flavor_get', _fake_flavor_get)
+
+
+def fake_suds_context(calls=None):
+    """Generate a suds client which automatically mocks all SOAP method calls.
+
+    Calls are stored in <calls>, indexed by the name of the call. If you need
+    to mock the behaviour of specific API calls you can pre-populate <calls>
+    with appropriate Mock objects.
+    """
+
+    calls = calls or {}
+
+    class fake_factory:
+        def create(self, name):
+            return mock.NonCallableMagicMock(name=name)
+
+    class fake_service:
+        def __getattr__(self, attr_name):
+            if attr_name in calls:
+                return calls[attr_name]
+
+            mock_call = mock.MagicMock(name=attr_name)
+            calls[attr_name] = mock_call
+            return mock_call
+
+    class fake_client:
+        def __init__(self, wdsl_url, **kwargs):
+            self.service = fake_service()
+            self.factory = fake_factory()
+
+    return contextlib.nested(
+        mock.patch('suds.client.Client', fake_client),
+
+        # As we're not connecting to a real host there's no need to wait
+        # between retries
+        mock.patch.object(driver, 'TIME_BETWEEN_API_CALL_RETRIES', 0)
+    )