valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-07 08:58:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Test cases

Browse Source
This commit is contained in:
rajvidhimar 2017-03-06 14:55:40 +05:30
parent a06d5e961d
commit 1f8fe142dd
2 changed files with 27 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
salt/modules/junos.py
View File

@ -77,6 +77,8 @@ def facts_refresh():
except Exception as exception:
ret['message'] = 'Execution failed due to "{0}"'.format(exception)
ret['out'] = False
return ret
ret['facts'] = __proxy__['junos.get_serialized_facts']()
try:
@ -885,13 +887,9 @@ def install_config(path=None, **kwargs):
check = conn.cu.commit_check()
except Exception as exception:
ret['message'] = \
<<<<<<< HEAD
'Commit check threw the following exception: "{0}"'\
.format(exception)
=======
'Commit check threw the following exception: "{0}"' \
.format(exception)
>>>>>>> bdc8d2d196417a4ce9247d5ea9a9770ab5d1adae
ret['out'] = False
return ret
@ -901,13 +899,8 @@ def install_config(path=None, **kwargs):
ret['message'] = 'Successfully loaded and committed!'
except Exception as exception:
ret['message'] = \
<<<<<<< HEAD
'Commit check successful but commit failed with "{0}"'\
.format(exception)
=======
'Commit check successful but commit failed with "{0}"' \
.format(exception)
>>>>>>> bdc8d2d196417a4ce9247d5ea9a9770ab5d1adae
ret['out'] = False
return ret
else:

42
tests/unit/modules/junos_test.py
View File

@ -10,14 +10,13 @@ from mock import patch, MagicMock, ANY
from jnpr.junos.utils.config import Config
from jnpr.junos.utils.sw import SW
from jnpr.junos.device import Device
from jnpr.junos.facts.swver import version_info
import salt.modules.junos as junos
@attr('unit')
class Test_Junos_Module(unittest.TestCase):
def setUp(self):
junos.__proxy__ = {'junos.conn': self.make_connect}
junos.__proxy__ = {'junos.conn': self.make_connect, 'junos.get_serialized_facts': self.get_facts}
junos.__salt__ = {'cp.get_template': MagicMock}
@patch('ncclient.manager.connect')
@ -35,7 +34,7 @@ class Test_Junos_Module(unittest.TestCase):
def raise_exception(self, *args ,**kwargs):
raise Exception('dummy exception')
def get_facts(self, key):
def get_facts(self):
facts = {'2RE': True,
'HOME': '/var/home/regress',
'RE0': {'last_reboot_reason': '0x200:normal shutdown',
@ -92,14 +91,15 @@ class Test_Junos_Module(unittest.TestCase):
'version': '16.1I20160413_0837_aamish',
'version_RE0': '16.1I20160413_0837_aamish',
'version_RE1': '16.1I20160413_0837_aamish',
'version_info': version_info("16.1I20160413_0837_aamish"),
'version_info': {'build': None,
'major': (16, 1),
'minor': '20160413_0837_aamish',
'type': 'I'},
'virtual': True}
return facts[key]
return facts
@patch('salt.modules.saltutil.sync_grains')
@patch('jnpr.junos.factcache._FactCache.__getitem__')
def test_facts_refresh(self, mock_facts, mock_sync_grains):
mock_facts.side_effect = self.get_facts
def test_facts_refresh(self, mock_sync_grains):
ret = dict()
ret['facts'] = {'2RE': True,
'HOME': '/var/home/regress',
@ -163,10 +163,17 @@ class Test_Junos_Module(unittest.TestCase):
'type': 'I'},
'virtual': True}
ret['out'] = True
self.assertEqual(junos.facts_refresh(), ret)
@patch('jnpr.junos.factcache._FactCache.__getitem__')
def test_facts(self, mock_facts):
mock_facts.side_effect = self.get_facts
@patch('jnpr.junos.device.Device.facts_refresh')
def test_facts_refresh_exception(self, mock_facts_refresh):
mock_facts_refresh.side_effect = self.raise_exception
ret = dict()
ret['message'] = 'Execution failed due to "dummy exception"'
ret['out'] = False
self.assertEqual(junos.facts_refresh(), ret)
def test_facts(self):
ret = dict()
ret['facts'] = {'2RE': True,
'HOME': '/var/home/regress',
@ -232,13 +239,12 @@ class Test_Junos_Module(unittest.TestCase):
ret['out'] = True
self.assertEqual(junos.facts(), ret)
# @patch('jnpr.junos.device.Device.facts')
# def test_facts_exception(self, mock_facts):
# mock_facts.side_effect = self.raise_exception
# ret = dict()
# ret['message'] = 'Could not display facts due to "{0}"'
# ret['out'] = False
# self.assertEqual(junos.facts(), ret)
def test_facts_exception(self):
junos.__proxy__ = {'junos.get_serialized_facts': self.raise_exception}
ret = dict()
ret['message'] = 'Could not display facts due to "dummy exception"'
ret['out'] = False
self.assertEqual(junos.facts(), ret)
def test_set_hostname_without_args(self):
ret = dict()