salt/tests/integration/modules/test_decorators.py

# -*- coding: utf-8 -*-

# Import Python libs
from __future__ import absolute_import, print_function, unicode_literals

# Import Salt Testing libs
from tests.support.case import ModuleCase


class DecoratorTest(ModuleCase):
    def test_module(self):
        self.assertTrue(
                self.run_function(
                    'runtests_decorators.working_function'
                    )
                )

    def test_depends(self):
        ret = self.run_function('runtests_decorators.depends')
        self.assertTrue(isinstance(ret, dict))
        self.assertTrue(ret['ret'])
        self.assertTrue(isinstance(ret['time'], float))

    def test_missing_depends(self):
        self.assertEqual(
                {'runtests_decorators.missing_depends_will_fallback': None,
                 'runtests_decorators.missing_depends': "'runtests_decorators.missing_depends' is not available."},
                self.run_function('runtests_decorators.missing_depends'
                    )
                )

    def test_bool_depends(self):
        # test True
        self.assertTrue(
                self.run_function(
                    'runtests_decorators.booldependsTrue'
                    )
                )

        # test False
        self.assertIn(
                'is not available',
                self.run_function('runtests_decorators.booldependsFalse'
                    )
                )

    def test_depends_will_not_fallback(self):
        ret = self.run_function('runtests_decorators.depends_will_not_fallback')
        self.assertTrue(isinstance(ret, dict))
        self.assertTrue(ret['ret'])
        self.assertTrue(isinstance(ret['time'], float))

    def test_missing_depends_will_fallback(self):
        self.assertListEqual(
                [False, 'fallback'],
                self.run_function(
                    'runtests_decorators.missing_depends_will_fallback'
                    )
                )

    def test_command_success_retcode(self):
        ret = self.run_function('runtests_decorators.command_success_retcode')
        self.assertIs(ret, True)

    def test_command_failure_retcode(self):
        ret = self.run_function('runtests_decorators.command_failure_retcode')
        self.assertEqual(
            ret,
            "'runtests_decorators.command_failure_retcode' is not available."
        )

    def test_command_success_nonzero_retcode_true(self):
        ret = self.run_function('runtests_decorators.command_success_nonzero_retcode_true')
        self.assertIs(ret, True)

    def test_command_failure_nonzero_retcode_true(self):
        ret = self.run_function('runtests_decorators.command_failure_nonzero_retcode_true')
        self.assertEqual(
            ret,
            "'runtests_decorators.command_failure_nonzero_retcode_true' is not available."
        )

    def test_command_success_nonzero_retcode_false(self):
        ret = self.run_function('runtests_decorators.command_success_nonzero_retcode_false')
        self.assertIs(ret, True)

    def test_command_failure_nonzero_retcode_false(self):
        ret = self.run_function('runtests_decorators.command_failure_nonzero_retcode_false')
        self.assertEqual(
            ret,
            "'runtests_decorators.command_failure_nonzero_retcode_false' is not available."
        )