"""Wakaama integration tests (pytest) fixtures"""
from pathlib import Path
from time import sleep

import pexpect
import pytest


BASE_PATH = str(Path(__file__).parent.absolute())
REPO_BASE_PATH = BASE_PATH + "/../../"


# pylint: disable=no-member
class HelperBase:
    """Provides helper methods for integration tests. Wraps "pexpect" API"""

    def __dump_text(self):
        """Internal method that prints debug help"""
        print("Debug help: actual output---------------------")
        print(self.pexpectobj.before)
        print("----------------------------------------------")

    def command_response(self, cmd, resp):
        """Issue CLI command and check response"""
        self.pexpectobj.sendline(cmd)
        try:
            self.pexpectobj.expect_exact(resp)
        except pexpect.exceptions.TIMEOUT:
            self.__dump_text()
            return False
        return True

    def wait_for_packet(self):
        """Wait for "packet printout" and return text output"""
        try:
            self.pexpectobj.expect_exact("bytes received from")
        except pexpect.exceptions.TIMEOUT:
            self.__dump_text()
            assert False
        try:
            self.pexpectobj.expect_exact("\r\r\n>")
        except pexpect.exceptions.TIMEOUT:
            self.__dump_text()
            assert False
        return self.pexpectobj.before

    def wait_for_text(self, txt):
        """Wait for text printout and return True if found"""
        try:
            self.pexpectobj.expect_exact(txt)
        except pexpect.exceptions.TIMEOUT:
            self.__dump_text()
            return False
        return True

    def wait_for_time(self, timedelay):
        """Wait for time and return output since last command"""
        sleep(timedelay)
        # this is a "hack" to be able to return output between commands
        self.command_response("help", "help")
        return self.pexpectobj.before

    def quit(self):
        """Quit client or server"""
        self.pexpectobj.sendline("q")  # exit, if client
        self.pexpectobj.sendline("quit")  # exit, if server
        self.pexpectobj.expect(pexpect.EOF)  # make sure exited


class Lwm2mServer(HelperBase):
    """Server subclass of HelperBase"""

    def __init__(self, arguments="", timeout=3, encoding="utf8"):
        self.pexpectobj = pexpect.spawn(REPO_BASE_PATH +
                                        "/build-wakaama-server/lwm2mserver " +
                                        arguments,
                                        encoding=encoding,
                                        timeout=timeout)
        # pylint: disable=consider-using-with
        self.pexpectobj.logfile = open(BASE_PATH +
                                       "/lwm2mserver_log.txt",
                                       "w",
                                       encoding="utf-8")


class Lwm2mClient(HelperBase):
    """Client subclass of HelperBase"""

    def __init__(self, arguments="", timeout=3, encoding="utf8"):
        self.pexpectobj = pexpect.spawn(REPO_BASE_PATH +
                                        "/build-wakaama-client/udp/lwm2mclient "
                                        + arguments,
                                        encoding=encoding,
                                        timeout=timeout)
        # pylint: disable=consider-using-with
        self.pexpectobj.logfile = open(BASE_PATH +
                                       "/lwm2mclient_log.txt",
                                       "w",
                                       encoding="utf-8")


class Lwm2mBootstrapServer(HelperBase):
    """Bootstrap-server subclass of HelperBase"""

    def __init__(self, arguments="", timeout=3, encoding="utf8"):
        self.pexpectobj = pexpect.spawn(REPO_BASE_PATH +
                                        "/build-wakaama-bootstrap_server/bootstrap_server "
                                        + arguments,
                                        encoding=encoding,
                                        timeout=timeout)
        # pylint: disable=consider-using-with
        self.pexpectobj.logfile = open(BASE_PATH +
                                       "/lwm2mbootstrapserver_log.txt",
                                       "w",
                                       encoding="utf-8")


@pytest.fixture
def lwm2mserver(request):
    """Provide lwm2mserver instance."""
    marker = request.node.get_closest_marker("server_args")
    if marker is None:
        args = ""
    else:
        args = marker.args[0]
    server = Lwm2mServer(args)
    yield server
    server.quit()


@pytest.fixture
def lwm2mclient(request):
    """Provide lwm2mclient instance."""
    marker = request.node.get_closest_marker("client_args")
    if marker is None:
        args = ""
    else:
        args = marker.args[0]
    client = Lwm2mClient(args)
    yield client
    client.quit()


@pytest.fixture
def lwm2mbootstrapserver():
    """Provide bootstrap_server instance."""
    ini_file = REPO_BASE_PATH + "/examples/bootstrap_server/bootstrap_server.ini"
    bootstrapserver = Lwm2mBootstrapServer(f"-f {ini_file}", timeout=13)
    bootstrapserver.wait_for_text(">")
    yield bootstrapserver
    bootstrapserver.quit()