[PATCH v2 20/21] avocado/tasst: Helpers to construct a simple network environment for tests

[David Gibson <david@gibson.dropbear.id.au>]

This constructs essentially the simplest sensible network for passt/pasta
to operate in.  We have one netns "simhost" to represent the host where we
will run passt or pasta, and a second "gw" to represent its default
gateway.

Signed-off-by: David Gibson <david@gibson.dropbear.id.au>
---
 avocado/tasst/scenario/__init__.py | 11 ++++
 avocado/tasst/scenario/simple.py   | 96 ++++++++++++++++++++++++++++++
 2 files changed, 107 insertions(+)
 create mode 100644 avocado/tasst/scenario/__init__.py
 create mode 100644 avocado/tasst/scenario/simple.py

diff --git a/avocado/tasst/scenario/__init__.py b/avocado/tasst/scenario/__init__.py
new file mode 100644
index 0000000..8ca64a6
--- /dev/null
+++ b/avocado/tasst/scenario/__init__.py
@@ -0,0 +1,11 @@
+#! /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/scenario - Helpers to set up various sample network topologies
+#
+# Copyright Red Hat
+# Author: David Gibson <david@gibson.dropbear.id.au>
diff --git a/avocado/tasst/scenario/simple.py b/avocado/tasst/scenario/simple.py
new file mode 100644
index 0000000..8f6b41e
--- /dev/null
+++ b/avocado/tasst/scenario/simple.py
@@ -0,0 +1,96 @@
+#! /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/scenario/simple - Smallest sensible network to use passt/pasta
+#
+# Copyright Red Hat
+# Author: David Gibson <david@gibson.dropbear.id.au>
+
+from tasst import Tasst
+from tasst.nstool import UnshareSite
+from tasst.address import IpiAllocator, TEST_NET_2, TEST_NET6_TASST_B
+from tasst.transfer import BaseTransferTasst, MetaTcpUploadTasst, MetaUdpTransferTasst
+
+
+class SimpleNetTasst(Tasst):
+    """Base class for tests in a simple network scenario
+
+    The sample network has 2 sites (network namespaces) connected with
+    a veth link:
+            [simhost]   <-veth->    [gw]
+
+    gw is set up as the default router for simhost.
+
+    simhost has addresses:
+        self.ip4 (IPv4), self.ip6 (IPv6), self.ip6_ll (IPv6 link local)
+
+    gw has addresses:
+        self.gw_ip4 (IPv4), self.gw_ip6 (IPv6), self.gw_ip6_ll (IPv6 link local)
+        self.remote_ip4 (IPv4), self.remote_ip6 (IPv6)
+
+    The "remote" addresses are on a different subnet from the others,
+    so the only way for simhost to reach them is via its default
+    route.  This helps to exercise that we're actually using that,
+    rather than just local net routes.
+
+    """
+    def setUp(self):
+        super().setUp()
+
+        self.simhost = UnshareSite(type(self).__name__ + '.simhost',
+                                   '-Ucnpf --mount-proc')
+        self.gw = UnshareSite(type(self).__name__ + '.gw', '-n',
+                              parent=self.simhost, sudo=True)
+
+        self.ifname = 'veth'
+        self.gw_ifname = 'gw' + self.ifname
+        self.simhost.veth(self.ifname, self.gw_ifname, self.gw)
+
+        ipa_local = IpiAllocator()
+        self.ip4, self.ip6 = ipa_local.next_ipis()
+        self.gw_ip4, self.gw_ip6 = ipa_local.next_ipis()
+
+        ipa_remote = IpiAllocator(TEST_NET_2, TEST_NET6_TASST_B)
+        self.remote_ip4, self.remote_ip6 = ipa_remote.next_ipis()
+
+        self.gw.ifup('lo')
+        self.gw.ifup(self.gw_ifname, self.gw_ip4, self.gw_ip6,
+                     self.remote_ip4, self.remote_ip6)
+
+        self.simhost.ifup('lo')
+        self.simhost.ifup(self.ifname, self.ip4, self.ip6)
+
+        # Once link is up on both sides, SLAAC will run
+        self.gw_ip6_ll = self.gw.addr_wait(self.gw_ifname, family='inet6', scope='link')[0]
+        self.ip6_ll = self.simhost.addr_wait(self.ifname, family='inet6', scope='link')[0]
+
+        # Set up the default route
+        self.simhost.fg('ip -4 route add default via {}'
+                        .format(self.gw_ip4.ip), sudo=True)
+        self.simhost.fg('ip -6 route add default via {} dev {}'
+                        .format(self.gw_ip6_ll.ip, self.ifname), sudo=True)
+
+    def tearDown(self):
+        self.gw.close()
+        self.simhost.close()
+        super().tearDown()
+
+
+class SimpleNetConnectivityTasst(SimpleNetTasst, MetaTcpUploadTasst, MetaUdpTransferTasst):
+    """
+    Check connectivity between the simulated host and the gateway,
+    before involving passt/pasta.
+
+    :avocado: tags=meta
+
+    """
+
+    def setUp(self):
+        super().setUp()
+
+        BaseTransferTasst.subsetup(self, 'test/small.bin', self.simhost, self.gw,
+                                   self.remote_ip4.ip, self.remote_ip6.ip, 10000)
-- 
@@ -0,0 +1,96 @@
+#! /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/scenario/simple - Smallest sensible network to use passt/pasta
+#
+# Copyright Red Hat
+# Author: David Gibson <david@gibson.dropbear.id.au>
+
+from tasst import Tasst
+from tasst.nstool import UnshareSite
+from tasst.address import IpiAllocator, TEST_NET_2, TEST_NET6_TASST_B
+from tasst.transfer import BaseTransferTasst, MetaTcpUploadTasst, MetaUdpTransferTasst
+
+
+class SimpleNetTasst(Tasst):
+    """Base class for tests in a simple network scenario
+
+    The sample network has 2 sites (network namespaces) connected with
+    a veth link:
+            [simhost]   <-veth->    [gw]
+
+    gw is set up as the default router for simhost.
+
+    simhost has addresses:
+        self.ip4 (IPv4), self.ip6 (IPv6), self.ip6_ll (IPv6 link local)
+
+    gw has addresses:
+        self.gw_ip4 (IPv4), self.gw_ip6 (IPv6), self.gw_ip6_ll (IPv6 link local)
+        self.remote_ip4 (IPv4), self.remote_ip6 (IPv6)
+
+    The "remote" addresses are on a different subnet from the others,
+    so the only way for simhost to reach them is via its default
+    route.  This helps to exercise that we're actually using that,
+    rather than just local net routes.
+
+    """
+    def setUp(self):
+        super().setUp()
+
+        self.simhost = UnshareSite(type(self).__name__ + '.simhost',
+                                   '-Ucnpf --mount-proc')
+        self.gw = UnshareSite(type(self).__name__ + '.gw', '-n',
+                              parent=self.simhost, sudo=True)
+
+        self.ifname = 'veth'
+        self.gw_ifname = 'gw' + self.ifname
+        self.simhost.veth(self.ifname, self.gw_ifname, self.gw)
+
+        ipa_local = IpiAllocator()
+        self.ip4, self.ip6 = ipa_local.next_ipis()
+        self.gw_ip4, self.gw_ip6 = ipa_local.next_ipis()
+
+        ipa_remote = IpiAllocator(TEST_NET_2, TEST_NET6_TASST_B)
+        self.remote_ip4, self.remote_ip6 = ipa_remote.next_ipis()
+
+        self.gw.ifup('lo')
+        self.gw.ifup(self.gw_ifname, self.gw_ip4, self.gw_ip6,
+                     self.remote_ip4, self.remote_ip6)
+
+        self.simhost.ifup('lo')
+        self.simhost.ifup(self.ifname, self.ip4, self.ip6)
+
+        # Once link is up on both sides, SLAAC will run
+        self.gw_ip6_ll = self.gw.addr_wait(self.gw_ifname, family='inet6', scope='link')[0]
+        self.ip6_ll = self.simhost.addr_wait(self.ifname, family='inet6', scope='link')[0]
+
+        # Set up the default route
+        self.simhost.fg('ip -4 route add default via {}'
+                        .format(self.gw_ip4.ip), sudo=True)
+        self.simhost.fg('ip -6 route add default via {} dev {}'
+                        .format(self.gw_ip6_ll.ip, self.ifname), sudo=True)
+
+    def tearDown(self):
+        self.gw.close()
+        self.simhost.close()
+        super().tearDown()
+
+
+class SimpleNetConnectivityTasst(SimpleNetTasst, MetaTcpUploadTasst, MetaUdpTransferTasst):
+    """
+    Check connectivity between the simulated host and the gateway,
+    before involving passt/pasta.
+
+    :avocado: tags=meta
+
+    """
+
+    def setUp(self):
+        super().setUp()
+
+        BaseTransferTasst.subsetup(self, 'test/small.bin', self.simhost, self.gw,
+                                   self.remote_ip4.ip, self.remote_ip6.ip, 10000)
-- 
2.40.1