mirror of
https://github.com/saltstack/salt.git
synced 2025-04-17 10:10:20 +00:00
Merge pull request #23532 from jayeshka/ipmi-states-unit-test
adding states/ipmi unit test case
This commit is contained in:
Justin Findlay
commit
e5421139d3
1 changed files with 165 additions and 0 deletions
165
tests/unit/states/ipmi_test.py
Normal file
165
tests/unit/states/ipmi_test.py
Normal file
|
|
@ -0,0 +1,165 @@
|
|||
# -*- 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 ipmi
|
||||
|
||||
ipmi.__salt__ = {}
|
||||
ipmi.__opts__ = {}
|
||||
|
||||
|
||||
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
||||
class IpmiTestCase(TestCase):
|
||||
'''
|
||||
Test cases for salt.states.ipmi
|
||||
'''
|
||||
# 'boot_device' function tests: 1
|
||||
|
||||
def test_boot_device(self):
|
||||
'''
|
||||
Test to request power state change.
|
||||
'''
|
||||
name = 'salt'
|
||||
|
||||
ret = {'name': name,
|
||||
'result': True,
|
||||
'comment': '',
|
||||
'changes': {}}
|
||||
|
||||
mock = MagicMock(return_value=name)
|
||||
with patch.dict(ipmi.__salt__, {'ipmi.get_bootdev': mock,
|
||||
'ipmi.set_bootdev': mock}):
|
||||
comt = ('system already in this state')
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(ipmi.boot_device(name), ret)
|
||||
|
||||
with patch.dict(ipmi.__opts__, {'test': False}):
|
||||
comt = ('changed boot device')
|
||||
ret.update({'name': 'default', 'comment': comt, 'result': True,
|
||||
'changes': {'new': 'default', 'old': 'salt'}})
|
||||
self.assertDictEqual(ipmi.boot_device(), ret)
|
||||
|
||||
with patch.dict(ipmi.__opts__, {'test': True}):
|
||||
comt = ('would change boot device')
|
||||
ret.update({'comment': comt, 'result': None})
|
||||
self.assertDictEqual(ipmi.boot_device(), ret)
|
||||
|
||||
# 'power' function tests: 1
|
||||
|
||||
def test_power(self):
|
||||
'''
|
||||
Test to request power state change
|
||||
'''
|
||||
ret = {'name': 'power_on',
|
||||
'result': True,
|
||||
'comment': '',
|
||||
'changes': {}}
|
||||
|
||||
mock = MagicMock(return_value='on')
|
||||
with patch.dict(ipmi.__salt__, {'ipmi.get_power': mock,
|
||||
'ipmi.set_power': mock}):
|
||||
comt = ('system already in this state')
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(ipmi.power(), ret)
|
||||
|
||||
with patch.dict(ipmi.__opts__, {'test': False}):
|
||||
comt = ('changed system power')
|
||||
ret.update({'name': 'off', 'comment': comt, 'result': True,
|
||||
'changes': {'new': 'off', 'old': 'on'}})
|
||||
self.assertDictEqual(ipmi.power('off'), ret)
|
||||
|
||||
with patch.dict(ipmi.__opts__, {'test': True}):
|
||||
comt = ('would power: off system')
|
||||
ret.update({'comment': comt, 'result': None})
|
||||
self.assertDictEqual(ipmi.power('off'), ret)
|
||||
|
||||
# 'user_present' function tests: 1
|
||||
|
||||
def test_user_present(self):
|
||||
'''
|
||||
Test to ensure IPMI user and user privileges.
|
||||
'''
|
||||
name = 'salt'
|
||||
|
||||
ret = {'name': name,
|
||||
'result': True,
|
||||
'comment': '',
|
||||
'changes': {}}
|
||||
|
||||
mock_ret = {'access': {'callback': False, 'link_auth': True,
|
||||
'ipmi_msg': True,
|
||||
'privilege_level': 'administrator'}}
|
||||
mock = MagicMock(return_value=mock_ret)
|
||||
mock_bool = MagicMock(side_effect=[True, False, False, False])
|
||||
with patch.dict(ipmi.__salt__, {'ipmi.get_user': mock,
|
||||
'ipmi.set_user_password': mock_bool,
|
||||
'ipmi.ensure_user': mock_bool}):
|
||||
comt = ('user already present')
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(ipmi.user_present(name, 5, 'salt@123'), ret)
|
||||
|
||||
with patch.dict(ipmi.__opts__, {'test': True}):
|
||||
comt = ('would (re)create user')
|
||||
ret.update({'comment': comt, 'result': None,
|
||||
'changes': {'new': 'salt', 'old': mock_ret}})
|
||||
self.assertDictEqual(ipmi.user_present(name, 5, 'pw@123'), ret)
|
||||
|
||||
with patch.dict(ipmi.__opts__, {'test': False}):
|
||||
comt = ('(re)created user')
|
||||
ret.update({'comment': comt, 'result': True,
|
||||
'changes': {'new': mock_ret, 'old': mock_ret}})
|
||||
self.assertDictEqual(ipmi.user_present(name, 5, 'pw@123'), ret)
|
||||
|
||||
# 'user_absent' function tests: 1
|
||||
|
||||
def test_user_absent(self):
|
||||
'''
|
||||
Test to delete all user (uid) records having the matching name.
|
||||
'''
|
||||
name = 'salt'
|
||||
|
||||
ret = {'name': name,
|
||||
'result': True,
|
||||
'comment': '',
|
||||
'changes': {}}
|
||||
|
||||
mock = MagicMock(side_effect=[[], [5], [5]])
|
||||
mock_bool = MagicMock(return_value=True)
|
||||
with patch.dict(ipmi.__salt__, {'ipmi.get_name_uids': mock,
|
||||
'ipmi.delete_user': mock_bool}):
|
||||
comt = ('user already absent')
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(ipmi.user_absent(name), ret)
|
||||
|
||||
with patch.dict(ipmi.__opts__, {'test': True}):
|
||||
comt = ('would delete user(s)')
|
||||
ret.update({'comment': comt, 'result': None,
|
||||
'changes': {'delete': [5]}})
|
||||
self.assertDictEqual(ipmi.user_absent(name), ret)
|
||||
|
||||
with patch.dict(ipmi.__opts__, {'test': False}):
|
||||
comt = ('user(s) removed')
|
||||
ret.update({'comment': comt, 'result': False,
|
||||
'changes': {'new': 'None', 'old': [5]}})
|
||||
self.assertDictEqual(ipmi.user_absent(name), ret)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
from integration import run_tests
|
||||
run_tests(IpmiTestCase, needs_daemon=False)
|
||||
Loading…
Add table
Reference in a new issue