morethantext-web/tests/step_defs/server.py

"""Fixture for setting up a single server."""

from asyncio import create_subprocess_exec, get_event_loop, sleep
from pathlib import Path
from random import choices
from shutil import rmtree
from socket import gethostname, gethostbyname, socket
from string import ascii_lowercase


class Server:
    """Setup a single server."""

    def __init__(self):
        """Initialization of a server."""
        self.datadir = Path.cwd().joinpath("tests", "data", "".join(choices(ascii_lowercase, k=5)))
        self.datadir.mkdir(parents=True)
        self.env = {}
        self.process = None
        self.loop = get_event_loop()

    @property
    def address(self):
        """Get the server address."""
        address = "127.0.0.1"
        if "MMT_ADDRESS" in self.env:
            address = self.env["MMT_ADDRESS"]
        return address

    @address.setter
    def address(self, addr):
        """Sets the server address."""
        self.env["MTT_ADDRESS"] = addr

    @property
    def port(self):
        """Get the port of the server."""
        port = "9090"
        if "MTT_PORT" in self.env:
            port = self.env["MTT_PORT"]
        return port

    @port.setter
    def port(self, num):
        """Set the port for the server."""
        self.env["MTT_PORT"] = str(num)

    async def __start(self):
        """async start of the server."""
        self.process = await create_subprocess_exec(
            Path.cwd().joinpath("target", "release", "morethantext_web"), 
            env=self.env,
            cwd=self.datadir
        )
        await sleep(1)

    async def __stop(self):
        """async stop of the server."""
        if self.process is not None and self.process.returncode is None:
            self.process.terminate()
            await self.process.wait()

    def start(self):
        """Start the server."""
        self.loop.run_until_complete(self.__start())

    def stop(self):
        """Stop the server."""
        self.loop.run_until_complete(self.__stop())

    def destroy(self):
        """Removes the server instance."""
        self.stop()
        rmtree(self.datadir, ignore_errors=True)

    def set_safe_port(self):
        """Set the server port to something not being used."""
        while self.port == "9090":
            sock = socket()
            sock.bind((self.address, 0))
            self.port = sock.getsockname()[1]
            sock.close()

    def set_to_host_ip(self):
        """Set the server to use something other than localhost."""
        hostname = gethostname()
        self.address = gethostbyname(hostname)
        print(self.address)
        self.set_safe_port()