salt/tests/unit/modules/smf_test.py

# -*- coding: utf-8 -*-
'''
    :codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
'''
# Import Python libs
from __future__ import absolute_import

# Import Salt Testing Libs
from salttesting import skipIf, TestCase
from salttesting.mock import (
    NO_MOCK,
    NO_MOCK_REASON,
    MagicMock,
    patch)

from salttesting.helpers import ensure_in_syspath

ensure_in_syspath('../../')

# Import Salt Libs
from salt.modules import smf

# Globals
smf.__salt__ = {}


@skipIf(NO_MOCK, NO_MOCK_REASON)
class SmfTestCase(TestCase):
    '''
    Test cases for salt.modules.smf
    '''
    def test_get_running(self):
        '''
        Test to return the running services
        '''
        with patch.dict(smf.__salt__, {'cmd.run':
                                       MagicMock(return_value='A online\n')}):
            self.assertEqual(smf.get_running(), ['A'])

    def test_get_stopped(self):
        '''
        Test to return the stopped services
        '''
        with patch.dict(smf.__salt__,
                        {'cmd.run': MagicMock(return_value='A\n')}):
            self.assertListEqual(smf.get_stopped(), ['A'])

    def test_available(self):
        '''
        Test to returns ``True`` if the specified service is available,
        otherwise returns ``False``.
        '''
        with patch.dict(smf.__salt__, {'cmd.run': MagicMock(return_value='A')}):
            with patch.object(smf, 'get_all', return_value=('A')):
                self.assertTrue(smf.available('A'))

    def test_missing(self):
        '''
        The inverse of service.available.
        Returns ``True`` if the specified service is not available, otherwise
        returns ``False``.
        '''
        with patch.dict(smf.__salt__, {'cmd.run': MagicMock(return_value='A')}):
            with patch.object(smf, 'get_all', return_value=('A')):
                self.assertFalse(smf.missing('A'))

    def test_get_all(self):
        '''
        Test to return all installed services
        '''
        with patch.dict(smf.__salt__,
                        {'cmd.run': MagicMock(return_value='A\n')}):
            self.assertListEqual(smf.get_all(), ['A'])

    def test_start(self):
        '''
        Test to start the specified service
        '''
        with patch.dict(smf.__salt__,
                        {'cmd.retcode': MagicMock(side_effect=[False, 3, None,
                                                               False, 4])}):
            self.assertTrue(smf.start('name'))

            self.assertTrue(smf.start('name'))

            self.assertFalse(smf.start('name'))

    def test_stop(self):
        '''
        Test to stop the specified service
        '''
        with patch.dict(smf.__salt__,
                        {'cmd.retcode': MagicMock(return_value=False)}):
            self.assertTrue(smf.stop('name'))

    def test_restart(self):
        '''
        Test to restart the named service
        '''
        with patch.dict(smf.__salt__,
                        {'cmd.retcode': MagicMock(side_effect=[False, True])}):

            with patch.object(smf, 'start', return_value='A'):
                self.assertEqual(smf.restart('name'), 'A')

            self.assertFalse(smf.restart('name'))

    def test_reload_(self):
        '''
        Test to reload the named service
        '''
        with patch.dict(smf.__salt__,
                        {'cmd.retcode': MagicMock(side_effect=[False, True])}):

            with patch.object(smf, 'start', return_value='A'):
                self.assertEqual(smf.reload_('name'), 'A')

            self.assertFalse(smf.reload_('name'))

    def test_status(self):
        '''
        Test to return the status for a service, returns a bool whether the
        service is running.
        '''
        with patch.dict(smf.__salt__,
                        {'cmd.run': MagicMock(side_effect=['online',
                                                           'online1'])}):
            self.assertTrue(smf.status('name'))

            self.assertFalse(smf.status('name'))

    def test_enable(self):
        '''
        Test to enable the named service to start at boot
        '''
        with patch.dict(smf.__salt__,
                        {'cmd.retcode': MagicMock(return_value=False)}):
            self.assertTrue(smf.enable('name'))

    def test_disable(self):
        '''
        Test to disable the named service to start at boot
        '''
        with patch.dict(smf.__salt__,
                        {'cmd.retcode': MagicMock(return_value=False)}):
            self.assertTrue(smf.disable('name'))

    def test_enabled(self):
        '''
        Test to check to see if the named service is enabled to start on boot
        '''
        with patch.dict(smf.__salt__,
                        {'cmd.run': MagicMock(side_effect=['fmri',
                                                           'A B true',
                                                           'fmri',
                                                           'A B false'])}):
            self.assertTrue(smf.enabled('name'))

            self.assertFalse(smf.enabled('name'))

    def test_disabled(self):
        '''
        Test to check to see if the named service is disabled to start on boot
        '''
        with patch.object(smf, 'enabled', return_value=False):
            self.assertTrue(smf.disabled('name'))

    def test_get_enabled(self):
        '''
        Test to return the enabled services
        '''
        with patch.object(smf, '_get_enabled_disabled', return_value=True):
            self.assertTrue(smf.get_enabled())

    def test_get_disabled(self):
        '''
        Test to return the disabled services
        '''
        with patch.object(smf, '_get_enabled_disabled', return_value=True):
            self.assertTrue(smf.get_disabled())


if __name__ == '__main__':
    from integration import run_tests
    run_tests(SmfTestCase, needs_daemon=False)