Signed-iff-by: David Gibson <david(a)gibson.dropbear.id.au> --- avocado/tasst/ndp.py | 137 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 137 insertions(+) create mode 100644 avocado/tasst/ndp.py diff --git a/avocado/tasst/ndp.py b/avocado/tasst/ndp.py new file mode 100644 index 0000000..771cba8 --- /dev/null +++ b/avocado/tasst/ndp.py @@ -0,0 +1,137 @@ +#! /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/ndp.py - Helpers for testing NDP +# +# Copyright Red Hat +# Author: David Gibson <david(a)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 NdpTasstInfo: + def __init__(self, site, ifname, net, gw): + self.site = typecheck(site, Site) + self.ifname = typecheck(ifname, str) + self.net = typecheck(net, ipaddress.IPv6Network) + self.gw = typecheck(gw, ipaddress.IPv6Address) + + site.require_cmds('ip') + + +class BaseNdpTasst(Tasst): + """ + Test NDP behaviour. + + :avocado: disable + """ + + def setup_ndp(self): + raise NotImplementedError("{} must implement setup_ndp() method".format(type(self).__name__)) + + @contextlib.contextmanager + def check_setup_ndp(self): + with self.setup_ndp() as ndp: + if not isinstance(ndp, NdpTasstInfo): + raise TypeError("{}.setup_ndp() must yield a NdpTasstInfo instance".format(type(self).__name__)) + yield ndp + + def test_addr(self): + with self.check_setup_ndp() as ndp: + # Wait for NDP to do its thing + (addr,) = ndp.site.addr_wait(ndp.ifname, family='inet6', scope='global') + + # The SLAAC address is derived from the guest ns MAC, so + # probably won't exactly match the host address (we need + # DHCPv6 for that). It should be in the right network though. + self.assertEquals(addr.network, ndp.net) + + def test_route(self): + with self.check_setup_ndp() as ndp: + defroutes = ndp.site.routes6(dst='default') + while not defroutes: + defroutes = ndp.site.routes6(dst='default') + + self.assertEquals(len(defroutes), 1) + gateway = ipaddress.ip_address(defroutes[0]['gateway']) + self.assertEquals(gateway, ndp.gw) + + +class MetaNdpTasst(BaseNdpTasst): + """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() + + +class RadvdNdpTasst(MetaNdpTasst): + timeout = 15.0 + + @contextlib.contextmanager + def setup_ndp(self): + ifname = 'clientif' + router_ifname = 'routerif' + prefix = TEST_NET6_TASST_A + + with UnshareSite(type(self).__name__ + '.client', '-Un') as client, \ + UnshareSite(type(self).__name__ + '.router', '-n', parent=client, sudo=True) as router: + router.require_cmds('radvd') + + client.veth(ifname, router_ifname, router) + + # Configure the simulated router + ipa = IpiAllocator(prefix) + (router_ip6,) = ipa.next_ipis() + + confpath = os.path.join(self.workdir, 'radvd.conf') + pidpath = os.path.join(self.workdir, 'radvd.pid') + open(confpath, 'w').write( + ''' + interface {} {{ + AdvSendAdvert on; + prefix {} {{ + }}; + }}; + '''.format(router_ifname, prefix)) + + router.ifup('lo') + router.ifup('routerif', router_ip6) + + # Configure the client + client.ifup('lo') + client.ifup(ifname) + + # Get the router's link-local-address + (router_ll,) = router.addr_wait(router_ifname, family='inet6', scope='link') + + # Run radvd + router.fg('radvd -c -C {}'.format(confpath)) + with router.bg('radvd -C {} -p {} -n -d 5'.format(confpath, pidpath), sudo=True) as radvd: + yield NdpTasstInfo(client, ifname, prefix, router_ll.ip) + + pid = int(open(pidpath).read()) + router.fg('kill {}'.format(pid)) + status = radvd.wait() + self.assertEquals(status, 0) -- 2.40.1