Merge pull request #22982 from jayeshka/qemu_img-unit-test

adding qemu_img unit test case

tests/unit/modules/qemu_img_test.py

@@ -0,0 +1,61 @@
+# -*- coding: utf-8 -*-
+'''
+    :codeauthor: :email:`Rupesh Tare <rupesht@saltstack.com>`
+'''
+
+# Import Python Libs
+from __future__ import absolute_import
+
+# Import Salt Testing Libs
+from salttesting import TestCase, skipIf
+from salttesting.mock import (
+    MagicMock,
+    patch,
+    NO_MOCK,
+    NO_MOCK_REASON
+)
+
+from salttesting.helpers import ensure_in_syspath
+
+ensure_in_syspath('../../')
+
+# Import Salt Libs
+from salt.modules import qemu_img
+import os
+
+# Globals
+qemu_img.__salt__ = {}
+
+
+@skipIf(NO_MOCK, NO_MOCK_REASON)
+class QemuimgTestCase(TestCase):
+    '''
+    Test cases for salt.modules.qemu_img
+    '''
+    def test_make_image(self):
+        '''
+        Test for create a blank virtual machine image file
+        of the specified size in megabytes
+        '''
+        with patch.object(os.path, 'isabs',
+                          MagicMock(side_effect=[False, True, True, True])):
+            self.assertEqual(qemu_img.make_image('location', 'size', 'fmt'), '')
+
+            with patch.object(os.path, 'isdir',
+                              MagicMock(side_effect=[False, True, True])):
+                self.assertEqual(qemu_img.make_image('location', 'size', 'fmt'),
+                                 '')
+
+                with patch.dict(qemu_img.__salt__,
+                                {'cmd.retcode': MagicMock(side_effect=[False,
+                                                                       True])}):
+                    self.assertEqual(qemu_img.make_image('location', 'size',
+                                                         'fmt'), 'location')
+
+                    self.assertEqual(qemu_img.make_image('location', 'size',
+                                                         'fmt'), '')
+
+
+if __name__ == '__main__':
+    from integration import run_tests
+    run_tests(QemuimgTestCase, needs_daemon=False)