def setUp(self):
super(TestVyosConfigModule, self).setUp()
self.mock_get_config = patch(
'ansible_collections.vyos.vyos.plugins.modules.vyos_config.get_config'
)
self.get_config = self.mock_get_config.start()
self.mock_load_config = patch(
'ansible_collections.vyos.vyos.plugins.modules.vyos_config.load_config'
)
self.load_config = self.mock_load_config.start()
self.mock_run_commands = patch(
'ansible_collections.vyos.vyos.plugins.modules.vyos_config.run_commands'
)
self.run_commands = self.mock_run_commands.start()
self.mock_get_connection = patch(
'ansible_collections.vyos.vyos.plugins.modules.vyos_config.get_connection'
)
self.get_connection = self.mock_get_connection.start()
self.cliconf_obj = Cliconf(MagicMock())
self.running_config = load_fixture('vyos_config_config.cfg')
self.conn = self.get_connection()
self.conn.edit_config = MagicMock()
self.running_config = load_fixture('vyos_config_config.cfg')
def test_vyos_config_lines(self):
commands = ['set system host-name foo']
set_module_args(dict(lines=commands))
candidate = '\n'.join(commands)
self.conn.get_diff = MagicMock(return_value=self.cliconf_obj.get_diff(
candidate, self.running_config))
self.execute_module(changed=True, commands=commands)
def test_vyos_config_unchanged(self):
src = load_fixture("vyos_config_config.cfg")
self.conn.get_diff = MagicMock(
return_value=self.cliconf_obj.get_diff(src, src)
)
set_module_args(dict(src=src))
self.execute_module()
def test_vyos_config_config(self):
config = 'set system host-name localhost'
new_config = ['set system host-name router']
set_module_args(dict(lines=new_config, config=config))
candidate = '\n'.join(new_config)
self.conn.get_diff = MagicMock(
return_value=self.cliconf_obj.get_diff(candidate, config))
self.execute_module(changed=True, commands=new_config)
def test_vyos_config_match_none(self):
lines = [
'set system interfaces ethernet eth0 address 1.2.3.4/24',
'set system interfaces ethernet eth0 description test string'
]
set_module_args(dict(lines=lines, match='none'))
candidate = '\n'.join(lines)
self.conn.get_diff = MagicMock(return_value=self.cliconf_obj.get_diff(
candidate, None, diff_match='none'))
self.execute_module(changed=True, commands=lines, sort=False)
def test_vyos_config_src_brackets(self):
src = load_fixture("vyos_config_src_brackets.cfg")
set_module_args(dict(src=src))
commands = [
"set interfaces ethernet eth0 address 10.10.10.10/24",
"set policy route testroute rule 1 set table 10",
"set system host-name foo",
]
self.conn.get_diff = MagicMock(side_effect=self.cliconf_obj.get_diff)
self.execute_module(changed=True, commands=commands)
def test_vyos_config_src_brackets(self):
src = load_fixture('vyos_config_src_brackets.cfg')
set_module_args(dict(src=src))
candidate = '\n'.join(self.module.format_commands(src.splitlines()))
commands = [
'set interfaces ethernet eth0 address 10.10.10.10/24',
'set system host-name foo'
]
self.conn.get_diff = MagicMock(return_value=self.cliconf_obj.get_diff(
candidate, self.running_config))
self.execute_module(changed=True, commands=commands)
def test_vyos_config_src(self):
src = load_fixture("vyos_config_src.cfg")
set_module_args(dict(src=src))
candidate = "\n".join(self.module.format_commands(src.splitlines()))
commands = [
"set system host-name foo",
"delete interfaces ethernet eth0 address",
]
self.conn.get_diff = MagicMock(return_value=self.cliconf_obj.get_diff(
candidate, self.running_config))
self.execute_module(changed=True, commands=commands)
from ansible_collections.vyos.vyos.tests.unit.compat.mock import MagicMock
from ansible.utils.path import unfrackpath
mock_unfrackpath_noop = MagicMock(spec_set=unfrackpath,
side_effect=lambda x, *args, **kwargs: x)