skeleton unit test

tests/unit/modules/test_vagrant.py

@@ -0,0 +1,89 @@
+# -*- coding: utf-8 -*-
+
+# Import python libs
+from __future__ import absolute_import
+import re
+
+# Import Salt Testing libs
+from tests.support.mixins import LoaderModuleMockMixin
+from tests.support.unit import skipIf, TestCase
+from tests.support.mock import NO_MOCK, NO_MOCK_REASON, MagicMock, patch
+
+# Import salt libs
+import salt.modules.vagrant as vagrant
+import salt.modules.config as config
+from salt._compat import ElementTree as ET
+import salt.utils
+
+# Import third party libs
+import yaml
+from salt.ext import six
+
+
+@skipIf(NO_MOCK, NO_MOCK_REASON)
+class VagrantTestCase(TestCase, LoaderModuleMockMixin):
+    '''
+    Unit TestCase for the salt.modules.vagrant module.
+    '''
+    # def setup_loader_modules(self):
+    #     return {vsphere: {'__virtual__': MagicMock(return_value='vsphere')}}
+
+    def setup_loader_modules(self):
+        loader_globals = {
+            '__salt__': {
+                'config.get': config.get,
+                'config.option': config.option,
+            }
+        }
+        return {vagrant: loader_globals, config: loader_globals}
+
+    def test_boot_default_dev(self):
+        diskp = vagrant._disk_profile('default', 'kvm')
+        nicp = vagrant._nic_profile('default', 'kvm')
+        xml_data = vagrant._gen_xml(
+            'hello',
+            1,
+            512,
+            diskp,
+            nicp,
+            'kvm'
+            )
+        root = ET.fromstring(xml_data)
+        self.assertEqual(root.find('os/boot').attrib['dev'], 'hd')
+
+
+    def test_gen_xml_for_telnet_console(self):
+        diskp = vagrant._disk_profile('default', 'kvm')
+        nicp = vagrant._nic_profile('default', 'kvm')
+        xml_data = vagrant._gen_xml(
+            'hello',
+            1,
+            512,
+            diskp,
+            nicp,
+            'kvm',
+            serial_type='tcp',
+            console=True,
+            telnet_port=22223
+            )
+        root = ET.fromstring(xml_data)
+        self.assertEqual(root.find('devices/serial').attrib['type'], 'tcp')
+        self.assertEqual(root.find('devices/console').attrib['type'], 'tcp')
+        self.assertEqual(root.find('devices/console/source').attrib['service'], '22223')
+
+
+    def test_controller_for_kvm(self):
+        diskp = vagrant._disk_profile('default', 'kvm')
+        nicp = vagrant._nic_profile('default', 'kvm')
+        xml_data = vagrant._gen_xml(
+            'hello',
+            1,
+            512,
+            diskp,
+            nicp,
+            'kvm'
+            )
+        root = ET.fromstring(xml_data)
+        controllers = root.findall('.//devices/controller')
+        # There should be no controller
+        self.assertTrue(len(controllers) == 0)