morethantext/test/test_single_boot.py

58 lines
1.7 KiB
Python

"""Tests for single server boot ups."""
import aiohttp
from asyncio import create_subprocess_exec
from pathlib import Path
from unittest import IsolatedAsyncioTestCase
class Servers:
"""Setup and run servers."""
def __init__(self):
"""Initialize class"""
self.servers = []
async def create_server(self, *args):
"""Create servers"""
app = Path.cwd().joinpath("target", "release", "morethantext")
if args:
cmd = list(args)
cmd.insert(0, app)
else:
cmd = [app]
self.servers.append(await create_subprocess_exec(*cmd))
async def destroy_servers(self):
"""destroy servers"""
for server in self.servers:
server.terminate()
await server.wait()
class SingleBootTC(IsolatedAsyncioTestCase):
"""Test single server boot up."""
async def asyncSetUp(self):
"""Test Case setup"""
self.servers = Servers()
async def asyncTearDown(self):
await self.servers.destroy_servers()
async def test_default_boot(self):
"Are the default settings available?"""
await self.servers.create_server()
async with aiohttp.ClientSession() as session:
async with session.get("http://localhost:3000") as response:
self.assertEqual(response.status, 200)
async def test_alt_port_boot(self):
"""Can the server boot off on alternate port?"""
port = 9025
await self.servers.create_server("-p", str(port))
async with aiohttp.ClientSession() as session:
async with session.get(f"http://localhost:{port}") as response:
self.assertEqual(response.status, 200)