Merge pull request #31411 from jtand/beacons_tests

Added some beacons execution module integration tests

tests/integration/files/conf/minion

@@ -10,6 +10,7 @@ sock_dir: minion_sock
 open_mode: True
 log_file: minion
 #loop_interval: 0.05
+config_dir: /tmp/salt-tests-tmpdir
 
 # module extension
 test.foo: baz

tests/integration/modules/beacons.py

@@ -0,0 +1,133 @@
+# -*- coding: utf-8 -*-
+'''
+    :codeauthor: :email:`Justin Anderson <janderson@saltstack.com>`
+'''
+
+# Python Libs
+from __future__ import absolute_import
+import os
+
+# Salt Libs
+from salt.modules import beacons
+from salt.exceptions import CommandExecutionError
+import integration
+
+# Salttesting libs
+from salttesting import skipIf
+from salttesting.helpers import destructiveTest, ensure_in_syspath
+
+ensure_in_syspath('../../')
+
+beacons.__opts__ = {}
+
+BEACON_CONF_DIR = os.path.join(integration.TMP, 'minion.d')
+if not os.path.exists(BEACON_CONF_DIR):
+    os.makedirs(BEACON_CONF_DIR)
+
+
+@skipIf(os.geteuid() != 0, 'You must be root to run these tests')
+@destructiveTest
+class BeaconsAddDeleteTest(integration.ModuleCase):
+    '''
+    Tests the add and delete functions
+    '''
+    def test_add_and_delete(self):
+        '''
+        Test adding and deleting a beacon
+        '''
+        _add = self.run_function('beacons.add', ['ps', {'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')
+
+
+@skipIf(os.geteuid() != 0, 'You must be root to run these tests')
+@destructiveTest
+class BeaconsTest(integration.ModuleCase):
+    '''
+    Tests the beacons execution module
+    '''
+    def setUp(self):
+        try:
+            # Add beacon to disable
+            self.run_function('beacons.add', ['ps', {'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)
+        self.assertFalse(_list['ps']['enabled'])
+
+    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'])
+
+        # 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': {'apache2': 'stopped'}, 'enabled': True})
+        else:
+            self.assertEqual(ret, {'ps': {'apache': 'stopped'}})
+
+
+if __name__ == '__main__':
+    from integration import run_tests
+    run_tests([BeaconsAddDeleteTest, BeaconsTest])