mirror of
https://github.com/retspen/webvirtcloud
synced 2026-11-08 11:16:47 +00:00
bdc8a6755c
Tests that build vrtManager objects and touch no view, form or model of an app lived in that app; vrtManager/tests already holds such tests, so the convention was split. They move to vrtManager/tests: the firmware, CPU mode and PCIe port definitions of a new VM (test_create), USB and PCI host devices, the console listener, host CPU reads, interface reads, deletes and XML, network filters, PCI passthrough rules, node devices, network changes, addresses and leases, pool XML, secrets, and the live tests that use vrtManager only (disk resize, ISO, snapshots, XML, live clone). livetest.py, the live-test helper six apps import, and the network test driver move with them. Others go to their app: the quota with a host down to instances, the 401/403 Logs rows (webvirtcloud's middleware) to webvirtcloud, the VM page's tab script test to instances, novncd's own tests out of the console views tests, the compute lock and input validation tests out of the computes views tests, the interface views out of the forms tests. instances/tests/test_views.py, entirely live, is test_live_views.py. Test modules imported fixtures from each other; shared ones are now in vrtManager/tests/fakes.py (migration fakes, node devices, IOMMU groups, leases, a NAT network, a Ceph secret), interfaces/tests/fakes.py, console/tests/daemons.py (novncd loaded once), livetest.wait_for and instances/tests/create_page.BODY. Every live test checks livetest.enabled(). The suite runs the same 2097 tests.
116 lines
5.3 KiB
Python
116 lines
5.3 KiB
Python
import unittest
|
|
|
|
from lxml import etree
|
|
from vrtManager.network import ip_networks, network_size
|
|
from django.test import SimpleTestCase
|
|
|
|
|
|
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()
|
|
|
|
|
|
class AddressTests(SimpleTestCase):
|
|
def test_an_address_without_a_prefix_takes_libvirts(self):
|
|
tree = etree.fromstring("<network><ip address='192.0.2.1'/><ip address='10.1.2.3'/><ip family='ipv6' address='2001:db8::1'/></network>")
|
|
self.assertEqual([str(ip["subnet"]) for ip in ip_networks(tree)], ["192.0.2.0/24", "10.0.0.0/8", "2001:db8::/64"])
|
|
|
|
def test_every_way_libvirt_writes_an_address(self):
|
|
compact = "<network><ip address='192.0.2.1' netmask='255.255.255.0'/></network>"
|
|
static = ("<network><ip address='192.0.2.1' prefix='24'><dhcp><host mac='52:54:00:00:00:01' ip='192.0.2.5'/>"
|
|
"</dhcp></ip><ip family='ipv6' address='2001:db8::1' prefix='64'/></network>")
|
|
(ip,) = ip_networks(etree.fromstring(compact))
|
|
self.assertEqual((ip["index"], ip["family"], str(ip["subnet"]), ip["dhcp"]), (0, "ipv4", "192.0.2.0/24", False))
|
|
v4, v6 = ip_networks(etree.fromstring(static))
|
|
# DHCP with fixed addresses only is DHCP too
|
|
self.assertEqual((v4["dhcp"], v4["range"], v4["hosts"][0]["ip"]), (True, None, "192.0.2.5"))
|
|
self.assertEqual((v6["index"], v6["family"], str(v6["subnet"])), (1, "ipv6", "2001:db8::/64"))
|