mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
3273bbdab7
Conflicts: - doc/ref/configuration/master.rst - doc/ref/modules/all/index.rst - doc/topics/grains/index.rst - doc/topics/releases/2016.3.4.rst - doc/topics/spm/spm_formula.rst - doc/topics/tutorials/cron.rst - doc/topics/tutorials/index.rst - doc/topics/tutorials/stormpath.rst - salt/engines/slack.py - salt/log/handlers/fluent_mod.py - salt/modules/cyg.py - salt/modules/junos.py - salt/modules/namecheap_dns.py - salt/modules/namecheap_domains.py - salt/modules/namecheap_ns.py - salt/modules/namecheap_ssl.py - salt/modules/namecheap_users.py - salt/modules/reg.py - salt/modules/tomcat.py - salt/modules/vault.py - salt/modules/win_file.py - salt/modules/zpool.py - salt/output/highstate.py - salt/renderers/pass.py - salt/runners/cache.py - salt/states/boto_apigateway.py - salt/states/boto_iam.py - salt/states/boto_route53.py - salt/states/msteams.py - salt/states/reg.py - salt/states/win_iis.py - tests/integration/modules/test_cmdmod.py - tests/integration/states/test_user.py - tests/support/helpers.py - tests/unit/cloud/clouds/test_openstack.py - tests/unit/fileserver/test_gitfs.py - tests/unit/modules/test_junos.py - tests/unit/pillar/test_git.py - tests/unit/states/test_win_path.py - tests/unit/test_pillar.py - tests/unit/utils/test_format_call.py - tests/unit/utils/test_utils.py - tests/unit/utils/test_warnings.py
149 lines
4.9 KiB
Python
149 lines
4.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: Justin Anderson <janderson@saltstack.com>
|
|
'''
|
|
|
|
# Python Libs
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
import os
|
|
|
|
# Salt Libs
|
|
from salt.exceptions import CommandExecutionError
|
|
|
|
# Salttesting libs
|
|
from tests.support.case import ModuleCase
|
|
from tests.support.unit import skipIf
|
|
|
|
|
|
class BeaconsAddDeleteTest(ModuleCase):
|
|
'''
|
|
Tests the add and delete functions
|
|
'''
|
|
def setUp(self):
|
|
self.minion_conf_d_dir = os.path.join(
|
|
self.minion_opts['config_dir'],
|
|
os.path.dirname(self.minion_opts['default_include']))
|
|
if not os.path.isdir(self.minion_conf_d_dir):
|
|
os.makedirs(self.minion_conf_d_dir)
|
|
self.beacons_config_file_path = os.path.join(self.minion_conf_d_dir, 'beacons.conf')
|
|
|
|
def tearDown(self):
|
|
if os.path.isfile(self.beacons_config_file_path):
|
|
os.unlink(self.beacons_config_file_path)
|
|
|
|
def test_add_and_delete(self):
|
|
'''
|
|
Test adding and deleting a beacon
|
|
'''
|
|
_add = self.run_function('beacons.add', ['ps', [{'processes': {'apache2': 'stopped'}}]])
|
|
self.assertTrue(_add['result'])
|
|
|
|
# save added beacon
|
|
_save = self.run_function('beacons.save')
|
|
self.assertTrue(_save['result'])
|
|
|
|
# delete the beacon
|
|
_delete = self.run_function('beacons.delete', ['ps'])
|
|
self.assertTrue(_delete['result'])
|
|
|
|
# save the results
|
|
self.run_function('beacons.save')
|
|
|
|
|
|
class BeaconsTest(ModuleCase):
|
|
'''
|
|
Tests the beacons execution module
|
|
'''
|
|
beacons_config_file_path = minion_conf_d_dir = None
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
if cls.beacons_config_file_path and os.path.isfile(cls.beacons_config_file_path):
|
|
os.unlink(cls.beacons_config_file_path)
|
|
|
|
def setUp(self):
|
|
if self.minion_conf_d_dir is None:
|
|
self.minion_conf_d_dir = os.path.join(
|
|
self.minion_opts['config_dir'],
|
|
os.path.dirname(self.minion_opts['default_include']))
|
|
if not os.path.isdir(self.minion_conf_d_dir):
|
|
os.makedirs(self.minion_conf_d_dir)
|
|
self.__class__.beacons_config_file_path = os.path.join(self.minion_conf_d_dir, 'beacons.conf')
|
|
try:
|
|
# Add beacon to disable
|
|
self.run_function('beacons.add', ['ps', [{'processes': {'apache2': 'stopped'}}]])
|
|
self.run_function('beacons.save')
|
|
except CommandExecutionError:
|
|
self.skipTest('Unable to add beacon')
|
|
|
|
def tearDown(self):
|
|
# delete added beacon
|
|
self.run_function('beacons.delete', ['ps'])
|
|
self.run_function('beacons.save')
|
|
|
|
def test_disable(self):
|
|
'''
|
|
Test disabling beacons
|
|
'''
|
|
# assert beacon exists
|
|
_list = self.run_function('beacons.list', return_yaml=False)
|
|
self.assertIn('ps', _list)
|
|
|
|
ret = self.run_function('beacons.disable')
|
|
self.assertTrue(ret['result'])
|
|
|
|
# assert beacons are disabled
|
|
_list = self.run_function('beacons.list', return_yaml=False)
|
|
self.assertFalse(_list['enabled'])
|
|
|
|
# disable added beacon
|
|
ret = self.run_function('beacons.disable_beacon', ['ps'])
|
|
self.assertTrue(ret['result'])
|
|
|
|
# assert beacon ps is disabled
|
|
_list = self.run_function('beacons.list', return_yaml=False)
|
|
for bdict in _list['ps']:
|
|
if 'enabled' in bdict:
|
|
self.assertFalse(bdict['enabled'])
|
|
break
|
|
|
|
def test_enable(self):
|
|
'''
|
|
Test enabling beacons
|
|
'''
|
|
# assert beacon exists
|
|
_list = self.run_function('beacons.list', return_yaml=False)
|
|
self.assertIn('ps', _list)
|
|
|
|
# enable beacons on minion
|
|
ret = self.run_function('beacons.enable')
|
|
self.assertTrue(ret['result'])
|
|
|
|
# assert beacons are enabled
|
|
_list = self.run_function('beacons.list', return_yaml=False)
|
|
self.assertTrue(_list['enabled'])
|
|
|
|
@skipIf(True, 'Skip until https://github.com/saltstack/salt/issues/31516 problems are resolved.')
|
|
def test_enabled_beacons(self):
|
|
'''
|
|
Test enabled specific beacon
|
|
'''
|
|
# enable added beacon
|
|
ret = self.run_function('beacons.enable_beacon', ['ps'])
|
|
self.assertTrue(ret['result'])
|
|
|
|
# assert beacon ps is enabled
|
|
_list = self.run_function('beacons.list', return_yaml=False)
|
|
self.assertTrue(_list['ps']['enabled'])
|
|
|
|
def test_list(self):
|
|
'''
|
|
Test lising the beacons
|
|
'''
|
|
# list beacons
|
|
ret = self.run_function('beacons.list', return_yaml=False)
|
|
if 'enabled' in ret:
|
|
self.assertEqual(ret, {'ps': [{'processes': {'apache2': 'stopped'}}], 'enabled': True})
|
|
else:
|
|
self.assertEqual(ret, {'ps': [{'processes': {'apache2': 'stopped'}}]})
|