def test_usage_dict(self):
with patch.dict(disk.__grains__, {"kernel": "Linux"}), patch(
"salt.modules.disk.usage",
MagicMock(return_value=STUB_DISK_USAGE)):
mock_cmd = MagicMock(return_value=1)
with patch.dict(disk.__salt__, {"cmd.run": mock_cmd}):
self.assertDictEqual(STUB_DISK_USAGE, disk.usage(args=None))
def test_usage_dict(self):
with patch.dict(disk.__grains__, {'kernel': 'Linux'}), \
patch('salt.modules.disk.usage',
MagicMock(return_value=STUB_DISK_USAGE)):
mock_cmd = MagicMock(return_value=1)
with patch.dict(disk.__salt__, {'cmd.run': mock_cmd}):
self.assertDictEqual(STUB_DISK_USAGE, disk.usage(args=None))
def test_usage_none(self):
with patch.dict(disk.__grains__, {'kernel': 'Linux'}):
mock_cmd = MagicMock(return_value=1)
with patch.dict(disk.__salt__, {'cmd.run': mock_cmd}):
self.assertEqual('', disk.usage(args=None))
def test_usage_none(self):
with patch.dict(disk.__grains__, {'kernel': 'Linux'}):
mock_cmd = MagicMock(return_value=1)
with patch.dict(disk.__salt__, {'cmd.run': mock_cmd}):
self.assertEqual('', disk.usage(args=None))
