diff --git a/tests/unit/modules/drbd_test.py b/tests/unit/modules/drbd_test.py new file mode 100644 index 0000000000..afd56ac238 --- /dev/null +++ b/tests/unit/modules/drbd_test.py @@ -0,0 +1,70 @@ +# -*- coding: utf-8 -*- +''' + :codeauthor: :email:`Jayesh Kariya ` +''' + +# Import Salt Testing Libs +from salttesting import TestCase, skipIf +from salttesting.mock import ( + MagicMock, + patch, + NO_MOCK, + NO_MOCK_REASON +) + +# Import Salt Libs +from salt.modules import drbd + +# Globals +drbd.__grains__ = {} +drbd.__salt__ = {} +drbd.__context__ = {} + + +@skipIf(NO_MOCK, NO_MOCK_REASON) +class DrbdTestCase(TestCase): + ''' + Test cases for salt.modules.drbd + ''' + # 'overview' function tests: 1 + + def test_overview(self): + ''' + Test if it shows status of the DRBD devices + ''' + ret = {'connection state': 'True', + 'device': 'Stack', + 'fs': 'None', + 'local disk state': 'UpToDate', + 'local role': 'master', + 'minor number': 'Salt', + 'mountpoint': 'True', + 'partner disk state': 'UpToDate', + 'partner role': 'minion', + 'percent': '888', + 'remains': '666', + 'total size': '50', + 'used': '50'} + mock = MagicMock(return_value='Salt:Stack True master/minion \ + UpToDate/UpToDate True None 50 50 666 888') + with patch.dict(drbd.__salt__, {'cmd.run': mock}): + self.assertDictEqual(drbd.overview(), ret) + + ret = {'connection state': 'True', + 'device': 'Stack', + 'local disk state': 'UpToDate', + 'local role': 'master', + 'minor number': 'Salt', + 'partner disk state': 'partner', + 'partner role': 'minion', + 'synched': '5050', + 'synchronisation: ': 'syncbar'} + mock = MagicMock(return_value='Salt:Stack True master/minion \ + UpToDate/partner syncbar None 50 50') + with patch.dict(drbd.__salt__, {'cmd.run': mock}): + self.assertDictEqual(drbd.overview(), ret) + + +if __name__ == '__main__': + from integration import run_tests + run_tests(DrbdTestCase, needs_daemon=False)