mirror of
https://github.com/retspen/webvirtcloud
synced 2026-11-08 19:26:48 +00:00
42ab293dc5
- 25 vrtManager test modules configured settings themselves when Django had none (MAC_OUI, and in test_network_ip a fake django module): under manage.py test settings are always configured, so it never ran. - webvirtcloud/tests/test_common_tags.py tested no tag: it fetched an unknown URL, as test_error_pages and test_logging already do. - .gitignore covers the parallel runs' test_db_N.sqlite3 files too.
98 lines
4.1 KiB
Python
98 lines
4.1 KiB
Python
import unittest
|
|
|
|
from lxml import etree
|
|
from vrtManager.network import ip_networks, network_size
|
|
|
|
|
|
class TestNetworkSize(unittest.TestCase):
|
|
"""Unit tests for the network_size function."""
|
|
|
|
def test_ipv4_slash_24_with_dhcp(self):
|
|
gateway, mask, dhcp_pool = network_size("192.168.100.0/24", dhcp=True)
|
|
self.assertEqual(gateway, "192.168.100.1")
|
|
self.assertEqual(mask, "255.255.255.0")
|
|
self.assertIsNotNone(dhcp_pool)
|
|
self.assertEqual(str(dhcp_pool[0]), "192.168.100.2")
|
|
self.assertEqual(str(dhcp_pool[1]), "192.168.100.254")
|
|
|
|
def test_ipv4_slash_24_without_dhcp(self):
|
|
gateway, mask, dhcp_pool = network_size("192.168.100.0/24", dhcp=False)
|
|
self.assertEqual(gateway, "192.168.100.1")
|
|
self.assertEqual(mask, "255.255.255.0")
|
|
self.assertIsNone(dhcp_pool)
|
|
|
|
def test_ipv4_slash_16_with_dhcp(self):
|
|
gateway, mask, dhcp_pool = network_size("172.16.0.0/16", dhcp=True)
|
|
self.assertEqual(gateway, "172.16.0.1")
|
|
self.assertEqual(mask, "255.255.0.0")
|
|
self.assertIsNotNone(dhcp_pool)
|
|
self.assertEqual(str(dhcp_pool[0]), "172.16.0.2")
|
|
self.assertEqual(str(dhcp_pool[1]), "172.16.255.254")
|
|
|
|
def test_ipv4_slash_28_with_dhcp(self):
|
|
gateway, mask, dhcp_pool = network_size("10.0.0.0/28", dhcp=True)
|
|
self.assertEqual(gateway, "10.0.0.1")
|
|
self.assertEqual(mask, "255.255.255.240")
|
|
self.assertIsNotNone(dhcp_pool)
|
|
self.assertEqual(str(dhcp_pool[0]), "10.0.0.2")
|
|
self.assertEqual(str(dhcp_pool[1]), "10.0.0.14")
|
|
|
|
def test_ipv4_slash_30_with_dhcp(self):
|
|
gateway, mask, dhcp_pool = network_size("10.0.0.0/30", dhcp=True)
|
|
self.assertEqual(gateway, "10.0.0.1")
|
|
self.assertEqual(mask, "255.255.255.252")
|
|
self.assertIsNotNone(dhcp_pool)
|
|
self.assertEqual(str(dhcp_pool[0]), "10.0.0.2")
|
|
self.assertEqual(str(dhcp_pool[1]), "10.0.0.2")
|
|
|
|
def test_ipv6_slash_64_with_dhcp(self):
|
|
gateway, mask, dhcp_pool = network_size("2001:db8:ca2:2::/64", dhcp=True)
|
|
self.assertEqual(gateway, "2001:db8:ca2:2::1")
|
|
self.assertEqual(mask, "64")
|
|
self.assertIsNotNone(dhcp_pool)
|
|
self.assertEqual(str(dhcp_pool[0]), "2001:db8:ca2:2::100")
|
|
self.assertEqual(str(dhcp_pool[1]), "2001:db8:ca2:2::1ff")
|
|
|
|
def test_ipv6_slash_64_without_dhcp(self):
|
|
gateway, mask, dhcp_pool = network_size("2001:db8:ca2:2::/64", dhcp=False)
|
|
self.assertEqual(gateway, "2001:db8:ca2:2::1")
|
|
self.assertEqual(mask, "64")
|
|
self.assertIsNone(dhcp_pool)
|
|
|
|
def test_invalid_subnet(self):
|
|
with self.assertRaises(ValueError):
|
|
network_size("invalid.subnet/99")
|
|
|
|
|
|
class TestIpNetworks(unittest.TestCase):
|
|
"""A network's addresses (ip_networks): subnet, DHCP range and fixed addresses of each family."""
|
|
|
|
DUAL_STACK_XML = """<network>
|
|
<name>default</name>
|
|
<ip address="192.168.100.1" netmask="255.255.255.0">
|
|
<dhcp>
|
|
<range start="192.168.100.2" end="192.168.100.254"/>
|
|
<host mac="52:54:00:12:34:56" name="vm1" ip="192.168.100.50"/>
|
|
</dhcp>
|
|
</ip>
|
|
<ip family="ipv6" address="2001:db8:ca2:2::1" prefix="64">
|
|
<dhcp>
|
|
<range start="2001:db8:ca2:2::100" end="2001:db8:ca2:2::1ff"/>
|
|
<host id="0:1:0:1" name="vm2" ip="2001:db8:ca2:2::50"/>
|
|
</dhcp>
|
|
</ip>
|
|
</network>"""
|
|
|
|
def test_dual_stack(self):
|
|
v4, v6 = ip_networks(etree.fromstring(self.DUAL_STACK_XML))
|
|
self.assertEqual((str(v4["subnet"]), v4["range"]), ("192.168.100.0/24", ("192.168.100.2", "192.168.100.254")))
|
|
self.assertEqual(v4["hosts"], [{"ip": "192.168.100.50", "mac": "52:54:00:12:34:56", "id": None, "name": "vm1"}])
|
|
self.assertEqual((str(v6["subnet"]), v6["range"]), ("2001:db8:ca2:2::/64", ("2001:db8:ca2:2::100", "2001:db8:ca2:2::1ff")))
|
|
self.assertEqual(v6["hosts"][0]["id"], "0:1:0:1")
|
|
|
|
def test_no_address(self):
|
|
self.assertEqual(ip_networks(etree.fromstring("<network><name>isolated</name><bridge name='virbr1'/></network>")), [])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|