diff --git a/tests/unit/modules/win_disk_test.py b/tests/unit/modules/win_disk_test.py new file mode 100644 index 0000000000..c035b24882 --- /dev/null +++ b/tests/unit/modules/win_disk_test.py @@ -0,0 +1,78 @@ +# -*- coding: utf-8 -*- +''' + :codeauthor: :email:`Jayesh Kariya ` +''' + +# Import Python Libs +from __future__ import absolute_import + +# Import Salt Testing Libs +from salttesting import TestCase, skipIf +from salttesting.mock import ( + NO_MOCK, + NO_MOCK_REASON +) + +from salttesting.helpers import ensure_in_syspath + +ensure_in_syspath('../../') + +# Import Salt Libs +from salt.modules import win_disk + + +class MockKernel32(object): + ''' + Mock windll class + ''' + def __init__(self): + pass + + @staticmethod + def GetLogicalDrives(): + ''' + Mock GetLogicalDrives method + ''' + return 1 + + +class MockWindll(object): + ''' + Mock windll class + ''' + def __init__(self): + self.kernel32 = MockKernel32() + + +class MockCtypes(object): + ''' + Mock ctypes class + ''' + def __init__(self): + self.windll = MockWindll() + +win_disk.ctypes = MockCtypes() + + +@skipIf(NO_MOCK, NO_MOCK_REASON) +class WinDiskTestCase(TestCase): + ''' + Test cases for salt.modules.win_disk + ''' + # 'usage' function tests: 1 + + def test_usage(self): + ''' + Test if it return usage information for volumes mounted on this minion. + ''' + self.assertDictEqual(win_disk.usage(), + {'A:\\': {'available': None, + '1K-blocks': None, + 'used': None, + 'capacity': None, + 'filesystem': 'A:\\'}}) + + +if __name__ == '__main__': + from integration import run_tests + run_tests(WinDiskTestCase, needs_daemon=False)