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

Added unit tests for new psutil features.

Browse Source
This commit is contained in:
Ted Strzalkowski 2014-02-01 13:22:44 -08:00
parent e4d3d38ed5
commit 488dd52e2c
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/unit/modules/ps_test.py
View File

@ -18,6 +18,8 @@ if HAS_PSUTIL:
import psutil
STUB_CPU_TIMES = psutil._compat.namedtuple('cputimes', 'user nice system idle')(1, 2, 3, 4)
STUB_VIRT_MEM = psutil._compat.namedtuple('vmem', 'total available percent used free')(1000, 500, 50, 500, 500)
STUB_SWAP_MEM = psutil._compat.namedtuple('swap', 'total used free percent sin sout')(1000, 500, 500, 50, 0, 0)
STUB_PHY_MEM_USAGE = psutil._compat.namedtuple('usage', 'total used free percent')(1000, 500, 500, 50)
STUB_DISK_PARTITION = psutil._compat.namedtuple('partition', 'device mountpoint fstype, opts')('/dev/disk0s2', '/',
'hfs',
@ -33,6 +35,8 @@ if HAS_PSUTIL:
0.0)
else:
(STUB_CPU_TIMES,
STUB_VIRT_MEM,
STUB_SWAP_MEM,
STUB_PHY_MEM_USAGE,
STUB_DISK_PARTITION,
STUB_DISK_USAGE,
@ -86,6 +90,14 @@ class PsTestCase(TestCase):
def test_cpu_times(self):
self.assertDictEqual({'idle': 4, 'nice': 2, 'system': 3, 'user': 1}, ps.cpu_times())
@patch('psutil.virtual_memory', new=MagicMock(return_value=STUB_VIRT_MEM))
def test_virtual_memory(self):
self.assertDictEqual({'used': 500, 'total': 1000, 'available': 500, 'percent': 50, 'free': 500}, ps.virtual_memory())
@patch('psutil.swap_memory', new=MagicMock(return_value=STUB_SWAP_MEM))
def test_swap_memory(self):
self.assertDictEqual({'used': 500, 'total': 1000, 'percent': 50, 'free': 500, 'sin': 0, 'sout': 0}, ps.swap_memory())
@patch('psutil.phymem_usage', new=MagicMock(return_value=STUB_PHY_MEM_USAGE))
def test_physical_memory_usage(self):
self.assertDictEqual({'used': 500, 'total': 1000, 'percent': 50, 'free': 500}, ps.physical_memory_usage())