CSIT-158: Tap interface tests

[csit.git] / resources / libraries / python / Namespaces.py

diff --git a/resources/libraries/python/Namespaces.py b/resources/libraries/python/Namespaces.py
new file mode 100644 (file)
index 0000000..d92dfd7
--- /dev/null
+++ b/resources/libraries/python/Namespaces.py
@@ -0,0 +1,96 @@
+# Copyright (c) 2016 Cisco and/or its affiliates.
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at:
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Linux namespace utilities library."""
+
+from resources.libraries.python.ssh import exec_cmd_no_error, exec_cmd, SSH
+
+
+class Namespaces(object):
+    """Linux namespace utilities."""
+    def __init__(self):
+        self._namespaces = []
+
+    def create_namespace(self, node, namespace_name):
+        """Create namespace and add the name to the list for later clean-up.
+
+        :param node: Where to create namespace.
+        :param namespace_name: Name for namespace.
+        :type node: dict
+        :type namespace_name: str
+        """
+        cmd = ('ip netns add {0}'.format(namespace_name))
+        exec_cmd_no_error(node, cmd, sudo=True)
+        self._namespaces.append(namespace_name)
+
+    @staticmethod
+    def attach_interface_to_namespace(node, namespace, interface):
+        """Attach specific interface to namespace.
+
+        :param node: Node where to execute command.
+        :param namespace: Namespace to execute command on.
+        :param interface: Interface in namespace.
+        :type node: dict
+        :type namespace: str
+        :type interface: str
+        :raises RuntimeError: Interface could not be attached.
+        """
+        cmd = 'ip link set {0} netns {1}'.format(interface, namespace)
+        (rc, _, stderr) = exec_cmd(node, cmd, timeout=5, sudo=True)
+        if rc != 0:
+            raise RuntimeError(
+                'Could not attach interface, reason:{}'.format(stderr))
+        cmd = 'ip netns exec {} ip link set {} up'.format(
+            namespace, interface)
+        (rc, _, stderr) = exec_cmd(node, cmd, timeout=5, sudo=True)
+        if rc != 0:
+            raise RuntimeError(
+                'Could not set interface state, reason:{}'.format(stderr))
+
+    @staticmethod
+    def create_bridge_for_int_in_namespace(
+            node, namespace, bridge_name, *interfaces):
+        """Setup bridge domain and add interfaces to it.
+
+        :param node: Node where to execute command.
+        :param namespace: Namespace to execute command on.
+        :param bridge_name: Name of the bridge to be created.
+        :param interfaces: List of interfaces to add to the namespace.
+        :type node: dict
+        :type namespace: str
+        :type bridge_name: str
+        :type interfaces: list
+        """
+        cmd = 'ip netns exec {} brctl addbr {}'.format(namespace, bridge_name)
+        exec_cmd_no_error(node, cmd, sudo=True)
+        for interface in interfaces:
+            cmd = 'ip netns exec {} brctl addif {} {}'.format(
+                namespace, bridge_name, interface)
+            exec_cmd_no_error(node, cmd, sudo=True)
+        cmd = 'ip netns exec {} ip link set dev {} up'.format(
+            namespace, bridge_name)
+        exec_cmd_no_error(node, cmd, sudo=True)
+
+    def clean_up_namespaces(self, node):
+        """Remove all old namespaces.
+
+        :param node: Node where to execute command.
+        :type node: dict
+        :raises RuntimeError: Namespaces could not be cleaned properly.
+        """
+        for namespace in self._namespaces:
+            print "Cleaning namespace {}".format(namespace)
+            cmd = 'ip netns delete {}'.format(namespace)
+            (rc, stdout, stderr) = exec_cmd(node, cmd, timeout=5, sudo=True)
+            if rc != 0:
+                raise RuntimeError('Could not delete namespace')