salt/tests/integration/states/test_beacon.py

# -*- coding: utf-8 -*-
'''
Integration tests for the beacon states
'''
from __future__ import absolute_import, print_function, unicode_literals

# Import Salt Testing Libs
from tests.support.case import ModuleCase
from tests.support.mixins import SaltReturnAssertsMixin


class BeaconStateTestCase(ModuleCase, SaltReturnAssertsMixin):
    '''
    Test beacon states
    '''
    def setUp(self):
        '''
        '''
        self.run_function('beacons.reset', f_timeout=300)
        self.wait_for_all_jobs()

    def tearDown(self):
        self.run_function('beacons.reset', f_timeout=300)

    def test_present_absent(self):
        kwargs = {'/': '38%', 'interval': 5}
        ret = self.run_state(
            'beacon.present',
            name='diskusage',
            f_timeout=300,
            **kwargs
        )
        self.assertSaltTrueReturn(ret)

        ret = self.run_function('beacons.list',
                                return_yaml=False,
                                f_timeout=300)
        self.assertTrue('diskusage' in ret)
        self.assertTrue({'interval': 5} in ret['diskusage'])
        self.assertTrue({'/': '38%'} in ret['diskusage'])

        ret = self.run_state(
            'beacon.absent',
            name='diskusage',
            f_timeout=300
        )
        self.assertSaltTrueReturn(ret)

        ret = self.run_function('beacons.list',
                                return_yaml=False,
                                f_timeout=300)
        self.assertEqual(ret, {'beacons': {}})