1 files changed, 113 insertions, 0 deletions
diff --git a/resources/libraries/python/DMM/SingleCliSer.py b/resources/libraries/python/DMM/SingleCliSer.py
new file mode 100644
index 0000000000..504f59c415
--- /dev/null
+++ b/resources/libraries/python/DMM/SingleCliSer.py
@@ -0,0 +1,113 @@
+# Copyright (c) 2018 Huawei Technologies Co.,Ltd.
+# 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.
+
+
+"""
+This module exists to provide the vs_epoll ping test for DMM on topology nodes.
+"""
+import time
+
+from resources.libraries.python.ssh import SSH
+from resources.libraries.python.DMM.DMMConstants import DMMConstants as con
+from resources.libraries.python.topology import Topology
+from robot.api import logger
+
+class SingleCliSer(object):
+    """Test the DMM vs_epoll ping function."""
+
+    @staticmethod
+    def exec_the_base_vs_epoll_test(dut1_node, dut2_node):
+        """Execute the vs_epoll on the dut1_node.
+
+        :param dut1_node: Will execute the vs_epoll on this node.
+        :param dut2_node: Will execute the vc_epoll on this node.
+        :type dut1_node: dict
+        :type dut2_node: dict
+        :returns: positive value if packets are sent and received
+        :raises RuntimeError:If failed to execute vs_epoll test on  dut1_node.
+        """
+        dut1_ip = Topology.get_node_hostname(dut1_node)
+        dut2_ip = Topology.get_node_hostname(dut2_node)
+
+        ssh = SSH()
+        ssh.connect(dut1_node)
+
+        cmd = 'cd {0}/{1} && ./run_dmm.sh {2} {3} {4} ' \
+        .format(con.REMOTE_FW_DIR, con.DMM_SCRIPTS, dut1_ip, dut2_ip, 0)
+
+        cmd += '2>&1 | tee log_run_dmm.txt &'
+
+        (ret_code, _, _) = ssh.exec_command(cmd, timeout=6000)
+        if ret_code != 0:
+            raise RuntimeError('Failed to execute vs_epoll test at node {0}'
+                               .format(dut1_node['host']))
+
+        time.sleep(10)
+
+        ssh = SSH()
+        ssh.connect(dut2_node)
+
+        cmd = 'cd {0}/{1} && ./run_dmm.sh {2} {3} {4} ' \
+        .format(con.REMOTE_FW_DIR, con.DMM_SCRIPTS, dut1_ip, dut2_ip, 1)
+
+        cmd += '2>&1 | tee log_run_dmm.txt'
+
+        (ret_code, stdout_cli, _) = ssh.exec_command(cmd, timeout=6000)
+        if ret_code != 0:
+            raise RuntimeError('Failed to execute vs_epoll test at node {0}'
+                               .format(dut1_node['host']))
+
+        return stdout_cli.find("send 50000")
+
+    @staticmethod
+    def get_the_test_result(dut_node):
+        """
+        After executing exec_the_base_vs_epoll_test, use this
+        to get the test result
+
+        :param dut_node: will get the test result in this dut node
+        :type dut_node: dict
+        :returns: str.
+        :rtype: str.
+        :raises RuntimeError: If failed to get the test result.
+        """
+        ssh = SSH()
+        ssh.connect(dut_node)
+        cmd = 'cat {0}/{1}/log_run_dmm.txt | grep "send 50000" | wc -l' \
+        .format(con.REMOTE_FW_DIR, con.DMM_SCRIPTS)
+
+        (ret_code, stdout, _) = ssh.exec_command(cmd, timeout=100)
+        if ret_code != 0:
+            raise RuntimeError('Failed to get test result at node {0}'
+                               .format(dut_node['host']))
+
+        return stdout
+
+    @staticmethod
+    def echo_dmm_logs(dut_node):
+        """
+        :param dut_node:
+        :return:
+        """
+        ssh = SSH()
+        ssh.connect(dut_node)
+        cmd = 'cat {0}/{1}/log_install_dmm.txt' \
+        .format(con.REMOTE_FW_DIR, con.DMM_SCRIPTS)
+
+        (ret_code, stdout, _) = ssh.exec_command(cmd, timeout=100)
+        if ret_code != 0:
+            raise RuntimeError('Failed to get log_install_dmm at node {0}'
+                               .format(dut_node['host']))
+        else:
+            logger.console('....log_install_dmm on node {1}.... {0}'
+                           .format(stdout, dut_node['host']))