import asyncio
from dataclasses import dataclass
from pathlib import Path

import aiohttp


@dataclass
class RunHttpServerChildProcess:
    server_path: Path

    async def __aenter__(self):
        self.subprocess = await asyncio.create_subprocess_exec(
            'pdm', 'run', 'python', self.server_path)
        self._session = await aiohttp.ClientSession().__aenter__()
        return self

    async def __aexit__(self, exc_type, exc, tb):
        await self._session.close()
        self.subprocess.terminate()
        await self.subprocess.wait()

    async def get(self, url: str, headers=None) -> aiohttp.ClientResponse:
        while True:
            try:
                return await self._session.get(url, headers=headers)
            except aiohttp.ClientConnectorError:
                await asyncio.sleep(0.05)