mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 08:58:59 +00:00
3273bbdab7
Conflicts: - doc/ref/configuration/master.rst - doc/ref/modules/all/index.rst - doc/topics/grains/index.rst - doc/topics/releases/2016.3.4.rst - doc/topics/spm/spm_formula.rst - doc/topics/tutorials/cron.rst - doc/topics/tutorials/index.rst - doc/topics/tutorials/stormpath.rst - salt/engines/slack.py - salt/log/handlers/fluent_mod.py - salt/modules/cyg.py - salt/modules/junos.py - salt/modules/namecheap_dns.py - salt/modules/namecheap_domains.py - salt/modules/namecheap_ns.py - salt/modules/namecheap_ssl.py - salt/modules/namecheap_users.py - salt/modules/reg.py - salt/modules/tomcat.py - salt/modules/vault.py - salt/modules/win_file.py - salt/modules/zpool.py - salt/output/highstate.py - salt/renderers/pass.py - salt/runners/cache.py - salt/states/boto_apigateway.py - salt/states/boto_iam.py - salt/states/boto_route53.py - salt/states/msteams.py - salt/states/reg.py - salt/states/win_iis.py - tests/integration/modules/test_cmdmod.py - tests/integration/states/test_user.py - tests/support/helpers.py - tests/unit/cloud/clouds/test_openstack.py - tests/unit/fileserver/test_gitfs.py - tests/unit/modules/test_junos.py - tests/unit/pillar/test_git.py - tests/unit/states/test_win_path.py - tests/unit/test_pillar.py - tests/unit/utils/test_format_call.py - tests/unit/utils/test_utils.py - tests/unit/utils/test_warnings.py
178 lines
8.0 KiB
Python
178 lines
8.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
|
|
'''
|
|
|
|
# Import Python libs
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
# Import Salt Testing libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import TestCase, skipIf
|
|
from tests.support.mock import MagicMock, patch
|
|
|
|
# Import Salt libs
|
|
import salt.modules.disk as disk
|
|
import salt.utils.path
|
|
|
|
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}}
|
|
|
|
|
|
class DiskTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
TestCase for salt.modules.disk module
|
|
'''
|
|
def setup_loader_modules(self):
|
|
return {disk: {}}
|
|
|
|
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'}), \
|
|
patch('salt.modules.disk.usage', MagicMock(return_value='')):
|
|
mock_cmd = MagicMock(return_value=1)
|
|
with patch.dict(disk.__salt__, {'cmd.run': mock_cmd}):
|
|
self.assertEqual('', disk.usage(args=None))
|
|
|
|
def test_inodeusage(self):
|
|
with patch.dict(disk.__grains__, {'kernel': 'OpenBSD'}), \
|
|
patch('salt.modules.disk.inodeusage',
|
|
MagicMock(return_value=STUB_DISK_INODEUSAGE)):
|
|
mock = MagicMock()
|
|
with patch.dict(disk.__salt__, {'cmd.run': mock}):
|
|
self.assertDictEqual(STUB_DISK_INODEUSAGE, disk.inodeusage(args=None))
|
|
|
|
def test_percent(self):
|
|
with patch.dict(disk.__grains__, {'kernel': 'Linux'}), \
|
|
patch('salt.modules.disk.percent',
|
|
MagicMock(return_value=STUB_DISK_PERCENT)):
|
|
mock = MagicMock()
|
|
with patch.dict(disk.__salt__, {'cmd.run': mock}):
|
|
self.assertDictEqual(STUB_DISK_PERCENT, disk.percent(args=None))
|
|
|
|
def test_percent_args(self):
|
|
with patch.dict(disk.__grains__, {'kernel': 'Linux'}), \
|
|
patch('salt.modules.disk.percent', MagicMock(return_value='/')):
|
|
mock = MagicMock()
|
|
with patch.dict(disk.__salt__, {'cmd.run': mock}):
|
|
self.assertEqual('/', disk.percent('/'))
|
|
|
|
def test_blkid(self):
|
|
with patch.dict(disk.__salt__, {'cmd.run_stdout': MagicMock(return_value=1)}), \
|
|
patch('salt.modules.disk.blkid', MagicMock(return_value=STUB_DISK_BLKID)):
|
|
self.assertDictEqual(STUB_DISK_BLKID, disk.blkid())
|
|
|
|
def test_dump(self):
|
|
mock = MagicMock(return_value={'retcode': 0, 'stdout': ''})
|
|
with patch.dict(disk.__salt__, {'cmd.run_all': mock}):
|
|
disk.dump('/dev/sda')
|
|
mock.assert_called_once_with(
|
|
'blockdev --getro --getsz --getss --getpbsz --getiomin '
|
|
'--getioopt --getalignoff --getmaxsect --getsize '
|
|
'--getsize64 --getra --getfra /dev/sda',
|
|
python_shell=False
|
|
)
|
|
|
|
def test_wipe(self):
|
|
mock = MagicMock(return_value={'retcode': 0, 'stdout': ''})
|
|
with patch.dict(disk.__salt__, {'cmd.run_all': mock}):
|
|
disk.wipe('/dev/sda')
|
|
mock.assert_called_once_with(
|
|
'wipefs -a /dev/sda',
|
|
python_shell=False
|
|
)
|
|
|
|
def test_tune(self):
|
|
mock = MagicMock(return_value='712971264\n512\n512\n512\n0\n0\n88\n712971264\n365041287168\n512\n512')
|
|
with patch.dict(disk.__salt__, {'cmd.run': mock}):
|
|
mock_dump = MagicMock(return_value={'retcode': 0, 'stdout': ''})
|
|
with patch('salt.modules.disk.dump', mock_dump):
|
|
kwargs = {'read-ahead': 512, 'filesystem-read-ahead': 1024}
|
|
disk.tune('/dev/sda', **kwargs)
|
|
|
|
self.assert_called_once(mock)
|
|
|
|
args, kwargs = mock.call_args
|
|
|
|
# Assert called once with either 'blockdev --setra 512 --setfra 512 /dev/sda' or
|
|
# 'blockdev --setfra 512 --setra 512 /dev/sda' and python_shell=False kwarg.
|
|
self.assertEqual(len(args), 1)
|
|
self.assertTrue(args[0].startswith('blockdev '))
|
|
self.assertTrue(args[0].endswith(' /dev/sda'))
|
|
self.assertIn(' --setra 512 ', args[0])
|
|
self.assertIn(' --setfra 1024 ', args[0])
|
|
self.assertEqual(len(args[0].split()), 6)
|
|
self.assertEqual(kwargs, {'python_shell': False})
|
|
|
|
def test_format(self):
|
|
'''
|
|
unit tests for disk.format
|
|
'''
|
|
device = '/dev/sdX1'
|
|
mock = MagicMock(return_value=0)
|
|
with patch.dict(disk.__salt__, {'cmd.retcode': mock}),\
|
|
patch('salt.utils.path.which', MagicMock(return_value=True)):
|
|
self.assertEqual(disk.format_(device), True)
|
|
|
|
@skipIf(not salt.utils.path.which('lsblk') and not salt.utils.path.which('df'),
|
|
'lsblk or df not found')
|
|
def test_fstype(self):
|
|
'''
|
|
unit tests for disk.fstype
|
|
'''
|
|
device = '/dev/sdX1'
|
|
fs_type = 'ext4'
|
|
mock = MagicMock(return_value='FSTYPE\n{0}'.format(fs_type))
|
|
with patch.dict(disk.__grains__, {'kernel': 'Linux'}), \
|
|
patch.dict(disk.__salt__, {'cmd.run': mock}), \
|
|
patch('salt.utils.path.which', MagicMock(return_value=True)):
|
|
self.assertEqual(disk.fstype(device), fs_type)
|
|
|
|
def test_resize2fs(self):
|
|
'''
|
|
unit tests for disk.resize2fs
|
|
'''
|
|
device = '/dev/sdX1'
|
|
mock = MagicMock()
|
|
with patch.dict(disk.__salt__, {'cmd.run_all': mock}), \
|
|
patch('salt.utils.path.which', MagicMock(return_value=True)):
|
|
disk.resize2fs(device)
|
|
mock.assert_called_once_with('resize2fs {0}'.format(device), python_shell=False)
|