diff --git a/tests/unit/modules/qemu_img_test.py b/tests/unit/modules/qemu_img_test.py new file mode 100644 index 0000000000..ae1541e9bc --- /dev/null +++ b/tests/unit/modules/qemu_img_test.py @@ -0,0 +1,61 @@ +# -*- coding: utf-8 -*- +''' + :codeauthor: :email:`Rupesh Tare ` +''' + +# 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)