salt/tests/unit/states/chef_test.py
2015-04-27 17:08:56 +05:30

78 lines
2.0 KiB
Python

# -*- 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.states import chef
chef.__salt__ = {}
chef.__opts__ = {}
@skipIf(NO_MOCK, NO_MOCK_REASON)
class ChefTestCase(TestCase):
'''
Test cases for salt.states.chef
'''
# 'client' function tests: 1
def test_client(self):
'''
Test to run chef-client
'''
name = 'my-chef-run'
ret = {'name': name,
'result': False,
'changes': {},
'comment': ''}
mock = MagicMock(return_value={'retcode': 1, 'stdout': '',
'stderr': 'error'})
with patch.dict(chef.__salt__, {'chef.client': mock}):
with patch.dict(chef.__opts__, {'test': True}):
comt = ('\nerror')
ret.update({'comment': comt})
self.assertDictEqual(chef.client(name), ret)
# 'solo' function tests: 1
def test_solo(self):
'''
Test to run chef-solo
'''
name = 'my-chef-run'
ret = {'name': name,
'result': False,
'changes': {},
'comment': ''}
mock = MagicMock(return_value={'retcode': 1, 'stdout': '',
'stderr': 'error'})
with patch.dict(chef.__salt__, {'chef.solo': mock}):
with patch.dict(chef.__opts__, {'test': True}):
comt = ('\nerror')
ret.update({'comment': comt})
self.assertDictEqual(chef.solo(name), ret)
if __name__ == '__main__':
from integration import run_tests
run_tests(ChefTestCase, needs_daemon=False)