Merge pull request #22530 from jayeshka/win_disk-unit-test

adding win_disk unit test case.

tests/unit/modules/win_disk_test.py

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+'''
+    :codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
+'''
+
+# 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)