mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
# coding: utf-8
|
|
|
|
# Python libs
|
|
from __future__ import absolute_import
|
|
from collections import namedtuple
|
|
|
|
# Salt testing libs
|
|
from tests.support.unit import skipIf, TestCase
|
|
from tests.support.mock import NO_MOCK, NO_MOCK_REASON, patch, MagicMock
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
|
|
# Salt libs
|
|
import salt.beacons.memusage as memusage
|
|
|
|
STUB_MEMORY_USAGE = namedtuple('vmem', 'total available percent used free active inactive buffers cached shared')(
|
|
15722012672, 9329594368, 40.7, 5137018880,
|
|
4678086656, 6991405056, 2078953472,
|
|
1156378624, 4750528512, 898908160)
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class MemUsageBeaconTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
Test case for salt.beacons.memusage
|
|
'''
|
|
|
|
def setup_loader_modules(self):
|
|
return {}
|
|
|
|
def test_non_list_config(self):
|
|
config = {}
|
|
|
|
ret = memusage.validate(config)
|
|
|
|
self.assertEqual(ret, (False, 'Configuration for memusage beacon must'
|
|
' be a list.'))
|
|
|
|
def test_empty_config(self):
|
|
config = [{}]
|
|
|
|
ret = memusage.validate(config)
|
|
|
|
self.assertEqual(ret, (False, 'Configuration for memusage beacon '
|
|
'requires percent.'))
|
|
|
|
def test_memusage_match(self):
|
|
with patch('psutil.virtual_memory',
|
|
MagicMock(return_value=STUB_MEMORY_USAGE)):
|
|
|
|
config = [{'percent': '40%'}, {'interval': 30}]
|
|
|
|
ret = memusage.validate(config)
|
|
|
|
self.assertEqual(ret, (True, 'Valid beacon configuration'))
|
|
|
|
ret = memusage.beacon(config)
|
|
self.assertEqual(ret, [{'memusage': 40.7}])
|
|
|
|
def test_memusage_nomatch(self):
|
|
with patch('psutil.virtual_memory',
|
|
MagicMock(return_value=STUB_MEMORY_USAGE)):
|
|
|
|
config = [{'percent': '70%'}]
|
|
|
|
ret = memusage.validate(config)
|
|
|
|
self.assertEqual(ret, (True, 'Valid beacon configuration'))
|
|
|
|
ret = memusage.beacon(config)
|
|
self.assertNotEqual(ret, [{'memusage': 50}])
|