valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-08 01:18:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'disk_unit_tests' of https://github.com/jayeshka/salt into jayeshka-disk_unit_tests

Browse Source
This commit is contained in:
Thomas S Hatch 2014-12-18 10:11:01 -07:00
commit 6afb8b64d6
1 changed files with 106 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

106
tests/unit/modules/disk_test.py Normal file
View File

@ -0,0 +1,106 @@
# -*- coding: utf-8 -*-
'''
:codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
'''
# Import Salt Testing libs
from salttesting import TestCase, skipIf
from salttesting.helpers import ensure_in_syspath
from salttesting.mock import NO_MOCK, NO_MOCK_REASON, MagicMock, patch, call, Mock
import psutil
from defer import return_value
ensure_in_syspath('../../')
import salt.modules.cmdmod
from salt.modules import disk
import salt.config
#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(args=None))
if __name__ == '__main__':
from integration import run_tests
run_tests(DiskTestCase, needs_daemon=False)