Signed-off-by: David Gibson
---
test/tasst/dhcp.py | 114 +++++++++++++++++++++++++++++++++++++++++++
test/tasst/dhcpv6.py | 74 ++++++++++++++++++++++++++++
2 files changed, 188 insertions(+)
create mode 100644 test/tasst/dhcp.py
create mode 100644 test/tasst/dhcpv6.py
diff --git a/test/tasst/dhcp.py b/test/tasst/dhcp.py
new file mode 100644
index 00000000..b8c18b8f
--- /dev/null
+++ b/test/tasst/dhcp.py
@@ -0,0 +1,114 @@
+#! /usr/bin/env avocado-runner-avocado-classless
+
+# SPDX-License-Identifier: GPL-2.0-or-later
+#
+# Copyright Red Hat
+# Author: David Gibson
+
+"""
+Test A Simple Socket Transport
+
+dhcp.py - Helpers for testing DHCP
+"""
+
+import contextlib
+import ipaddress
+import os
+import tempfile
+
+from avocado_classless.test import assert_eq, test_output
+
+from tasst.address import IpiAllocator, TEST_NET_1
+from tasst.nstool import unshare_site
+
+
+DHCLIENT = '/sbin/dhclient'
+
+
+@contextlib.contextmanager
+def dhclient(site, ifname, ipv='4'):
+ site.require_cmds(DHCLIENT)
+
+ with tempfile.TemporaryDirectory() as tmpdir:
+ pidfile = os.path.join(tmpdir, 'dhclient.pid')
+ leasefile = os.path.join(tmpdir, 'dhclient.leases')
+
+ # We need '-nc' because we may be running with
+ # capabilities but not UID 0. Without -nc dhclient drops
+ # capabilities before invoking dhclient-script, so it's
+ # unable to actually configure the interface
+ opts = f'-{ipv} -v -nc -pf {pidfile} -lf {leasefile} {ifname}'
+ with site.daemon(f'{DHCLIENT} {opts}', sudo=True, pidfile=pidfile):
+ yield
+
+
+def test_dhcp(ifname, expected_addr, gateway, mtu):
+ def dec(setupfn):
+ def test_addr(setup):
+ with setup as site, dhclient(site, ifname):
+ (actual_addr,) = site.addrs(ifname, family='inet',
+ scope='global')
+ assert_eq(actual_addr.ip, expected_addr)
+
+ def test_route(setup):
+ with setup as site, dhclient(site, ifname):
+ (defroute,) = site.routes4(dst='default')
+ assert_eq(ipaddress.ip_address(defroute['gateway']), gateway)
+
+ def test_mtu(setup):
+ with setup as site, dhclient(site, ifname):
+ assert_eq(site.mtu(ifname), mtu)
+
+ return test_output(test_addr, test_route, test_mtu)(setupfn)
+
+ return dec
+
+
+DHCPD = 'dhcpd'
+SUBNET = TEST_NET_1
+ipa = IpiAllocator(SUBNET)
+(SERVER_IP4,) = ipa.next_ipis()
+(CLIENT_IP4,) = ipa.next_ipis()
+IFNAME = 'clientif'
+
+
+@contextlib.contextmanager
+def setup_dhcpd_common(ifname, server_ifname):
+ with unshare_site('client', '-Un') as client, \
+ unshare_site('server', '-n', parent=client, sudo=True) as server:
+ server.require_cmds(DHCPD)
+
+ client.veth(ifname, server_ifname, server)
+
+ with tempfile.TemporaryDirectory() as tmpdir:
+ yield (client, server, tmpdir)
+
+
+@test_dhcp(IFNAME, CLIENT_IP4.ip, SERVER_IP4.ip, 1500)
+@contextlib.contextmanager
+def setup_dhcpd():
+ server_ifname = 'serverif'
+
+ with setup_dhcpd_common(IFNAME, server_ifname) as (client, server, tmpdir):
+ # Configure dhcpd
+ confpath = os.path.join(tmpdir, 'dhcpd.conf')
+ open(confpath, 'w', encoding='UTF-8').write(
+ f'''subnet {SUBNET.network_address} netmask {SUBNET.netmask} {{
+ option routers {SERVER_IP4.ip};
+ range {CLIENT_IP4.ip} {CLIENT_IP4.ip};
+ }}'''
+ )
+ pidfile = os.path.join(tmpdir, 'dhcpd.pid')
+ leasepath = os.path.join(tmpdir, 'dhcpd.leases')
+ open(leasepath, 'wb').write(b'')
+
+ server.ifup('lo')
+ server.ifup(server_ifname, SERVER_IP4)
+
+ opts = f'-f -d -4 -cf {confpath} -lf {leasepath} -pf {pidfile}'
+ server.fg(f'{DHCPD} -t {opts}') # test config
+ with server.bg(f'{DHCPD} {opts}', sudo=True, ignore_status=True,
+ pidfile=pidfile):
+ # Configure the client
+ client.ifup('lo')
+ yield client
diff --git a/test/tasst/dhcpv6.py b/test/tasst/dhcpv6.py
new file mode 100644
index 00000000..816c0ea9
--- /dev/null
+++ b/test/tasst/dhcpv6.py
@@ -0,0 +1,74 @@
+#! /usr/bin/env avocado-runner-avocado-classless
+
+# SPDX-License-Identifier: GPL-2.0-or-later
+#
+# Copyright Red Hat
+# Author: David Gibson
+
+"""
+Test A Simple Socket Transport
+
+dhcpv6.py - Helpers for testing DHCPv6
+"""
+
+import contextlib
+import os
+
+from avocado_classless.test import assert_in, test_output
+
+from tasst.address import IpiAllocator, TEST_NET6_TASST_A
+from tasst.dhcp import dhclient, setup_dhcpd_common
+
+
+def dhclientv6(site, ifname):
+ return dhclient(site, ifname, '6')
+
+
+def test_dhcpv6(ifname, expected_addr):
+ def dec(setupfn):
+ def test_addr(setup):
+ with setup as site, dhclientv6(site, ifname):
+ addrs = [a.ip for a in site.addrs(ifname, family='inet6',
+ scope='global')]
+ assert_in(expected_addr, addrs) # Might also have a SLAAC address
+
+ return test_output(test_addr)(setupfn)
+ return dec
+
+
+DHCPD = 'dhcpd'
+SUBNET = TEST_NET6_TASST_A
+ipa = IpiAllocator(SUBNET)
+(SERVER_IP6,) = ipa.next_ipis()
+(CLIENT_IP6,) = ipa.next_ipis()
+IFNAME = 'clientif'
+
+
+@test_dhcpv6(IFNAME, CLIENT_IP6.ip)
+@contextlib.contextmanager
+def setup_dhcpdv6():
+ server_ifname = 'serverif'
+
+ with setup_dhcpd_common(IFNAME, server_ifname) as (client, server, tmpdir):
+ # Sort out link local addressing
+ server.ifup('lo')
+ server.ifup(server_ifname, SERVER_IP6)
+ client.ifup('lo')
+ client.ifup(IFNAME)
+ server.addr_wait(server_ifname, family='inet6', scope='link')
+
+ # Configure the DHCP server
+ confpath = os.path.join(tmpdir, 'dhcpd.conf')
+ open(confpath, 'w', encoding='UTF-8').write(
+ f'''subnet6 {SUBNET} {{
+ range6 {CLIENT_IP6.ip} {CLIENT_IP6.ip};
+ }}''')
+ pidfile = os.path.join(tmpdir, 'dhcpd.pid')
+ leasepath = os.path.join(tmpdir, 'dhcpd.leases')
+ open(leasepath, 'wb').write(b'')
+
+ opts = f'-f -d -6 -cf {confpath} -lf {leasepath} -pf {pidfile}'
+ server.fg(f'{DHCPD} -t {opts}') # test config
+ with server.bg(f'{DHCPD} {opts}', sudo=True, ignore_status=True,
+ pidfile=pidfile):
+ yield client
--
2.41.0