Add support for IPv6 `address_group` and `network_group` (#202)

Add support for IPv6 `address_group` and `network_group`

SUMMARY
This adds support for ipv6 in network and address groups by adding an afi parameter, but defaulting it to ipv4 for backwards compatibility.
Fixes #137.
ISSUE TYPE

Feature Pull Request

COMPONENT NAME
vyos_firewall_global

Reviewed-by: GomathiselviS <None>
Reviewed-by: Shawn Wilsher <None>
Reviewed-by: None <None>

3 files changed, 101 insertions, 8 deletions

diff --git a/tests/integration/targets/vyos_firewall_global/vars/main.yaml b/tests/integration/targets/vyos_firewall_global/vars/main.yaml
index 4a1e7a8..faa0ff1 100644
--- a/tests/integration/targets/vyos_firewall_global/vars/main.yaml
+++ b/tests/integration/targets/vyos_firewall_global/vars/main.yaml
@@ -35,11 +35,13 @@ merged:
             - address: 192.0.1.5
           description: This group has the Management hosts address list
           name: MGMT-HOSTS
+          afi: ipv4
       network_group:
         - members:
             - address: 192.0.1.0/24
           description: This group has the Management network addresses
           name: MGMT
+          afi: ipv4
     log_martians: true
     ping:
       all: true
@@ -88,11 +90,13 @@ populate:
           - address: 192.0.1.1
           - address: 192.0.1.3
           - address: 192.0.1.5
+        afi: ipv4
     network_group:
       - name: MGMT
         description: This group has the Management network addresses
         members:
           - address: 192.0.1.0/24
+        afi: ipv4
 replaced:
   commands:
     - delete firewall group address-group MGMT-HOSTS
@@ -116,17 +120,20 @@ replaced:
             - address: 192.0.3.2
           description: Sales office hosts address list
           name: ENG-HOSTS
+          afi: ipv4
         - members:
             - address: 192.0.2.1
             - address: 192.0.2.2
             - address: 192.0.2.3
           description: Sales office hosts address list
           name: SALES-HOSTS
+          afi: ipv4
       network_group:
         - members:
             - address: 192.0.1.0/24
           description: This group has the Management network addresses
           name: MGMT
+          afi: ipv4
     log_martians: true
     ping:
       all: true
@@ -210,8 +217,10 @@ round_trip:
             - address: 192.0.1.1
             - address: 192.0.1.3
             - address: 192.0.1.5
+          afi: ipv4
       network_group:
         - name: MGMT
           description: This group has the Management network addresses
           members:
             - address: 192.0.1.0/24
+          afi: ipv4
diff --git a/tests/unit/modules/network/vyos/fixtures/vyos_firewall_global_config.cfg b/tests/unit/modules/network/vyos/fixtures/vyos_firewall_global_config.cfg
index 6c275a3..16b7a14 100644
--- a/tests/unit/modules/network/vyos/fixtures/vyos_firewall_global_config.cfg
+++ b/tests/unit/modules/network/vyos/fixtures/vyos_firewall_global_config.cfg
@@ -2,7 +2,12 @@ set firewall group address-group RND-HOSTS address 192.0.2.1
 set firewall group address-group RND-HOSTS address 192.0.2.3
 set firewall group address-group RND-HOSTS address 192.0.2.5
 set firewall group address-group RND-HOSTS description 'This group has the Management hosts address lists'
+set firewall group ipv6-address-group LOCAL-v6 address ::1
+set firewall group ipv6-address-group LOCAL-v6 address fdec:2503:89d6:59b3::1
+set firewall group ipv6-address-group LOCAL-v6 description 'This group has the hosts address lists of this machine'
 set firewall group network-group RND network 192.0.2.0/24
 set firewall group network-group RND description 'This group has the Management network addresses'
+set firewall group ipv6-network-group UNIQUE-LOCAL-v6 network fc00::/7
+set firewall group ipv6-network-group UNIQUE-LOCAL-v6 description 'This group encompasses the ULA address space in IPv6'
 set firewall group port-group SSH port 22
 set firewall group port-group SSH description 'This group has the ssh ports'
diff --git a/tests/unit/modules/network/vyos/test_vyos_firewall_global.py b/tests/unit/modules/network/vyos/test_vyos_firewall_global.py
index fa40d7c..0e62be8 100644
--- a/tests/unit/modules/network/vyos/test_vyos_firewall_global.py
+++ b/tests/unit/modules/network/vyos/test_vyos_firewall_global.py
@@ -106,6 +106,7 @@ class TestVyosFirewallRulesModule(TestVyosModule):
                     group=dict(
                         address_group=[
                             dict(
+                                afi="ipv4",
                                 name="MGMT-HOSTS",
                                 description="This group has the Management hosts address lists",
                                 members=[
@@ -113,14 +114,32 @@ class TestVyosFirewallRulesModule(TestVyosModule):
                                     dict(address="192.0.1.3"),
                                     dict(address="192.0.1.5"),
                                 ],
-                            )
+                            ),
+                            dict(
+                                afi="ipv6",
+                                name="GOOGLE-DNS-v6",
+                                members=[
+                                    dict(address="2001:4860:4860::8888"),
+                                    dict(address="2001:4860:4860::8844"),
+                                ],
+                            ),
                         ],
                         network_group=[
                             dict(
+                                afi="ipv4",
                                 name="MGMT",
                                 description="This group has the Management network addresses",
                                 members=[dict(address="192.0.1.0/24")],
-                            )
+                            ),
+                            dict(
+                                afi="ipv6",
+                                name="DOCUMENTATION-v6",
+                                description="IPv6 Addresses reserved for documentation per RFC 3849",
+                                members=[
+                                    dict(address="2001:0DB8::/32"),
+                                    dict(address="3FFF:FFFF::/32"),
+                                ],
+                            ),
                         ],
                         port_group=[
                             dict(
@@ -140,9 +159,16 @@ class TestVyosFirewallRulesModule(TestVyosModule):
             "set firewall group address-group MGMT-HOSTS address 192.0.1.5",
             "set firewall group address-group MGMT-HOSTS description 'This group has the Management hosts address lists'",
             "set firewall group address-group MGMT-HOSTS",
+            "set firewall group ipv6-address-group GOOGLE-DNS-v6 address 2001:4860:4860::8888",
+            "set firewall group ipv6-address-group GOOGLE-DNS-v6 address 2001:4860:4860::8844",
+            "set firewall group ipv6-address-group GOOGLE-DNS-v6",
             "set firewall group network-group MGMT network 192.0.1.0/24",
             "set firewall group network-group MGMT description 'This group has the Management network addresses'",
             "set firewall group network-group MGMT",
+            "set firewall group ipv6-network-group DOCUMENTATION-v6 network 2001:0DB8::/32",
+            "set firewall group ipv6-network-group DOCUMENTATION-v6 network 3FFF:FFFF::/32",
+            "set firewall group ipv6-network-group DOCUMENTATION-v6 description 'IPv6 Addresses reserved for documentation per RFC 3849'",
+            "set firewall group ipv6-network-group DOCUMENTATION-v6",
             "set firewall group port-group TELNET port 23",
             "set firewall group port-group TELNET description 'This group has the telnet ports'",
             "set firewall group port-group TELNET",
@@ -169,6 +195,7 @@ class TestVyosFirewallRulesModule(TestVyosModule):
                     group=dict(
                         address_group=[
                             dict(
+                                afi="ipv4",
                                 name="RND-HOSTS",
                                 description="This group has the Management hosts address lists",
                                 members=[
@@ -176,14 +203,30 @@ class TestVyosFirewallRulesModule(TestVyosModule):
                                     dict(address="192.0.2.3"),
                                     dict(address="192.0.2.5"),
                                 ],
-                            )
+                            ),
+                            dict(
+                                afi="ipv6",
+                                name="LOCAL-v6",
+                                description="This group has the hosts address lists of this machine",
+                                members=[
+                                    dict(address="::1"),
+                                    dict(address="fdec:2503:89d6:59b3::1"),
+                                ],
+                            ),
                         ],
                         network_group=[
                             dict(
+                                afi="ipv4",
                                 name="RND",
                                 description="This group has the Management network addresses",
                                 members=[dict(address="192.0.2.0/24")],
-                            )
+                            ),
+                            dict(
+                                afi="ipv6",
+                                name="UNIQUE-LOCAL-v6",
+                                description="This group encompasses the ULA address space in IPv6",
+                                members=[dict(address="fc00::/7")],
+                            ),
                         ],
                         port_group=[
                             dict(
@@ -206,6 +249,7 @@ class TestVyosFirewallRulesModule(TestVyosModule):
                     group=dict(
                         address_group=[
                             dict(
+                                afi="ipv4",
                                 name="RND-HOSTS",
                                 description="This group has the Management hosts address lists",
                                 members=[
@@ -213,14 +257,30 @@ class TestVyosFirewallRulesModule(TestVyosModule):
                                     dict(address="192.0.2.7"),
                                     dict(address="192.0.2.9"),
                                 ],
-                            )
+                            ),
+                            dict(
+                                afi="ipv6",
+                                name="LOCAL-v6",
+                                description="This group has the hosts address lists of this machine",
+                                members=[
+                                    dict(address="::1"),
+                                    dict(address="fdec:2503:89d6:59b3::2"),
+                                ],
+                            ),
                         ],
                         network_group=[
                             dict(
+                                afi="ipv4",
                                 name="RND",
                                 description="This group has the Management network addresses",
                                 members=[dict(address="192.0.2.0/24")],
-                            )
+                            ),
+                            dict(
+                                afi="ipv6",
+                                name="UNIQUE-LOCAL-v6",
+                                description="This group encompasses the ULA address space in IPv6",
+                                members=[dict(address="fc00::/7")],
+                            ),
                         ],
                         port_group=[
                             dict(
@@ -239,6 +299,8 @@ class TestVyosFirewallRulesModule(TestVyosModule):
             "delete firewall group address-group RND-HOSTS address 192.0.2.5",
             "set firewall group address-group RND-HOSTS address 192.0.2.7",
             "set firewall group address-group RND-HOSTS address 192.0.2.9",
+            "delete firewall group ipv6-address-group LOCAL-v6 address fdec:2503:89d6:59b3::1",
+            "set firewall group ipv6-address-group LOCAL-v6 address fdec:2503:89d6:59b3::2",
             "delete firewall group port-group SSH port 22",
             "set firewall group port-group SSH port 2222",
         ]
@@ -251,6 +313,7 @@ class TestVyosFirewallRulesModule(TestVyosModule):
                     group=dict(
                         address_group=[
                             dict(
+                                afi="ipv4",
                                 name="RND-HOSTS",
                                 description="This group has the Management hosts address lists",
                                 members=[
@@ -258,14 +321,30 @@ class TestVyosFirewallRulesModule(TestVyosModule):
                                     dict(address="192.0.2.3"),
                                     dict(address="192.0.2.5"),
                                 ],
-                            )
+                            ),
+                            dict(
+                                afi="ipv6",
+                                name="LOCAL-v6",
+                                description="This group has the hosts address lists of this machine",
+                                members=[
+                                    dict(address="::1"),
+                                    dict(address="fdec:2503:89d6:59b3::1"),
+                                ],
+                            ),
                         ],
                         network_group=[
                             dict(
+                                afi="ipv4",
                                 name="RND",
                                 description="This group has the Management network addresses",
                                 members=[dict(address="192.0.2.0/24")],
-                            )
+                            ),
+                            dict(
+                                afi="ipv6",
+                                name="UNIQUE-LOCAL-v6",
+                                description="This group encompasses the ULA address space in IPv6",
+                                members=[dict(address="fc00::/7")],
+                            ),
                         ],
                         port_group=[
                             dict(