mirror of
https://github.com/saltstack/salt.git
synced 2025-04-17 10:10:20 +00:00
adding states/lxc unit test case
This commit is contained in:
Jayesh Kariya
parent
8837d0038e
commit
32a0d03093
1 changed files with 325 additions and 0 deletions
325
tests/unit/states/lxc_test.py
Normal file
325
tests/unit/states/lxc_test.py
Normal file
|
|
@ -0,0 +1,325 @@
|
|||
# -*- 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 lxc
|
||||
import salt.utils
|
||||
|
||||
lxc.__salt__ = {}
|
||||
lxc.__opts__ = {}
|
||||
|
||||
|
||||
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
||||
class LxcTestCase(TestCase):
|
||||
'''
|
||||
Test cases for salt.states.lxc
|
||||
'''
|
||||
# 'present' function tests: 1
|
||||
|
||||
def test_present(self):
|
||||
'''
|
||||
Test to verify the named container if it exist.
|
||||
'''
|
||||
name = 'web01'
|
||||
|
||||
ret = {'name': name,
|
||||
'result': False,
|
||||
'comment': '',
|
||||
'changes': {}}
|
||||
|
||||
mock = MagicMock(side_effect=[False, True, True, True, True, True,
|
||||
True])
|
||||
mock_t = MagicMock(side_effect=[None, True, 'frozen', 'frozen',
|
||||
'stopped', 'running', 'running'])
|
||||
with patch.dict(lxc.__salt__, {'lxc.exists': mock,
|
||||
'lxc.state': mock_t}):
|
||||
comt = ("Clone source 'True' does not exist")
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(lxc.present(name, clone_from=True), ret)
|
||||
|
||||
with patch.dict(lxc.__opts__, {'test': True}):
|
||||
comt = ("Container 'web01' will be cloned from True")
|
||||
ret.update({'comment': comt, 'result': None})
|
||||
self.assertDictEqual(lxc.present(name, clone_from=True), ret)
|
||||
|
||||
comt = ("Container 'web01' already exists")
|
||||
ret.update({'comment': comt, 'result': True})
|
||||
self.assertDictEqual(lxc.present(name, clone_from=True), ret)
|
||||
|
||||
comt = ("Container 'web01' would be unfrozen")
|
||||
ret.update({'comment': comt, 'result': None})
|
||||
self.assertDictEqual(lxc.present(name, running=True,
|
||||
clone_from=True), ret)
|
||||
|
||||
comt = ('Container \'{0}\' would be stopped'.format(name))
|
||||
ret.update({'comment': comt, 'result': None})
|
||||
self.assertDictEqual(lxc.present(name, running=False,
|
||||
clone_from=True), ret)
|
||||
|
||||
comt = ("Container 'web01' already exists and is stopped")
|
||||
ret.update({'comment': comt, 'result': True})
|
||||
self.assertDictEqual(lxc.present(name, running=False,
|
||||
clone_from=True), ret)
|
||||
|
||||
with patch.dict(lxc.__opts__, {'test': False}):
|
||||
comt = ("Container 'web01' already exists")
|
||||
ret.update({'comment': comt, 'result': True})
|
||||
self.assertDictEqual(lxc.present(name, clone_from=True), ret)
|
||||
|
||||
# 'absent' function tests: 1
|
||||
|
||||
def test_absent(self):
|
||||
'''
|
||||
Test to ensure a container is not present, destroying it if present.
|
||||
'''
|
||||
name = 'web01'
|
||||
|
||||
ret = {'name': name,
|
||||
'result': True,
|
||||
'comment': '',
|
||||
'changes': {}}
|
||||
|
||||
mock = MagicMock(side_effect=[False, True, True])
|
||||
mock_des = MagicMock(return_value={'state': True})
|
||||
with patch.dict(lxc.__salt__, {'lxc.exists': mock,
|
||||
'lxc.destroy': mock_des}):
|
||||
comt = ('Container \'{0}\' does not exist'.format(name))
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(lxc.absent(name), ret)
|
||||
|
||||
with patch.dict(lxc.__opts__, {'test': True}):
|
||||
comt = ('Container \'{0}\' would be destroyed'.format(name))
|
||||
ret.update({'comment': comt, 'result': None})
|
||||
self.assertDictEqual(lxc.absent(name), ret)
|
||||
|
||||
with patch.dict(lxc.__opts__, {'test': False}):
|
||||
comt = ('Container \'{0}\' was destroyed'.format(name))
|
||||
ret.update({'comment': comt, 'result': True,
|
||||
'changes': {'state': True}})
|
||||
self.assertDictEqual(lxc.absent(name), ret)
|
||||
|
||||
# 'running' function tests: 1
|
||||
|
||||
def test_running(self):
|
||||
'''
|
||||
Test to ensure that a container is running.
|
||||
'''
|
||||
name = 'web01'
|
||||
|
||||
ret = {'name': name,
|
||||
'result': False,
|
||||
'comment': '',
|
||||
'changes': {}}
|
||||
|
||||
mock = MagicMock(return_value={'state': {'new': 'stop'}})
|
||||
mock_t = MagicMock(side_effect=[None, 'running', 'stopped', 'start'])
|
||||
with patch.dict(lxc.__salt__, {'lxc.exists': mock,
|
||||
'lxc.state': mock_t,
|
||||
'lxc.start': mock}):
|
||||
comt = ('Container \'{0}\' does not exist'.format(name))
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(lxc.running(name), ret)
|
||||
|
||||
comt = ("Container 'web01' is already running")
|
||||
ret.update({'comment': comt, 'result': True})
|
||||
self.assertDictEqual(lxc.running(name), ret)
|
||||
|
||||
with patch.dict(lxc.__opts__, {'test': True}):
|
||||
comt = ("Container 'web01' would be started")
|
||||
ret.update({'comment': comt, 'result': None})
|
||||
self.assertDictEqual(lxc.running(name), ret)
|
||||
|
||||
with patch.dict(lxc.__opts__, {'test': False}):
|
||||
comt = ("Unable to start container 'web01'")
|
||||
ret.update({'comment': comt, 'result': False, 'changes':
|
||||
{'state': {'new': 'stop', 'old': 'start'}}})
|
||||
self.assertDictEqual(lxc.running(name), ret)
|
||||
|
||||
# 'frozen' function tests: 1
|
||||
|
||||
def test_frozen(self):
|
||||
'''
|
||||
Test to ensure that a container is frozen.
|
||||
'''
|
||||
name = 'web01'
|
||||
|
||||
ret = {'name': name,
|
||||
'result': True,
|
||||
'comment': '',
|
||||
'changes': {}}
|
||||
|
||||
mock = MagicMock(return_value={'state': {'new': 'stop'}})
|
||||
mock_t = MagicMock(side_effect=['frozen', 'stopped', 'stopped'])
|
||||
with patch.dict(lxc.__salt__, {'lxc.freeze': mock,
|
||||
'lxc.state': mock_t}):
|
||||
comt = ('Container \'{0}\' is already frozen'.format(name))
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(lxc.frozen(name), ret)
|
||||
|
||||
with patch.dict(lxc.__opts__, {'test': True}):
|
||||
comt = ("Container 'web01' would be started and frozen")
|
||||
ret.update({'comment': comt, 'result': None})
|
||||
self.assertDictEqual(lxc.frozen(name), ret)
|
||||
|
||||
with patch.dict(lxc.__opts__, {'test': False}):
|
||||
comt = ("Unable to start and freeze container 'web01'")
|
||||
ret.update({'comment': comt, 'result': False, 'changes':
|
||||
{'state': {'new': 'stop', 'old': 'stopped'}}})
|
||||
self.assertDictEqual(lxc.frozen(name), ret)
|
||||
|
||||
# 'stopped' function tests: 1
|
||||
|
||||
def test_stopped(self):
|
||||
'''
|
||||
Test to ensure that a container is stopped.
|
||||
'''
|
||||
name = 'web01'
|
||||
|
||||
ret = {'name': name,
|
||||
'result': False,
|
||||
'comment': '',
|
||||
'changes': {}}
|
||||
|
||||
mock = MagicMock(return_value={'state': {'new': 'stop'}})
|
||||
mock_t = MagicMock(side_effect=[None, 'stopped', 'frozen', 'frozen'])
|
||||
with patch.dict(lxc.__salt__, {'lxc.stop': mock,
|
||||
'lxc.state': mock_t}):
|
||||
comt = ('Container \'{0}\' does not exist'.format(name))
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(lxc.stopped(name), ret)
|
||||
|
||||
comt = ('Container \'{0}\' is already stopped'.format(name))
|
||||
ret.update({'comment': comt, 'result': True})
|
||||
self.assertDictEqual(lxc.stopped(name), ret)
|
||||
|
||||
with patch.dict(lxc.__opts__, {'test': True}):
|
||||
comt = ("Container 'web01' would be stopped")
|
||||
ret.update({'comment': comt, 'result': None})
|
||||
self.assertDictEqual(lxc.stopped(name), ret)
|
||||
|
||||
with patch.dict(lxc.__opts__, {'test': False}):
|
||||
comt = ("Unable to stop container 'web01'")
|
||||
ret.update({'comment': comt, 'result': False, 'changes':
|
||||
{'state': {'new': 'stop', 'old': 'frozen'}}})
|
||||
self.assertDictEqual(lxc.stopped(name), ret)
|
||||
|
||||
# 'created' function tests: 1
|
||||
|
||||
def test_created(self):
|
||||
'''
|
||||
Test to execute create func.
|
||||
'''
|
||||
name = 'web01'
|
||||
|
||||
ret = {'name': name,
|
||||
'result': False,
|
||||
'comment': '',
|
||||
'changes': {}}
|
||||
|
||||
mock = MagicMock(return_value=False)
|
||||
with patch.dict(lxc.__salt__, {'lxc.exists': mock}):
|
||||
with patch.object(salt.utils, 'warn_until', MagicMock()):
|
||||
comt = ("Clone source 'True' does not exist")
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(lxc.created(name, clone_from=True), ret)
|
||||
|
||||
# 'started' function tests: 1
|
||||
|
||||
def test_started(self):
|
||||
'''
|
||||
Test to execute started func.
|
||||
'''
|
||||
name = 'web01'
|
||||
|
||||
ret = {'name': name,
|
||||
'result': False,
|
||||
'comment': '',
|
||||
'changes': {}}
|
||||
|
||||
mock = MagicMock(return_value=None)
|
||||
with patch.dict(lxc.__salt__, {'lxc.state': mock}):
|
||||
with patch.object(salt.utils, 'warn_until', MagicMock()):
|
||||
comt = ("Container 'web01' does not exist")
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(lxc.started(name), ret)
|
||||
|
||||
# 'cloned' function tests: 1
|
||||
|
||||
def test_cloned(self):
|
||||
'''
|
||||
Test to execute cloned func.
|
||||
'''
|
||||
name = 'web01'
|
||||
|
||||
ret = {'name': name,
|
||||
'result': False,
|
||||
'comment': '',
|
||||
'changes': {}}
|
||||
|
||||
mock = MagicMock(return_value=False)
|
||||
with patch.dict(lxc.__salt__, {'lxc.exists': mock}):
|
||||
with patch.object(salt.utils, 'warn_until', MagicMock()):
|
||||
comt = ("Clone source 'True' does not exist")
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(lxc.cloned(name, True), ret)
|
||||
|
||||
# 'set_pass' function tests: 1
|
||||
|
||||
def test_set_pass(self):
|
||||
'''
|
||||
Test to execute set_pass func.
|
||||
'''
|
||||
comment = ('The lxc.set_pass state is no longer supported. Please see '
|
||||
'the LXC states documentation for further information.')
|
||||
ret = {'name': 'web01',
|
||||
'comment': comment,
|
||||
'result': False,
|
||||
'changes': {}}
|
||||
|
||||
self.assertDictEqual(lxc.set_pass('web01'), ret)
|
||||
|
||||
# 'edited_conf' function tests: 1
|
||||
|
||||
def test_edited_conf(self):
|
||||
'''
|
||||
Test to edit LXC configuration options
|
||||
'''
|
||||
name = 'web01'
|
||||
|
||||
comment = ('{0} lxc.conf will be edited'.format(name))
|
||||
|
||||
ret = {'name': name,
|
||||
'result': True,
|
||||
'comment': comment,
|
||||
'changes': {}}
|
||||
|
||||
with patch.object(salt.utils, 'warn_until', MagicMock()):
|
||||
with patch.dict(lxc.__opts__, {'test': True}):
|
||||
self.assertDictEqual(lxc.edited_conf(name), ret)
|
||||
|
||||
with patch.dict(lxc.__opts__, {'test': False}):
|
||||
mock = MagicMock(return_value={})
|
||||
with patch.dict(lxc.__salt__, {'lxc.update_lxc_conf': mock}):
|
||||
self.assertDictEqual(lxc.edited_conf(name),
|
||||
{'name': 'web01'})
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
from integration import run_tests
|
||||
run_tests(LxcTestCase, needs_daemon=False)
|
||||
Loading…
Add table
Reference in a new issue