salt/tests/unit/modules/win_autoruns_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 TestCase, skipIf
from salttesting.mock import (
    MagicMock,
    patch
)

from salttesting.helpers import ensure_in_syspath

ensure_in_syspath('../../')

# Import Salt Libs
from salt.modules import win_autoruns

# Globals
win_autoruns.__salt__ = {}
win_autoruns.__grains__ = {}

# Make sure this module runs on Windows system
IS_WIN = win_autoruns.__virtual__()

KEY = ['HKLM\\Software\\Microsoft\\Windows\\CurrentVersion\\Run',
       'HKLM\\Software\\Microsoft\\Windows\\CurrentVersion\\Run /reg:64',
       'HKCU\\Software\\Microsoft\\Windows\\CurrentVersion\\Run']


@skipIf(not IS_WIN, "This test case runs only on Windows system")
class WinAutorunsTestCase(TestCase):
    '''
    Test cases for salt.modules.win_autoruns
    '''
    # 'list_' function tests: 1

    @patch('os.listdir', MagicMock(return_value=[]))
    def test_list(self):
        '''
        Test if it enables win_autoruns the service on the server
        '''
        ret = {KEY[0]: ['SALT'], KEY[1]: ['SALT'], KEY[2]: ['SALT']}
        mock = MagicMock(return_value='Windows 7')
        with patch.dict(win_autoruns.__grains__, {'osfullname': mock}):
            mock = MagicMock(return_value='SALT')
            with patch.dict(win_autoruns.__salt__, {'cmd.run': mock}):
                self.assertDictEqual(win_autoruns.list_(), ret)


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