mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
105 lines
4.8 KiB
Python
105 lines
4.8 KiB
Python
# -*- 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
|
|
from salttesting.helpers import ensure_in_syspath
|
|
from salttesting.mock import MagicMock, patch
|
|
ensure_in_syspath('../../')
|
|
|
|
from salt.modules import disk
|
|
#usage_size = {'filesystem': None,'1K-blocks':10000,'used':10000,'available':10000,'capacity':10000}
|
|
|
|
STUB_DISK_USAGE = {
|
|
'/': {'filesystem': None, '1K-blocks': 10000, 'used': 10000, 'available': 10000, 'capacity': 10000},
|
|
'/dev': {'filesystem': None, '1K-blocks': 10000, 'used': 10000, 'available': 10000, 'capacity': 10000},
|
|
'/run': {'filesystem': None, '1K-blocks': 10000, 'used': 10000, 'available': 10000, 'capacity': 10000},
|
|
'/run/lock': {'filesystem': None, '1K-blocks': 10000, 'used': 10000, 'available': 10000, 'capacity': 10000},
|
|
'/run/shm': {'filesystem': None, '1K-blocks': 10000, 'used': 10000, 'available': 10000, 'capacity': 10000},
|
|
'/run/user': {'filesystem': None, '1K-blocks': 10000, 'used': 10000, 'available': 10000, 'capacity': 10000},
|
|
'/sys/fs/cgroup': {'filesystem': None, '1K-blocks': 10000, 'used': 10000, 'available': 10000, 'capacity': 10000}
|
|
}
|
|
|
|
|
|
STUB_DISK_INODEUSAGE = {
|
|
'/': {'inodes': 10000, 'used': 10000, 'free': 10000, 'use': 10000, 'filesystem': None},
|
|
'/dev': {'inodes': 10000, 'used': 10000, 'free': 10000, 'use': 10000, 'filesystem': None},
|
|
'/run': {'inodes': 10000, 'used': 10000, 'free': 10000, 'use': 10000, 'filesystem': None},
|
|
'/run/lock': {'inodes': 10000, 'used': 10000, 'free': 10000, 'use': 10000, 'filesystem': None},
|
|
'/run/shm': {'inodes': 10000, 'used': 10000, 'free': 10000, 'use': 10000, 'filesystem': None},
|
|
'/run/user': {'inodes': 10000, 'used': 10000, 'free': 10000, 'use': 10000, 'filesystem': None},
|
|
'/sys/fs/cgroup': {'inodes': 10000, 'used': 10000, 'free': 10000, 'use': 10000, 'filesystem': None}
|
|
}
|
|
|
|
STUB_DISK_PERCENT = {
|
|
'/': 50,
|
|
'/dev': 10,
|
|
'/run': 10,
|
|
'/run/lock': 10,
|
|
'/run/shm': 10,
|
|
'/run/user': 10,
|
|
'/sys/fs/cgroup': 10
|
|
}
|
|
|
|
STUB_DISK_BLKID = {'/dev/sda': {'TYPE': 'ext4', 'UUID': None}}
|
|
|
|
disk.__grains__ = {}
|
|
|
|
disk.__salt__ = {} # {'cmd.run': salt.modules.cmdmod._run_quiet}
|
|
|
|
|
|
class DiskTestCase(TestCase):
|
|
'''
|
|
TestCase for salt.modules.disk module
|
|
'''
|
|
|
|
@patch('salt.modules.disk.usage', MagicMock(return_value=STUB_DISK_USAGE))
|
|
def test_usage_dict(self):
|
|
with patch.dict(disk.__grains__, {'kernel': 'Linux'}):
|
|
mock_cmd = MagicMock(return_value=1)
|
|
with patch.dict(disk.__salt__, {'cmd.run': mock_cmd}):
|
|
self.assertDictEqual(STUB_DISK_USAGE, disk.usage(args=None))
|
|
|
|
@patch('salt.modules.disk.usage', MagicMock(return_value=''))
|
|
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))
|
|
|
|
@patch('salt.modules.disk.inodeusage', MagicMock(return_value=STUB_DISK_INODEUSAGE))
|
|
def test_inodeusage(self):
|
|
with patch.dict(disk.__grains__, {'kernel': 'OpenBSD'}):
|
|
mock = MagicMock()
|
|
with patch.dict(disk.__salt__, {'cmd.run': mock}):
|
|
self.assertDictEqual(STUB_DISK_INODEUSAGE, disk.inodeusage(args=None))
|
|
|
|
@patch('salt.modules.disk.percent', MagicMock(return_value=STUB_DISK_PERCENT))
|
|
def test_percent(self):
|
|
with patch.dict(disk.__grains__, {'kernel': 'Linux'}):
|
|
mock = MagicMock()
|
|
with patch.dict(disk.__salt__, {'cmd.run': mock}):
|
|
self.assertDictEqual(STUB_DISK_PERCENT, disk.percent(args=None))
|
|
|
|
@patch('salt.modules.disk.percent', MagicMock(return_value='/'))
|
|
def test_percent_args(self):
|
|
with patch.dict(disk.__grains__, {'kernel': 'Linux'}):
|
|
mock = MagicMock()
|
|
with patch.dict(disk.__salt__, {'cmd.run': mock}):
|
|
self.assertEqual('/', disk.percent('/'))
|
|
|
|
@patch('salt.modules.disk.blkid', MagicMock(return_value=STUB_DISK_BLKID))
|
|
def test_blkid(self):
|
|
with patch.dict(disk.__salt__, {'cmd.run_stdout': MagicMock(return_value=1)}):
|
|
self.assertDictEqual(STUB_DISK_BLKID, disk.blkid())
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from integration import run_tests
|
|
run_tests(DiskTestCase, needs_daemon=False)
|