morethantext/release_tests/support/cluster.py

"""Create a cluster."""

from release_tests.support.mttserver import MTTServer
from release_tests.support.translate import Translate


class Cluster:
    """cluster of MoreThanText."""

    def __init__(self, num=2):
        """initialization"""
        self.translate = Translate()
        self.servers = []
        for _ in range(num):
            self.servers.append(MTTServer())

    async def start(self):
        """Start the cluster."""
        await self.translate.start()
        for server in self.servers:
            await server.start()

    async def stop(self):
        """Stops cluster."""
        await self.translate.stop()
        for server in self.servers:
            await server.stop()