You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ansible/test/units/modules/network/icx/test_icx_lldp.py

100 lines
4.5 KiB
Python

# Copyright: (c) 2019, Ansible Project
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type
from units.compat.mock import patch
from ansible.modules.network.icx import icx_lldp
from units.modules.utils import set_module_args
from .icx_module import TestICXModule, load_fixture
class TestICXlldpModule(TestICXModule):
module = icx_lldp
def setUp(self):
super(TestICXlldpModule, self).setUp()
self.mock_load_config = patch('ansible.modules.network.icx.icx_lldp.load_config')
self.load_config = self.mock_load_config.start()
self.mock_run_commands = patch('ansible.modules.network.icx.icx_lldp.run_commands')
self.run_commands = self.mock_run_commands.start()
self.set_running_config()
def tearDown(self):
super(TestICXlldpModule, self).tearDown()
self.mock_load_config.stop()
self.mock_run_commands.stop()
def load_fixtures(self, commands=None):
def load_from_file(*args, **kwargs):
compares = None
module, commands = args
state = module.params['state']
if module.params['check_running_config'] is True:
return load_fixture('icx_lldp_%s' % state).strip()
else:
return ''
self.run_commands.side_effect = load_from_file
def test_icx_lldp_enable_state_None(self):
interfaces_spec = [dict(name='ethernet 1/1/9', state='present')]
set_module_args(dict(interfaces=interfaces_spec))
if not self.ENV_ICX_USE_DIFF:
result = self.execute_module(failed=True)
else:
result = self.execute_module(failed=True)
def test_icx_lldp_enable_state_absent_compare(self):
interfaces_spec = [dict(name='ethernet 1/1/9', state='present')]
set_module_args(dict(interfaces=interfaces_spec, state='absent', check_running_config=True))
if self.get_running_config(compare=True):
if not self.ENV_ICX_USE_DIFF:
result = self.execute_module(changed=True)
self.assertEqual(result['commands'], ['no lldp run'])
else:
result = self.execute_module(changed=True)
self.assertEqual(result['commands'], ['no lldp run'])
def test_icx_lldp_enable_state_present(self):
interfaces_spec = [dict(name='ethernet 1/1/9', state='present')]
set_module_args(dict(interfaces=interfaces_spec, state='present'))
if not self.ENV_ICX_USE_DIFF:
result = self.execute_module(changed=True)
self.assertEqual(result['commands'], ['lldp enable ports ethernet 1/1/9'])
else:
result = self.execute_module(changed=True)
self.assertEqual(result['commands'], ['lldp enable ports ethernet 1/1/9'])
def test_icx_lldp_multi_enable_state_present(self):
interfaces_spec = [dict(name=['ethernet 1/1/9', 'ethernet 1/1/1 to 1/1/6'], state='present')]
set_module_args(dict(interfaces=interfaces_spec, state='present'))
if not self.ENV_ICX_USE_DIFF:
result = self.execute_module(changed=True)
self.assertEqual(result['commands'], ['lldp enable ports ethernet 1/1/9', 'lldp enable ports ethernet 1/1/1 to 1/1/6'])
else:
result = self.execute_module(changed=True)
self.assertEqual(result['commands'], ['lldp enable ports ethernet 1/1/9', 'lldp enable ports ethernet 1/1/1 to 1/1/6'])
def test_icx_lldp_multi_disable_state_present(self):
interfaces_spec = [dict(name=['ethernet 1/1/9', 'ethernet 1/1/1 to 1/1/6'], state='absent')]
set_module_args(dict(interfaces=interfaces_spec, state='present'))
if not self.ENV_ICX_USE_DIFF:
result = self.execute_module(changed=True)
self.assertEqual(result['commands'], ['no lldp enable ports ethernet 1/1/9', 'no lldp enable ports ethernet 1/1/1 to 1/1/6'])
else:
result = self.execute_module(changed=True)
self.assertEqual(result['commands'], ['no lldp enable ports ethernet 1/1/9', 'no lldp enable ports ethernet 1/1/1 to 1/1/6'])
def test_icx_lldp_all_error(self):
interfaces_spec = [dict(name=['ethernet all'], state='absent')]
set_module_args(dict(interfaces=interfaces_spec, state='present'))
if not self.ENV_ICX_USE_DIFF:
self.execute_module(failed=True)
else:
self.execute_module(failed=True)