morethantext/release_tests/support/test_mttserver.py

"""tests for MoreThanText servers."""

from unittest import IsolatedAsyncioTestCase
from aiohttp import ClientSession
from release_tests.support import get_port
from release_tests.support.mttserver import MTTServer


class MTTServerTC(IsolatedAsyncioTestCase):
    """Testcases for controling MoreThanText release servers."""

    async def test_defaukt_start_up(self):
        """Test default start up."""
        mtt = MTTServer()
        await mtt.start()
        url = mtt.baseurl
        self.assertEqual(url, "http://127.0.0.1:3000")
        async with ClientSession() as session:
            async with session.get(url) as resp:
                text = await resp.text()
                self.assertEqual(resp.status, 200, text)
        await mtt.cleanup()

    async def test_flags(self):
        """Make sure flags are working."""
        host = "127.28.56.13"
        port = get_port()
        mtt = MTTServer("-a", host, "-p", port)
        self.assertEqual(mtt.baseurl, f"http://{host}:{port}")
        await mtt.start()
        async with ClientSession() as session:
            async with session.get(mtt.baseurl) as resp:
                text = await resp.text()
                self.assertEqual(resp.status, 200, text)
        await mtt.cleanup()