public inbox for passt-dev@passt.top
 help / color / mirror / code / Atom feed
From: David Gibson <david@gibson.dropbear.id.au>
To: passt-dev@passt.top, Stefano Brivio <sbrivio@redhat.com>
Cc: Cleber Rosa <crosa@redhat.com>,
	jarichte@redhat.com, David Gibson <david@gibson.dropbear.id.au>
Subject: [PATCH v3 18/20] avocado/tasst: Helpers for testing DHCP & DHCPv6 behaviour
Date: Wed, 31 May 2023 11:58:47 +1000	[thread overview]
Message-ID: <20230531015849.3229596-19-david@gibson.dropbear.id.au> (raw)
In-Reply-To: <20230531015849.3229596-1-david@gibson.dropbear.id.au>

Signed-off-by: David Gibson <david@gibson.dropbear.id.au>
---
 avocado/tasst/dhcp.py   | 152 ++++++++++++++++++++++++++++++++++++++++
 avocado/tasst/dhcpv6.py | 135 +++++++++++++++++++++++++++++++++++
 2 files changed, 287 insertions(+)
 create mode 100644 avocado/tasst/dhcp.py
 create mode 100644 avocado/tasst/dhcpv6.py

diff --git a/avocado/tasst/dhcp.py b/avocado/tasst/dhcp.py
new file mode 100644
index 0000000..be226e4
--- /dev/null
+++ b/avocado/tasst/dhcp.py
@@ -0,0 +1,152 @@
+#! /usr/bin/python3
+
+# SPDX-License-Identifier: GPL-2.0-or-later
+#
+# tasst - Test A Simple Socket Transport
+#         library of test helpers for passt & pasta
+#
+# tasst/dhcp.py - Helpers for testing DHCP
+#
+# Copyright Red Hat
+# Author: David Gibson <david@gibson.dropbear.id.au>
+
+import contextlib
+import ipaddress
+import os
+
+from tasst import Tasst, TasstSubData
+from tasst.address import IpiAllocator, TEST_NET_1
+from tasst.nstool import UnshareSite
+from tasst.site import Site
+from tasst.typing import typecheck
+
+
+class DhcpTasstInfo:
+    def __init__(self, site, ifname, addr, gw, mtu):
+        self.site = typecheck(site, Site)
+        self.ifname = typecheck(ifname, str)
+        self.addr = typecheck(addr, ipaddress.IPv4Address)
+        self.gw = typecheck(gw, ipaddress.IPv4Address)
+        self.mtu = typecheck(mtu, int)
+
+        site.require_cmds('ip')
+
+
+class BaseDhcpTasst(Tasst):
+    """
+    Test DHCP behaviour.
+
+    :avocado: disable
+    """
+
+    DHCLIENT = '/sbin/dhclient'
+
+    def setup_dhcp(self):
+        raise NotImplementedError("{} must implement setup_dhcp() method".format(type(self).__name__))
+
+    @contextlib.contextmanager
+    def dhclient(self):
+        with self.setup_dhcp() as dti:
+            dti = typecheck(dti, DhcpTasstInfo)
+
+            dti.site.require_cmds(self.DHCLIENT, 'cat', 'kill')
+
+            pidfile = os.path.join(self.workdir, 'dhclient.pid')
+            leasefile = os.path.join(self.workdir, '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
+            dti.site.fg('{} -4 -v -nc -pf {} -lf {} {}'
+                        .format(self.DHCLIENT, pidfile, leasefile, dti.ifname), sudo=True)
+
+            try:
+                yield dti
+            finally:
+                pid = int(dti.site.output('cat {}'.format(pidfile)))
+                dti.site.fg('kill {}'.format(pid))
+
+    def test_addr(self):
+        with self.dhclient() as dti:
+            (addr,) = dti.site.addrs(dti.ifname, family='inet', scope='global')
+            self.assertEquals(addr.ip, dti.addr)
+
+    def test_route(self):
+        with self.dhclient() as dti:
+            (defroute,) = dti.site.routes4(dst='default')
+            self.assertEquals(ipaddress.ip_address(defroute['gateway']), dti.gw)
+
+    def test_mtu(self):
+        with self.dhclient() as dti:
+            self.assertEquals(dti.site.mtu(dti.ifname), dti.mtu)
+
+
+class MetaDhcpTasst(BaseDhcpTasst):
+    """Ugly workaround for
+    https://github.com/avocado-framework/avocado/issues/5680.
+    Explicitly apply the "meta" tag to inherited tests
+
+    :avocado: disable
+    :avocado: tags=meta
+
+    """
+
+    def test_addr(self):
+        super().test_addr()
+
+    def test_route(self):
+        super().test_route()
+
+    def test_mtu(self):
+        super().test_mtu()
+
+
+class DhcpdTasst(MetaDhcpTasst):
+    DHCPD = 'dhcpd'
+    SUBNET = TEST_NET_1
+
+    @contextlib.contextmanager
+    def setup_dhcp(self):
+        ifname = 'clientif'
+        server_ifname = 'serverif'
+
+        with UnshareSite(type(self).__name__ + '.client', '-Un') as client, \
+             UnshareSite(type(self).__name__ + '.server', '-n',
+                         parent=client, sudo=True) as server:
+
+            server.require_cmds(self.DHCPD)
+
+            client.veth(ifname, server_ifname, server)
+
+            # Configure the DHCP server
+            ipa = IpiAllocator(self.SUBNET)
+            (server_ip4,) = ipa.next_ipis()
+            (client_ip4,) = ipa.next_ipis()
+
+            confpath = os.path.join(self.workdir, 'dhcpd.conf')
+            open(confpath, 'w').write('''
+            subnet {} netmask {} {{
+                option routers {};
+                range {} {};
+            }}
+            '''.format(self.SUBNET.network_address, self.SUBNET.netmask,
+                       server_ip4.ip, client_ip4.ip, client_ip4.ip))
+            pidpath = os.path.join(self.workdir, 'dhcpd.pid')
+            leasepath = os.path.join(self.workdir, 'dhcpd.leases')
+            open(leasepath, 'w').write('')
+
+            server.ifup('lo')
+            server.ifup(server_ifname, server_ip4)
+
+            opts = ('-f -d -4 -cf {} -lf {} -pf {}'.format(confpath, leasepath, pidpath))
+            server.fg('{} -t {}'.format(self.DHCPD, opts)) # test config
+            with server.bg('{} {}'.format(self.DHCPD, opts), sudo=True) as dhcpd:
+                # Configure the client
+                client.ifup('lo')
+
+                yield DhcpTasstInfo(client, ifname, client_ip4.ip, server_ip4.ip, 1500)
+
+                pid = int(open(pidpath).read())
+                server.fg('kill {}'.format(pid))
+                status = dhcpd.wait()
diff --git a/avocado/tasst/dhcpv6.py b/avocado/tasst/dhcpv6.py
new file mode 100644
index 0000000..5a0c166
--- /dev/null
+++ b/avocado/tasst/dhcpv6.py
@@ -0,0 +1,135 @@
+#! /usr/bin/python3
+
+# SPDX-License-Identifier: GPL-2.0-or-later
+#
+# tasst - Test A Simple Socket Transport
+#         library of test helpers for passt & pasta
+#
+# tasst/dhcpv6.py - Helpers for testing DHCPv6
+#
+# Copyright Red Hat
+# Author: David Gibson <david@gibson.dropbear.id.au>
+
+import contextlib
+import ipaddress
+import os
+
+from tasst import Tasst, TasstSubData
+from tasst.address import IpiAllocator, TEST_NET6_TASST_A
+from tasst.nstool import UnshareSite
+from tasst.site import Site
+from tasst.typing import typecheck
+
+
+class Dhcpv6TasstInfo:
+    def __init__(self, site, ifname, addr):
+        self.site = typecheck(site, Site)
+        self.ifname = typecheck(ifname, str)
+        self.addr = typecheck(addr, ipaddress.IPv6Address)
+
+        site.require_cmds('ip')
+
+
+class BaseDhcpv6Tasst(Tasst):
+    """
+    Test DHCPv6 behaviour.
+
+    :avocado: disable
+    """
+
+    DHCLIENT = '/sbin/dhclient'
+
+    def setup_dhcpv6(self):
+        raise NotImplementedError("{} must implement setup_dhcpv6() method".format(type(self).__name__))
+
+    @contextlib.contextmanager
+    def dhclientv6(self):
+        with self.setup_dhcpv6() as d6ti:
+            d6ti.site.require_cmds(self.DHCLIENT)
+
+            pidfile = os.path.join(self.workdir, 'dhclient.pid')
+            leasefile = os.path.join(self.workdir, '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
+            d6ti.site.fg('{} -6 -v -nc -pf {} -lf {} {}'
+                         .format(self.DHCLIENT, pidfile, leasefile, d6ti.ifname), sudo=True)
+
+            yield d6ti
+
+            pid = int(d6ti.site.output('cat {}'.format(pidfile)))
+            d6ti.site.fg('kill {}'.format(pid))
+
+    def test_addr(self):
+        with self.dhclientv6() as d6ti:
+            addrs = [a.ip for a in d6ti.site.addrs(d6ti.ifname, family='inet6', scope='global')]
+            self.assertIn(d6ti.addr, addrs) # Might also have a SLAAC address
+
+
+class MetaDhcpv6Tasst(BaseDhcpv6Tasst):
+    """Ugly workaround for
+    https://github.com/avocado-framework/avocado/issues/5680.
+    Explicitly apply the "meta" tag to inherited tests
+
+    :avocado: disable
+    :avocado: tags=meta
+
+    """
+
+    def test_addr(self):
+        super().test_addr()
+
+
+class Dhcpd6Tasst(MetaDhcpv6Tasst):
+    """
+    :avocado: tags=meta
+    """
+
+    DHCPD = 'dhcpd'
+    SUBNET = TEST_NET6_TASST_A
+
+    @contextlib.contextmanager
+    def setup_dhcpv6(self):
+        ifname = 'clientif'
+        server_ifname = 'serverif'
+
+        with UnshareSite(type(self).__name__ + '.client', '-Un') as client, \
+             UnshareSite(type(self).__name__ + '.server', '-n',
+                         parent=client, sudo=True) as server:
+            server.require_cmds(self.DHCPD)
+
+            client.veth(ifname, server_ifname, server)
+
+            # Allocate IPs, and sort out link local addressing
+            ipa = IpiAllocator(self.SUBNET)
+            (server_ip6,) = ipa.next_ipis()
+            (client_ip6,) = ipa.next_ipis()
+
+            server.ifup('lo')
+            server.ifup(server_ifname, server_ip6)
+            client.ifup('lo')
+            client.ifup(ifname)
+            (server_ip6_ll,) = server.addr_wait(server_ifname, family='inet6', scope='link')
+
+            # Configure the DHCP server
+            confpath = os.path.join(self.workdir, 'dhcpd.conf')
+            open(confpath, 'w').write('''
+            subnet6 {} {{
+                range6 {} {};
+            }}
+            '''.format(self.SUBNET, client_ip6.ip, client_ip6.ip))
+            pidpath = os.path.join(self.workdir, 'dhcpd.pid')
+            leasepath = os.path.join(self.workdir, 'dhcpd.leases')
+            open(leasepath, 'w').write('')
+
+            opts = ('-f -d -6 -cf {} -lf {} -pf {}'.format(confpath, leasepath, pidpath))
+            server.fg('{} -t {}'.format(self.DHCPD, opts)) # test config
+            with server.bg('{} {}'.format(self.DHCPD, opts), sudo=True) as dhcpd:
+                yield Dhcpv6TasstInfo(client, ifname, client_ip6.ip)
+
+                pid = int(open(pidpath).read())
+                server.fg('kill {}'.format(pid))
+                status = dhcpd.wait()
+
-- 
@@ -0,0 +1,135 @@
+#! /usr/bin/python3
+
+# SPDX-License-Identifier: GPL-2.0-or-later
+#
+# tasst - Test A Simple Socket Transport
+#         library of test helpers for passt & pasta
+#
+# tasst/dhcpv6.py - Helpers for testing DHCPv6
+#
+# Copyright Red Hat
+# Author: David Gibson <david@gibson.dropbear.id.au>
+
+import contextlib
+import ipaddress
+import os
+
+from tasst import Tasst, TasstSubData
+from tasst.address import IpiAllocator, TEST_NET6_TASST_A
+from tasst.nstool import UnshareSite
+from tasst.site import Site
+from tasst.typing import typecheck
+
+
+class Dhcpv6TasstInfo:
+    def __init__(self, site, ifname, addr):
+        self.site = typecheck(site, Site)
+        self.ifname = typecheck(ifname, str)
+        self.addr = typecheck(addr, ipaddress.IPv6Address)
+
+        site.require_cmds('ip')
+
+
+class BaseDhcpv6Tasst(Tasst):
+    """
+    Test DHCPv6 behaviour.
+
+    :avocado: disable
+    """
+
+    DHCLIENT = '/sbin/dhclient'
+
+    def setup_dhcpv6(self):
+        raise NotImplementedError("{} must implement setup_dhcpv6() method".format(type(self).__name__))
+
+    @contextlib.contextmanager
+    def dhclientv6(self):
+        with self.setup_dhcpv6() as d6ti:
+            d6ti.site.require_cmds(self.DHCLIENT)
+
+            pidfile = os.path.join(self.workdir, 'dhclient.pid')
+            leasefile = os.path.join(self.workdir, '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
+            d6ti.site.fg('{} -6 -v -nc -pf {} -lf {} {}'
+                         .format(self.DHCLIENT, pidfile, leasefile, d6ti.ifname), sudo=True)
+
+            yield d6ti
+
+            pid = int(d6ti.site.output('cat {}'.format(pidfile)))
+            d6ti.site.fg('kill {}'.format(pid))
+
+    def test_addr(self):
+        with self.dhclientv6() as d6ti:
+            addrs = [a.ip for a in d6ti.site.addrs(d6ti.ifname, family='inet6', scope='global')]
+            self.assertIn(d6ti.addr, addrs) # Might also have a SLAAC address
+
+
+class MetaDhcpv6Tasst(BaseDhcpv6Tasst):
+    """Ugly workaround for
+    https://github.com/avocado-framework/avocado/issues/5680.
+    Explicitly apply the "meta" tag to inherited tests
+
+    :avocado: disable
+    :avocado: tags=meta
+
+    """
+
+    def test_addr(self):
+        super().test_addr()
+
+
+class Dhcpd6Tasst(MetaDhcpv6Tasst):
+    """
+    :avocado: tags=meta
+    """
+
+    DHCPD = 'dhcpd'
+    SUBNET = TEST_NET6_TASST_A
+
+    @contextlib.contextmanager
+    def setup_dhcpv6(self):
+        ifname = 'clientif'
+        server_ifname = 'serverif'
+
+        with UnshareSite(type(self).__name__ + '.client', '-Un') as client, \
+             UnshareSite(type(self).__name__ + '.server', '-n',
+                         parent=client, sudo=True) as server:
+            server.require_cmds(self.DHCPD)
+
+            client.veth(ifname, server_ifname, server)
+
+            # Allocate IPs, and sort out link local addressing
+            ipa = IpiAllocator(self.SUBNET)
+            (server_ip6,) = ipa.next_ipis()
+            (client_ip6,) = ipa.next_ipis()
+
+            server.ifup('lo')
+            server.ifup(server_ifname, server_ip6)
+            client.ifup('lo')
+            client.ifup(ifname)
+            (server_ip6_ll,) = server.addr_wait(server_ifname, family='inet6', scope='link')
+
+            # Configure the DHCP server
+            confpath = os.path.join(self.workdir, 'dhcpd.conf')
+            open(confpath, 'w').write('''
+            subnet6 {} {{
+                range6 {} {};
+            }}
+            '''.format(self.SUBNET, client_ip6.ip, client_ip6.ip))
+            pidpath = os.path.join(self.workdir, 'dhcpd.pid')
+            leasepath = os.path.join(self.workdir, 'dhcpd.leases')
+            open(leasepath, 'w').write('')
+
+            opts = ('-f -d -6 -cf {} -lf {} -pf {}'.format(confpath, leasepath, pidpath))
+            server.fg('{} -t {}'.format(self.DHCPD, opts)) # test config
+            with server.bg('{} {}'.format(self.DHCPD, opts), sudo=True) as dhcpd:
+                yield Dhcpv6TasstInfo(client, ifname, client_ip6.ip)
+
+                pid = int(open(pidpath).read())
+                server.fg('kill {}'.format(pid))
+                status = dhcpd.wait()
+
-- 
2.40.1


  parent reply	other threads:[~2023-05-31  1:59 UTC|newest]

Thread overview: 21+ messages / expand[flat|nested]  mbox.gz  Atom feed  top
2023-05-31  1:58 [PATCH v3 00/20] RFCv3: Proof-of-concept conversion of some tests to Avocado framework David Gibson
2023-05-30 23:59 ` [PATCH v3 01/20] avocado: Make a duplicate copy of testsuite for comparison purposes David Gibson
2023-05-31  1:58 ` [PATCH v3 02/20] avocado: Don't double download assets for test/ and oldtest/ David Gibson
2023-05-31  1:58 ` [PATCH v3 03/20] avocado: Move static checkers to avocado David Gibson
2023-05-31  1:58 ` [PATCH v3 04/20] avocado/tasst: Helper functions for executing commands in different places David Gibson
2023-05-31  1:58 ` [PATCH v3 05/20] avocado/tasst: Type checking helpers David Gibson
2023-05-31  1:58 ` [PATCH v3 06/20] avocado: Convert build tests to avocado David Gibson
2023-05-31  1:58 ` [PATCH v3 07/20] avocado/tasst: Add helpers for running background commands on sites David Gibson
2023-05-31  1:58 ` [PATCH v3 08/20] avocado/tasst: Add helper to get network interface names for a site David Gibson
2023-05-31  1:58 ` [PATCH v3 09/20] avocado/tasst: Add helpers to run commands with nstool David Gibson
2023-05-31  1:58 ` [PATCH v3 10/20] avocado/tasst: Add ifup and network address helpers to Site David Gibson
2023-05-31  1:58 ` [PATCH v3 11/20] avocado/tasst: Helper for creating veth devices between namespaces David Gibson
2023-05-31  1:58 ` [PATCH v3 12/20] avocado/tasst: Add helper for getting MTU of a network interface David Gibson
2023-05-31  1:58 ` [PATCH v3 13/20] avocado/tasst: Add helper to wait for IP address to appear David Gibson
2023-05-31  1:58 ` [PATCH v3 14/20] avocado/tasst: Add helpers for getting a site's routes David Gibson
2023-05-31  1:58 ` [PATCH v3 15/20] avocado/tasst: Helpers for test transferring data between sites David Gibson
2023-05-31  1:58 ` [PATCH v3 16/20] avocado/tasst: IP address allocation helpers David Gibson
2023-05-31  1:58 ` [PATCH v3 17/20] avocado/tasst: Helpers for testing NDP behaviour David Gibson
2023-05-31  1:58 ` David Gibson [this message]
2023-05-31  1:58 ` [PATCH v3 19/20] avocado/tasst: Helpers to construct a simple network environment for tests David Gibson
2023-05-31  1:58 ` [PATCH v3 20/20] avocado: Convert basic pasta tests David Gibson

Reply instructions:

You may reply publicly to this message via plain-text email
using any one of the following methods:

* Save the following mbox file, import it into your mail client,
  and reply-to-all from there: mbox

  Avoid top-posting and favor interleaved quoting:
  https://en.wikipedia.org/wiki/Posting_style#Interleaved_style

* Reply using the --to, --cc, and --in-reply-to
  switches of git-send-email(1):

  git send-email \
    --in-reply-to=20230531015849.3229596-19-david@gibson.dropbear.id.au \
    --to=david@gibson.dropbear.id.au \
    --cc=crosa@redhat.com \
    --cc=jarichte@redhat.com \
    --cc=passt-dev@passt.top \
    --cc=sbrivio@redhat.com \
    /path/to/YOUR_REPLY

  https://kernel.org/pub/software/scm/git/docs/git-send-email.html

* If your mail client supports setting the In-Reply-To header
  via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line before the message body.
Code repositories for project(s) associated with this public inbox

	https://passt.top/passt

This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox;
as well as URLs for IMAP folder(s).