# 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']))