summaryrefslogtreecommitdiff
path: root/tests/unit
diff options
context:
space:
mode:
authorCaptTrews <capttrews@gmail.com>2020-02-19 19:31:27 +0000
committerCaptTrews <capttrews@gmail.com>2020-02-19 19:31:27 +0000
commit45f223636c73ba69d3fea3c8aab8edd41de01388 (patch)
treec5ff33aa0e4770a13340a52b1e2bda7531541a5f /tests/unit
parentf63b5c97edbf598f7b2a4c044386de3dddfda100 (diff)
downloadvyos-ansible-collection-45f223636c73ba69d3fea3c8aab8edd41de01388.tar.gz
vyos-ansible-collection-45f223636c73ba69d3fea3c8aab8edd41de01388.zip
Updated from network content collector
Signed-off-by: CaptTrews <capttrews@gmail.com>
Diffstat (limited to 'tests/unit')
-rw-r--r--tests/unit/modules/network/vyos/fixtures/vyos_firewall_rules_config.cfg13
-rw-r--r--tests/unit/modules/network/vyos/fixtures/vyos_static_routes_config.cfg2
-rw-r--r--tests/unit/modules/network/vyos/test_vyos_firewall_rules.py1039
-rw-r--r--tests/unit/modules/network/vyos/test_vyos_static_route.py8
-rw-r--r--tests/unit/modules/network/vyos/test_vyos_static_routes.py293
5 files changed, 1351 insertions, 4 deletions
diff --git a/tests/unit/modules/network/vyos/fixtures/vyos_firewall_rules_config.cfg b/tests/unit/modules/network/vyos/fixtures/vyos_firewall_rules_config.cfg
new file mode 100644
index 0000000..f65b386
--- /dev/null
+++ b/tests/unit/modules/network/vyos/fixtures/vyos_firewall_rules_config.cfg
@@ -0,0 +1,13 @@
+set firewall name V4-INGRESS default-action 'accept'
+set firewall ipv6-name V6-INGRESS default-action 'accept'
+set firewall name V4-INGRESS description 'This is IPv4 V4-INGRESS rule set'
+set firewall name V4-INGRESS enable-default-log
+set firewall name V4-INGRESS rule 101 protocol 'icmp'
+set firewall name V4-INGRESS rule 101 description 'Rule 101 is configured by Ansible'
+set firewall name V4-INGRESS rule 101 fragment 'match-frag'
+set firewall name V4-INGRESS rule 101
+set firewall name V4-INGRESS rule 101 disabled
+set firewall name V4-INGRESS rule 101 action 'accept'
+set firewall name V4-INGRESS rule 101 ipsec 'match-ipsec'
+set firewall name V4-EGRESS default-action 'reject'
+set firewall ipv6-name V6-EGRESS default-action 'reject'
diff --git a/tests/unit/modules/network/vyos/fixtures/vyos_static_routes_config.cfg b/tests/unit/modules/network/vyos/fixtures/vyos_static_routes_config.cfg
new file mode 100644
index 0000000..0411dc9
--- /dev/null
+++ b/tests/unit/modules/network/vyos/fixtures/vyos_static_routes_config.cfg
@@ -0,0 +1,2 @@
+'set protocols static route 192.0.2.32/28 next-hop 192.0.2.9'
+'set protocols static route 192.0.2.32/28 next-hop 192.0.2.10'
diff --git a/tests/unit/modules/network/vyos/test_vyos_firewall_rules.py b/tests/unit/modules/network/vyos/test_vyos_firewall_rules.py
new file mode 100644
index 0000000..86fcc65
--- /dev/null
+++ b/tests/unit/modules/network/vyos/test_vyos_firewall_rules.py
@@ -0,0 +1,1039 @@
+# (c) 2016 Red Hat Inc.
+#
+# This file is part of Ansible
+#
+# Ansible is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Ansible is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
+
+# Make coding more python3-ish
+from __future__ import absolute_import, division, print_function
+
+__metaclass__ = type
+
+from ansible_collections.vyos.vyos.tests.unit.compat.mock import patch
+from ansible_collections.vyos.vyos.plugins.modules import vyos_firewall_rules
+from ansible_collections.vyos.vyos.tests.unit.modules.utils import (
+ set_module_args,
+)
+from .vyos_module import TestVyosModule, load_fixture
+
+
+class TestVyosFirewallRulesModule(TestVyosModule):
+
+ module = vyos_firewall_rules
+
+ def setUp(self):
+ super(TestVyosFirewallRulesModule, self).setUp()
+ self.mock_get_config = patch(
+ "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.network.Config.get_config"
+ )
+ self.get_config = self.mock_get_config.start()
+
+ self.mock_load_config = patch(
+ "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.network.Config.load_config"
+ )
+ self.load_config = self.mock_load_config.start()
+
+ self.mock_get_resource_connection_config = patch(
+ "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.cfg.base.get_resource_connection"
+ )
+ self.get_resource_connection_config = (
+ self.mock_get_resource_connection_config.start()
+ )
+
+ self.mock_get_resource_connection_facts = patch(
+ "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.facts.facts.get_resource_connection"
+ )
+ self.get_resource_connection_facts = (
+ self.mock_get_resource_connection_facts.start()
+ )
+ self.mock_execute_show_command = patch(
+ "ansible_collections.vyos.vyos.plugins.module_utils.network.vyos.facts.static_routes.static_routes.Static_routesFacts.get_device_data"
+ )
+
+ self.mock_execute_show_command = patch(
+ "ansible_collections.vyos.vyos.plugins.module_utils.network.vyos.facts.firewall_rules.firewall_rules.Firewall_rulesFacts.get_device_data"
+ )
+ self.execute_show_command = self.mock_execute_show_command.start()
+
+ def tearDown(self):
+ super(TestVyosFirewallRulesModule, self).tearDown()
+ self.mock_get_resource_connection_config.stop()
+ self.mock_get_resource_connection_facts.stop()
+ self.mock_get_config.stop()
+ self.mock_load_config.stop()
+ self.mock_execute_show_command.stop()
+
+ def load_fixtures(self, commands=None):
+ def load_from_file(*args, **kwargs):
+ return load_fixture("vyos_firewall_rules_config.cfg")
+
+ self.execute_show_command.side_effect = load_from_file
+
+ def test_vyos_firewall_rule_set_01_merged(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv6",
+ rule_sets=[
+ dict(
+ name="V6-INBOUND",
+ description="This is IPv6 INBOUND rule set",
+ default_action="reject",
+ enable_default_log=True,
+ rules=[],
+ ),
+ dict(
+ name="V6-OUTBOUND",
+ description="This is IPv6 OUTBOUND rule set",
+ default_action="accept",
+ enable_default_log=False,
+ rules=[],
+ ),
+ ],
+ ),
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="V4-INBOUND",
+ description="This is IPv4 INBOUND rule set",
+ default_action="reject",
+ enable_default_log=True,
+ rules=[],
+ ),
+ dict(
+ name="V4-OUTBOUND",
+ description="This is IPv4 OUTBOUND rule set",
+ default_action="accept",
+ enable_default_log=False,
+ rules=[],
+ ),
+ ],
+ ),
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set firewall ipv6-name V6-INBOUND default-action 'reject'",
+ "set firewall ipv6-name V6-INBOUND description 'This is IPv6 INBOUND rule set'",
+ "set firewall ipv6-name V6-INBOUND enable-default-log",
+ "set firewall ipv6-name V6-OUTBOUND default-action 'accept'",
+ "set firewall ipv6-name V6-OUTBOUND description 'This is IPv6 OUTBOUND rule set'",
+ "set firewall name V4-INBOUND default-action 'reject'",
+ "set firewall name V4-INBOUND description 'This is IPv4 INBOUND rule set'",
+ "set firewall name V4-INBOUND enable-default-log",
+ "set firewall name V4-OUTBOUND default-action 'accept'",
+ "set firewall name V4-OUTBOUND description 'This is IPv4 OUTBOUND rule set'",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_rule_set_02_merged(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv6",
+ rule_sets=[
+ dict(
+ name="V6-INBOUND",
+ description="This is IPv6 INBOUND rule set",
+ default_action="reject",
+ enable_default_log=True,
+ rules=[],
+ ),
+ dict(
+ name="V6-OUTBOUND",
+ description="This is IPv6 OUTBOUND rule set",
+ default_action="accept",
+ enable_default_log=False,
+ rules=[],
+ ),
+ ],
+ ),
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="V4-INBOUND",
+ description="This is IPv4 INBOUND rule set",
+ default_action="reject",
+ enable_default_log=True,
+ rules=[],
+ ),
+ dict(
+ name="V4-OUTBOUND",
+ description="This is IPv4 OUTBOUND rule set",
+ default_action="accept",
+ enable_default_log=False,
+ rules=[],
+ ),
+ ],
+ ),
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set firewall ipv6-name V6-INBOUND default-action 'reject'",
+ "set firewall ipv6-name V6-INBOUND description 'This is IPv6 INBOUND rule set'",
+ "set firewall ipv6-name V6-INBOUND enable-default-log",
+ "set firewall ipv6-name V6-OUTBOUND default-action 'accept'",
+ "set firewall ipv6-name V6-OUTBOUND description 'This is IPv6 OUTBOUND rule set'",
+ "set firewall name V4-INBOUND default-action 'reject'",
+ "set firewall name V4-INBOUND description 'This is IPv4 INBOUND rule set'",
+ "set firewall name V4-INBOUND enable-default-log",
+ "set firewall name V4-OUTBOUND default-action 'accept'",
+ "set firewall name V4-OUTBOUND description 'This is IPv4 OUTBOUND rule set'",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v4_rule_sets_rule_merged_01(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="INBOUND",
+ description="This is IPv4 INBOUND rule set",
+ default_action="accept",
+ enable_default_log=True,
+ rules=[
+ dict(
+ number="101",
+ action="accept",
+ description="Rule 101 is configured by Ansible",
+ ipsec="match-ipsec",
+ protocol="icmp",
+ fragment="match-frag",
+ disabled=True,
+ )
+ ],
+ ),
+ ],
+ )
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set firewall name INBOUND default-action 'accept'",
+ "set firewall name INBOUND description 'This is IPv4 INBOUND rule set'",
+ "set firewall name INBOUND enable-default-log",
+ "set firewall name INBOUND rule 101 protocol 'icmp'",
+ "set firewall name INBOUND rule 101 description 'Rule 101 is configured by Ansible'",
+ "set firewall name INBOUND rule 101 fragment 'match-frag'",
+ "set firewall name INBOUND rule 101",
+ "set firewall name INBOUND rule 101 disabled",
+ "set firewall name INBOUND rule 101 action 'accept'",
+ "set firewall name INBOUND rule 101 ipsec 'match-ipsec'",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v4_rule_sets_rule_merged_02(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="INBOUND",
+ rules=[
+ dict(
+ number="101",
+ protocol="tcp",
+ source=dict(
+ address="192.0.2.0",
+ mac_address="38:00:25:19:76:0c",
+ port=2127,
+ ),
+ destination=dict(
+ address="192.0.1.0", port=2124
+ ),
+ limit=dict(
+ burst=10,
+ rate=dict(
+ number=20, unit="second"
+ ),
+ ),
+ recent=dict(count=10, time=20),
+ state=dict(
+ established=True,
+ related=True,
+ invalid=True,
+ new=True,
+ ),
+ )
+ ],
+ ),
+ ],
+ )
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set firewall name INBOUND rule 101 protocol 'tcp'",
+ "set firewall name INBOUND rule 101 destination address 192.0.1.0",
+ "set firewall name INBOUND rule 101 destination port 2124",
+ "set firewall name INBOUND rule 101",
+ "set firewall name INBOUND rule 101 source address 192.0.2.0",
+ "set firewall name INBOUND rule 101 source mac-address 38:00:25:19:76:0c",
+ "set firewall name INBOUND rule 101 source port 2127",
+ "set firewall name INBOUND rule 101 state new enable",
+ "set firewall name INBOUND rule 101 state invalid enable",
+ "set firewall name INBOUND rule 101 state related enable",
+ "set firewall name INBOUND rule 101 state established enable",
+ "set firewall name INBOUND rule 101 limit burst 10",
+ "set firewall name INBOUND rule 101 limit rate 20/second",
+ "set firewall name INBOUND rule 101 recent count 10",
+ "set firewall name INBOUND rule 101 recent time 20",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v4_rule_sets_rule_merged_03(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="INBOUND",
+ rules=[
+ dict(
+ number="101",
+ destination=dict(
+ group=dict(
+ address_group="OUT-ADDR-GROUP",
+ network_group="OUT-NET-GROUP",
+ port_group="OUT-PORT-GROUP",
+ )
+ ),
+ source=dict(
+ group=dict(
+ address_group="IN-ADDR-GROUP",
+ network_group="IN-NET-GROUP",
+ port_group="IN-PORT-GROUP",
+ )
+ ),
+ )
+ ],
+ ),
+ ],
+ )
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set firewall name INBOUND rule 101 source group address-group IN-ADDR-GROUP",
+ "set firewall name INBOUND rule 101 source group network-group IN-NET-GROUP",
+ "set firewall name INBOUND rule 101 source group port-group IN-PORT-GROUP",
+ "set firewall name INBOUND rule 101 destination group address-group OUT-ADDR-GROUP",
+ "set firewall name INBOUND rule 101 destination group network-group OUT-NET-GROUP",
+ "set firewall name INBOUND rule 101 destination group port-group OUT-PORT-GROUP",
+ "set firewall name INBOUND rule 101",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v4_rule_sets_rule_merged_04(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="INBOUND",
+ rules=[
+ dict(
+ number="101",
+ time=dict(
+ monthdays="2",
+ startdate="2020-01-24",
+ starttime="13:20:00",
+ stopdate="2020-01-28",
+ stoptime="13:30:00",
+ weekdays="!Sat,Sun",
+ utc=True,
+ ),
+ tcp=dict(flags="ALL"),
+ )
+ ],
+ ),
+ ],
+ )
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set firewall name INBOUND rule 101",
+ "set firewall name INBOUND rule 101 tcp flags ALL",
+ "set firewall name INBOUND rule 101 time utc",
+ "set firewall name INBOUND rule 101 time monthdays 2",
+ "set firewall name INBOUND rule 101 time startdate 2020-01-24",
+ "set firewall name INBOUND rule 101 time stopdate 2020-01-28",
+ "set firewall name INBOUND rule 101 time weekdays !Sat,Sun",
+ "set firewall name INBOUND rule 101 time stoptime 13:30:00",
+ "set firewall name INBOUND rule 101 time starttime 13:20:00",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v6_rule_sets_rule_merged_01(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv6",
+ rule_sets=[
+ dict(
+ name="INBOUND",
+ description="This is IPv6 INBOUND rule set",
+ default_action="accept",
+ enable_default_log=True,
+ rules=[
+ dict(
+ number="101",
+ action="accept",
+ description="Rule 101 is configured by Ansible",
+ ipsec="match-ipsec",
+ protocol="icmp",
+ disabled=True,
+ )
+ ],
+ ),
+ ],
+ )
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set firewall ipv6-name INBOUND default-action 'accept'",
+ "set firewall ipv6-name INBOUND description 'This is IPv6 INBOUND rule set'",
+ "set firewall ipv6-name INBOUND enable-default-log",
+ "set firewall ipv6-name INBOUND rule 101 protocol 'icmp'",
+ "set firewall ipv6-name INBOUND rule 101 description 'Rule 101 is configured by Ansible'",
+ "set firewall ipv6-name INBOUND rule 101",
+ "set firewall ipv6-name INBOUND rule 101 disabled",
+ "set firewall ipv6-name INBOUND rule 101 action 'accept'",
+ "set firewall ipv6-name INBOUND rule 101 ipsec 'match-ipsec'",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v6_rule_sets_rule_merged_02(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv6",
+ rule_sets=[
+ dict(
+ name="INBOUND",
+ rules=[
+ dict(
+ number="101",
+ protocol="tcp",
+ source=dict(
+ address="2001:db8::12",
+ mac_address="38:00:25:19:76:0c",
+ port=2127,
+ ),
+ destination=dict(
+ address="2001:db8::11", port=2124
+ ),
+ limit=dict(
+ burst=10,
+ rate=dict(
+ number=20, unit="second"
+ ),
+ ),
+ recent=dict(count=10, time=20),
+ state=dict(
+ established=True,
+ related=True,
+ invalid=True,
+ new=True,
+ ),
+ )
+ ],
+ ),
+ ],
+ )
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set firewall ipv6-name INBOUND rule 101 protocol 'tcp'",
+ "set firewall ipv6-name INBOUND rule 101 destination address 2001:db8::11",
+ "set firewall ipv6-name INBOUND rule 101 destination port 2124",
+ "set firewall ipv6-name INBOUND rule 101",
+ "set firewall ipv6-name INBOUND rule 101 source address 2001:db8::12",
+ "set firewall ipv6-name INBOUND rule 101 source mac-address 38:00:25:19:76:0c",
+ "set firewall ipv6-name INBOUND rule 101 source port 2127",
+ "set firewall ipv6-name INBOUND rule 101 state new enable",
+ "set firewall ipv6-name INBOUND rule 101 state invalid enable",
+ "set firewall ipv6-name INBOUND rule 101 state related enable",
+ "set firewall ipv6-name INBOUND rule 101 state established enable",
+ "set firewall ipv6-name INBOUND rule 101 limit burst 10",
+ "set firewall ipv6-name INBOUND rule 101 recent count 10",
+ "set firewall ipv6-name INBOUND rule 101 recent time 20",
+ "set firewall ipv6-name INBOUND rule 101 limit rate 20/second",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v6_rule_sets_rule_merged_03(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv6",
+ rule_sets=[
+ dict(
+ name="INBOUND",
+ rules=[
+ dict(
+ number="101",
+ destination=dict(
+ group=dict(
+ address_group="OUT-ADDR-GROUP",
+ network_group="OUT-NET-GROUP",
+ port_group="OUT-PORT-GROUP",
+ )
+ ),
+ source=dict(
+ group=dict(
+ address_group="IN-ADDR-GROUP",
+ network_group="IN-NET-GROUP",
+ port_group="IN-PORT-GROUP",
+ )
+ ),
+ )
+ ],
+ ),
+ ],
+ )
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set firewall ipv6-name INBOUND rule 101 source group address-group IN-ADDR-GROUP",
+ "set firewall ipv6-name INBOUND rule 101 source group network-group IN-NET-GROUP",
+ "set firewall ipv6-name INBOUND rule 101 source group port-group IN-PORT-GROUP",
+ "set firewall ipv6-name INBOUND rule 101 destination group address-group OUT-ADDR-GROUP",
+ "set firewall ipv6-name INBOUND rule 101 destination group network-group OUT-NET-GROUP",
+ "set firewall ipv6-name INBOUND rule 101 destination group port-group OUT-PORT-GROUP",
+ "set firewall ipv6-name INBOUND rule 101",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v6_rule_sets_rule_merged_04(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv6",
+ rule_sets=[
+ dict(
+ name="INBOUND",
+ rules=[
+ dict(
+ number="101",
+ time=dict(
+ monthdays="2",
+ startdate="2020-01-24",
+ starttime="13:20:00",
+ stopdate="2020-01-28",
+ stoptime="13:30:00",
+ weekdays="!Sat,Sun",
+ utc=True,
+ ),
+ tcp=dict(flags="ALL"),
+ )
+ ],
+ ),
+ ],
+ )
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set firewall ipv6-name INBOUND rule 101",
+ "set firewall ipv6-name INBOUND rule 101 tcp flags ALL",
+ "set firewall ipv6-name INBOUND rule 101 time utc",
+ "set firewall ipv6-name INBOUND rule 101 time monthdays 2",
+ "set firewall ipv6-name INBOUND rule 101 time startdate 2020-01-24",
+ "set firewall ipv6-name INBOUND rule 101 time stopdate 2020-01-28",
+ "set firewall ipv6-name INBOUND rule 101 time weekdays !Sat,Sun",
+ "set firewall ipv6-name INBOUND rule 101 time stoptime 13:30:00",
+ "set firewall ipv6-name INBOUND rule 101 time starttime 13:20:00",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v6_rule_sets_rule_merged_icmp_01(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv6",
+ rule_sets=[
+ dict(
+ name="INBOUND",
+ rules=[
+ dict(
+ number="101",
+ protocol="icmp",
+ icmp=dict(
+ type_name="port-unreachable"
+ ),
+ )
+ ],
+ ),
+ ],
+ )
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set firewall ipv6-name INBOUND rule 101 icmpv6 type port-unreachable",
+ "set firewall ipv6-name INBOUND rule 101 protocol 'icmp'",
+ "set firewall ipv6-name INBOUND rule 101",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v4_rule_sets_rule_merged_icmp_01(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="INBOUND",
+ rules=[
+ dict(
+ number="101",
+ protocol="icmp",
+ icmp=dict(type=1, code=1),
+ )
+ ],
+ ),
+ ],
+ )
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set firewall name INBOUND rule 101 icmp type 1",
+ "set firewall name INBOUND rule 101 icmp code 1",
+ "set firewall name INBOUND rule 101 protocol 'icmp'",
+ "set firewall name INBOUND rule 101",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v4_rule_sets_rule_merged_icmp_02(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="INBOUND",
+ rules=[
+ dict(
+ number="101",
+ protocol="icmp",
+ icmp=dict(type_name="echo-request"),
+ )
+ ],
+ ),
+ ],
+ )
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set firewall name INBOUND rule 101 icmp type-name echo-request",
+ "set firewall name INBOUND rule 101 protocol 'icmp'",
+ "set firewall name INBOUND rule 101",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v4_rule_sets_del_01(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(afi="ipv4", rule_sets=[dict(name="V4-INGRESS"),])
+ ],
+ state="deleted",
+ )
+ )
+ commands = ["delete firewall name V4-INGRESS"]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v4v6_rule_sets_del_02(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(afi="ipv4", rule_sets=[dict(name="V4-INGRESS"),]),
+ dict(afi="ipv6", rule_sets=[dict(name="V6-INGRESS"),]),
+ ],
+ state="deleted",
+ )
+ )
+ commands = [
+ "delete firewall name V4-INGRESS",
+ "delete firewall ipv6-name V6-INGRESS",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v4v6_rule_sets_del_03(self):
+ set_module_args(dict(config=[], state="deleted"))
+ commands = ["delete firewall name", "delete firewall ipv6-name"]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v4v6_rule_sets_del_04(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(afi="ipv4", rule_sets=[dict(name="V4-ING"),]),
+ dict(afi="ipv6", rule_sets=[dict(name="V6-ING"),]),
+ ],
+ state="deleted",
+ )
+ )
+ self.execute_module(changed=False, commands=[])
+
+ def test_vyos_firewall_v4v6_rule_sets_rule_rep_01(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="V4-INGRESS",
+ description="This is IPv4 INGRESS rule set",
+ default_action="accept",
+ enable_default_log=True,
+ rules=[
+ dict(
+ number="101",
+ action="reject",
+ description="Rule 101 is configured by Ansible RM",
+ ipsec="match-ipsec",
+ protocol="tcp",
+ fragment="match-frag",
+ disabled=False,
+ ),
+ dict(
+ number="102",
+ action="accept",
+ description="Rule 102 is configured by Ansible RM",
+ protocol="icmp",
+ disabled=True,
+ ),
+ ],
+ ),
+ ],
+ ),
+ dict(
+ afi="ipv6",
+ rule_sets=[
+ dict(
+ name="V6-INGRESS",
+ default_action="accept",
+ description="This rule-set is configured by Ansible RM",
+ ),
+ dict(
+ name="V6-EGRESS",
+ default_action="reject",
+ description="This rule-set is configured by Ansible RM",
+ ),
+ ],
+ ),
+ ],
+ state="replaced",
+ )
+ )
+ commands = [
+ "delete firewall name V4-INGRESS rule 101 disabled",
+ "delete firewall name V4-EGRESS default-action",
+ "set firewall name V4-INGRESS description 'This is IPv4 INGRESS rule set'",
+ "set firewall name V4-INGRESS rule 101 protocol 'tcp'",
+ "set firewall name V4-INGRESS rule 101 description 'Rule 101 is configured by Ansible RM'",
+ "set firewall name V4-INGRESS rule 101 action 'reject'",
+ "set firewall name V4-INGRESS rule 102 disabled",
+ "set firewall name V4-INGRESS rule 102 action 'accept'",
+ "set firewall name V4-INGRESS rule 102 protocol 'icmp'",
+ "set firewall name V4-INGRESS rule 102 description 'Rule 102 is configured by Ansible RM'",
+ "set firewall name V4-INGRESS rule 102",
+ "set firewall ipv6-name V6-INGRESS description 'This rule-set is configured by Ansible RM'",
+ "set firewall ipv6-name V6-EGRESS description 'This rule-set is configured by Ansible RM'",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v4v6_rule_sets_rule_rep_02(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="V4-INGRESS",
+ description="This is IPv4 V4-INGRESS rule set",
+ default_action="accept",
+ enable_default_log=False,
+ rules=[
+ dict(
+ number="101",
+ action="accept",
+ description="Rule 101 is configured by Ansible",
+ ipsec="match-ipsec",
+ protocol="icmp",
+ fragment="match-frag",
+ disabled=True,
+ ),
+ ],
+ ),
+ ],
+ ),
+ dict(
+ afi="ipv6",
+ rule_sets=[
+ dict(name="V6-INGRESS", default_action="accept",),
+ dict(name="V6-EGRESS", default_action="reject",),
+ ],
+ ),
+ ],
+ state="replaced",
+ )
+ )
+ commands = [
+ "delete firewall name V4-INGRESS enable-default-log",
+ "delete firewall name V4-EGRESS default-action",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v4v6_rule_sets_rule_rep_idem_01(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="V4-INGRESS",
+ description="This is IPv4 V4-INGRESS rule set",
+ default_action="accept",
+ enable_default_log=True,
+ rules=[
+ dict(
+ number="101",
+ action="accept",
+ description="Rule 101 is configured by Ansible",
+ ipsec="match-ipsec",
+ protocol="icmp",
+ fragment="match-frag",
+ disabled=True,
+ )
+ ],
+ ),
+ dict(name="V4-EGRESS", default_action="reject",),
+ ],
+ ),
+ dict(
+ afi="ipv6",
+ rule_sets=[
+ dict(name="V6-INGRESS", default_action="accept",),
+ dict(name="V6-EGRESS", default_action="reject",),
+ ],
+ ),
+ ],
+ state="replaced",
+ )
+ )
+ self.execute_module(changed=False, commands=[])
+
+ def test_vyos_firewall_v4v6_rule_sets_rule_mer_idem_01(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="V4-INGRESS",
+ description="This is IPv4 V4-INGRESS rule set",
+ default_action="accept",
+ enable_default_log=True,
+ rules=[
+ dict(
+ number="101",
+ action="accept",
+ description="Rule 101 is configured by Ansible",
+ ipsec="match-ipsec",
+ protocol="icmp",
+ fragment="match-frag",
+ disabled=True,
+ )
+ ],
+ ),
+ dict(name="V4-EGRESS", default_action="reject",),
+ ],
+ ),
+ dict(
+ afi="ipv6",
+ rule_sets=[
+ dict(name="V6-INGRESS", default_action="accept",),
+ dict(name="V6-EGRESS", default_action="reject",),
+ ],
+ ),
+ ],
+ state="merged",
+ )
+ )
+ self.execute_module(changed=False, commands=[])
+
+ def test_vyos_firewall_v4v6_rule_sets_rule_ovr_01(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="V4-IN",
+ description="This is IPv4 INGRESS rule set",
+ default_action="accept",
+ enable_default_log=True,
+ rules=[
+ dict(
+ number="1",
+ action="reject",
+ description="Rule 1 is configured by Ansible RM",
+ ipsec="match-ipsec",
+ protocol="tcp",
+ fragment="match-frag",
+ disabled=False,
+ ),
+ dict(
+ number="2",
+ action="accept",
+ description="Rule 102 is configured by Ansible RM",
+ protocol="icmp",
+ disabled=True,
+ ),
+ ],
+ ),
+ ],
+ ),
+ dict(
+ afi="ipv6",
+ rule_sets=[
+ dict(
+ name="V6-IN",
+ default_action="accept",
+ description="This rule-set is configured by Ansible RM",
+ ),
+ dict(
+ name="V6-EG",
+ default_action="reject",
+ description="This rule-set is configured by Ansible RM",
+ ),
+ ],
+ ),
+ ],
+ state="overridden",
+ )
+ )
+ commands = [
+ "delete firewall ipv6-name V6-INGRESS",
+ "delete firewall ipv6-name V6-EGRESS",
+ "delete firewall name V4-INGRESS",
+ "delete firewall name V4-EGRESS",
+ "set firewall name V4-IN default-action 'accept'",
+ "set firewall name V4-IN description 'This is IPv4 INGRESS rule set'",
+ "set firewall name V4-IN enable-default-log",
+ "set firewall name V4-IN rule 1 protocol 'tcp'",
+ "set firewall name V4-IN rule 1 description 'Rule 1 is configured by Ansible RM'",
+ "set firewall name V4-IN rule 1 fragment 'match-frag'",
+ "set firewall name V4-IN rule 1",
+ "set firewall name V4-IN rule 1 action 'reject'",
+ "set firewall name V4-IN rule 1 ipsec 'match-ipsec'",
+ "set firewall name V4-IN rule 2 disabled",
+ "set firewall name V4-IN rule 2 action 'accept'",
+ "set firewall name V4-IN rule 2 protocol 'icmp'",
+ "set firewall name V4-IN rule 2 description 'Rule 102 is configured by Ansible RM'",
+ "set firewall name V4-IN rule 2",
+ "set firewall ipv6-name V6-IN default-action 'accept'",
+ "set firewall ipv6-name V6-IN description 'This rule-set is configured by Ansible RM'",
+ "set firewall ipv6-name V6-EG default-action 'reject'",
+ "set firewall ipv6-name V6-EG description 'This rule-set is configured by Ansible RM'",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_firewall_v4v6_rule_sets_rule_ovr_idem_01(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ afi="ipv4",
+ rule_sets=[
+ dict(
+ name="V4-INGRESS",
+ description="This is IPv4 V4-INGRESS rule set",
+ default_action="accept",
+ enable_default_log=True,
+ rules=[
+ dict(
+ number="101",
+ action="accept",
+ description="Rule 101 is configured by Ansible",
+ ipsec="match-ipsec",
+ protocol="icmp",
+ fragment="match-frag",
+ disabled=True,
+ )
+ ],
+ ),
+ dict(name="V4-EGRESS", default_action="reject",),
+ ],
+ ),
+ dict(
+ afi="ipv6",
+ rule_sets=[
+ dict(name="V6-INGRESS", default_action="accept",),
+ dict(name="V6-EGRESS", default_action="reject",),
+ ],
+ ),
+ ],
+ state="overridden",
+ )
+ )
+ self.execute_module(changed=False, commands=[])
diff --git a/tests/unit/modules/network/vyos/test_vyos_static_route.py b/tests/unit/modules/network/vyos/test_vyos_static_route.py
index e020ca5..762508c 100644
--- a/tests/unit/modules/network/vyos/test_vyos_static_route.py
+++ b/tests/unit/modules/network/vyos/test_vyos_static_route.py
@@ -21,7 +21,7 @@ from __future__ import absolute_import, division, print_function
__metaclass__ = type
from ansible_collections.vyos.vyos.tests.unit.compat.mock import patch
-from ansible_collections.vyos.vyos.plugins.modules import vyos_static_route
+from ansible.modules.network.vyos import _vyos_static_route
from ansible_collections.vyos.vyos.tests.unit.modules.utils import (
set_module_args,
)
@@ -30,18 +30,18 @@ from .vyos_module import TestVyosModule
class TestVyosStaticRouteModule(TestVyosModule):
- module = vyos_static_route
+ module = _vyos_static_route
def setUp(self):
super(TestVyosStaticRouteModule, self).setUp()
self.mock_get_config = patch(
- "ansible_collections.vyos.vyos.plugins.modules.vyos_static_route.get_config"
+ "ansible.modules.network.vyos._vyos_static_route.get_config"
)
self.get_config = self.mock_get_config.start()
self.mock_load_config = patch(
- "ansible_collections.vyos.vyos.plugins.modules.vyos_static_route.load_config"
+ "ansible.modules.network.vyos._vyos_static_route.load_config"
)
self.load_config = self.mock_load_config.start()
diff --git a/tests/unit/modules/network/vyos/test_vyos_static_routes.py b/tests/unit/modules/network/vyos/test_vyos_static_routes.py
new file mode 100644
index 0000000..3646d61
--- /dev/null
+++ b/tests/unit/modules/network/vyos/test_vyos_static_routes.py
@@ -0,0 +1,293 @@
+# (c) 2016 Red Hat Inc.
+#
+# This file is part of Ansible
+#
+# Ansible is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Ansible is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
+
+# Make coding more python3-ish
+from __future__ import absolute_import, division, print_function
+
+__metaclass__ = type
+
+from ansible_collections.vyos.vyos.tests.unit.compat.mock import patch
+from ansible_collections.vyos.vyos.plugins.modules import vyos_static_routes
+from ansible_collections.vyos.vyos.tests.unit.modules.utils import (
+ set_module_args,
+)
+from .vyos_module import TestVyosModule, load_fixture
+
+
+class TestVyosStaticRoutesModule(TestVyosModule):
+
+ module = vyos_static_routes
+
+ def setUp(self):
+ super(TestVyosStaticRoutesModule, self).setUp()
+ self.mock_get_config = patch(
+ "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.network.Config.get_config"
+ )
+ self.get_config = self.mock_get_config.start()
+
+ self.mock_load_config = patch(
+ "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.network.Config.load_config"
+ )
+ self.load_config = self.mock_load_config.start()
+
+ self.mock_get_resource_connection_config = patch(
+ "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.cfg.base.get_resource_connection"
+ )
+ self.get_resource_connection_config = (
+ self.mock_get_resource_connection_config.start()
+ )
+
+ self.mock_get_resource_connection_facts = patch(
+ "ansible_collections.ansible.netcommon.plugins.module_utils.network.common.facts.facts.get_resource_connection"
+ )
+ self.get_resource_connection_facts = (
+ self.mock_get_resource_connection_facts.start()
+ )
+
+ self.mock_execute_show_command = patch(
+ "ansible_collections.vyos.vyos.plugins.module_utils.network.vyos.facts.static_routes.static_routes.Static_routesFacts.get_device_data"
+ )
+ self.execute_show_command = self.mock_execute_show_command.start()
+
+ def tearDown(self):
+ super(TestVyosStaticRoutesModule, self).tearDown()
+ self.mock_get_resource_connection_config.stop()
+ self.mock_get_resource_connection_facts.stop()
+ self.mock_get_config.stop()
+ self.mock_load_config.stop()
+ self.mock_execute_show_command.stop()
+
+ def load_fixtures(self, commands=None):
+ def load_from_file(*args, **kwargs):
+ return load_fixture("vyos_static_routes_config.cfg")
+
+ self.execute_show_command.side_effect = load_from_file
+
+ def test_vyos_static_routes_merged(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ address_families=[
+ dict(
+ afi="ipv4",
+ routes=[
+ dict(
+ dest="192.0.2.48/28",
+ next_hops=[
+ dict(
+ forward_router_address="192.0.2.9"
+ ),
+ dict(
+ forward_router_address="192.0.2.10"
+ ),
+ ],
+ )
+ ],
+ )
+ ]
+ )
+ ],
+ state="merged",
+ )
+ )
+ commands = [
+ "set protocols static route 192.0.2.48/28",
+ "set protocols static route 192.0.2.48/28 next-hop '192.0.2.9'",
+ "set protocols static route 192.0.2.48/28 next-hop '192.0.2.10'",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_static_routes_merged_idempotent(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ address_families=[
+ dict(
+ afi="ipv4",
+ routes=[
+ dict(
+ dest="192.0.2.32/28",
+ next_hops=[
+ dict(
+ forward_router_address="192.0.2.9"
+ ),
+ dict(
+ forward_router_address="192.0.2.10"
+ ),
+ ],
+ )
+ ],
+ )
+ ]
+ )
+ ],
+ state="merged",
+ )
+ )
+ self.execute_module(changed=False, commands=[])
+
+ def test_vyos_static_routes_replaced(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ address_families=[
+ dict(
+ afi="ipv4",
+ routes=[
+ dict(
+ dest="192.0.2.48/28",
+ next_hops=[
+ dict(
+ forward_router_address="192.0.2.9"
+ ),
+ dict(
+ forward_router_address="192.0.2.10"
+ ),
+ ],
+ )
+ ],
+ )
+ ]
+ )
+ ],
+ state="replaced",
+ )
+ )
+ commands = [
+ "set protocols static route 192.0.2.48/28",
+ "set protocols static route 192.0.2.48/28 next-hop '192.0.2.9'",
+ "set protocols static route 192.0.2.48/28 next-hop '192.0.2.10'",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_static_routes_replaced_idempotent(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ address_families=[
+ dict(
+ afi="ipv4",
+ routes=[
+ dict(
+ dest="192.0.2.32/28",
+ next_hops=[
+ dict(
+ forward_router_address="192.0.2.9"
+ ),
+ dict(
+ forward_router_address="192.0.2.10"
+ ),
+ ],
+ )
+ ],
+ )
+ ]
+ )
+ ],
+ state="replaced",
+ )
+ )
+
+ self.execute_module(changed=False, commands=[])
+
+ def test_vyos_static_routes_overridden(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ address_families=[
+ dict(
+ afi="ipv4",
+ routes=[
+ dict(
+ dest="192.0.2.48/28",
+ next_hops=[
+ dict(
+ forward_router_address="192.0.2.9"
+ ),
+ dict(
+ forward_router_address="192.0.2.10"
+ ),
+ ],
+ )
+ ],
+ )
+ ]
+ )
+ ],
+ state="overridden",
+ )
+ )
+ commands = [
+ "delete protocols static route 192.0.2.32/28",
+ "set protocols static route 192.0.2.48/28",
+ "set protocols static route 192.0.2.48/28 next-hop '192.0.2.9'",
+ "set protocols static route 192.0.2.48/28 next-hop '192.0.2.10'",
+ ]
+ self.execute_module(changed=True, commands=commands)
+
+ def test_vyos_static_routes_overridden_idempotent(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ address_families=[
+ dict(
+ afi="ipv4",
+ routes=[
+ dict(
+ dest="192.0.2.32/28",
+ next_hops=[
+ dict(
+ forward_router_address="192.0.2.9"
+ ),
+ dict(
+ forward_router_address="192.0.2.10"
+ ),
+ ],
+ )
+ ],
+ )
+ ]
+ )
+ ],
+ state="overridden",
+ )
+ )
+ self.execute_module(changed=False, commands=[])
+
+ def test_vyos_static_routes_deleted(self):
+ set_module_args(
+ dict(
+ config=[
+ dict(
+ address_families=[
+ dict(
+ afi="ipv4", routes=[dict(dest="192.0.2.32/28")]
+ )
+ ]
+ )
+ ],
+ state="deleted",
+ )
+ )
+ commands = ["delete protocols static route 192.0.2.32/28"]
+ self.execute_module(changed=True, commands=commands)