diff options
| author | Christian Poessinger <christian@poessinger.com> | 2020-11-01 13:44:55 +0100 |
|---|---|---|
| committer | Christian Poessinger <christian@poessinger.com> | 2020-11-01 13:51:44 +0100 |
| commit | 0bb7a7d32b6182064acf3203050464f22815ef5d (patch) | |
| tree | 1c80c6e94f60a15fd0d0743339137478e03c5a59 /src/tests | |
| parent | eb39427f4ff8bf0f22ce583de0f804cfb21dfdaa (diff) | |
| download | vyos-1x-0bb7a7d32b6182064acf3203050464f22815ef5d.tar.gz vyos-1x-0bb7a7d32b6182064acf3203050464f22815ef5d.zip | |
test: extend validate testcases
Test additional network combinations of address and prefixes.
Diffstat (limited to 'src/tests')
| -rw-r--r-- | src/tests/test_jinja_filters.py | 7 | ||||
| -rw-r--r-- | src/tests/test_validate.py | 8 |
2 files changed, 12 insertions, 3 deletions
diff --git a/src/tests/test_jinja_filters.py b/src/tests/test_jinja_filters.py index 57ff11070..17219802a 100644 --- a/src/tests/test_jinja_filters.py +++ b/src/tests/test_jinja_filters.py @@ -16,6 +16,7 @@ from unittest import TestCase +from ipaddress import ip_network from vyos.template import vyos_address_from_cidr from vyos.template import vyos_netmask_from_cidr from vyos.template import vyos_ipv4 @@ -26,9 +27,9 @@ class TestTeamplteHelpers(TestCase): pass def test_helpers_from_cidr(self): - network = '192.0.2.0/26' - self.assertEqual(vyos_address_from_cidr(network), '192.0.2.0') - self.assertEqual(vyos_netmask_from_cidr(network), '255.255.255.192') + network_v4 = '192.0.2.0/26' + self.assertEqual(vyos_address_from_cidr(network_v4), str(ip_network(network_v4).network_address)) + self.assertEqual(vyos_netmask_from_cidr(network_v4), str(ip_network(network_v4).netmask)) def test_helpers_ipv4(self): self.assertTrue(vyos_ipv4('192.0.2.1')) diff --git a/src/tests/test_validate.py b/src/tests/test_validate.py index 5e2c751ca..e9fe185ed 100644 --- a/src/tests/test_validate.py +++ b/src/tests/test_validate.py @@ -28,12 +28,20 @@ class TestVyOSValidate(TestCase): def test_is_ipv4(self): self.assertTrue(vyos.validate.is_ipv4('192.0.2.1')) + self.assertTrue(vyos.validate.is_ipv4('192.0.2.0/24')) + self.assertTrue(vyos.validate.is_ipv4('192.0.2.1/32')) + self.assertFalse(vyos.validate.is_ipv4('2001:db8::1')) + self.assertFalse(vyos.validate.is_ipv4('2001:db8::/64')) self.assertFalse(vyos.validate.is_ipv4('VyOS')) def test_is_ipv6(self): self.assertFalse(vyos.validate.is_ipv6('192.0.2.1')) + self.assertFalse(vyos.validate.is_ipv6('192.0.2.0/24')) + self.assertFalse(vyos.validate.is_ipv6('192.0.2.1/32')) self.assertTrue(vyos.validate.is_ipv6('2001:db8::1')) + self.assertTrue(vyos.validate.is_ipv6('2001:db8::/64')) + self.assertTrue(vyos.validate.is_ipv6('2001:db8::1/64')) self.assertFalse(vyos.validate.is_ipv6('VyOS')) def test_is_ipv6_link_local(self): |