salt/tests/unit/beacons/test_memusage_beacon.py

# 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, Mock, 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, (True, 'Valid beacon configuration'))

    def test_memusage_match(self):
        with patch('psutil.virtual_memory',
                   MagicMock(return_value=STUB_MEMORY_USAGE)):

            config = [{'percent': '40%'}]

            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}])