import asyncio

import pytest
from amqtt.client import MQTTClient

import bumper
from tests import HOST, MQTT_PORT


@pytest.fixture
async def mqtt_server():
    mqtt_server = bumper.MQTTServer(HOST, MQTT_PORT, password_file="tests/passwd")
    await mqtt_server.start()
    bumper.mqtt_server = mqtt_server
    while not mqtt_server.state == "started":
        await asyncio.sleep(0.1)

    yield

    await mqtt_server.shutdown()


@pytest.mark.usefixtures("mqtt_server")
@pytest.fixture
async def mqtt_client():
    client = MQTTClient(
        client_id="helperbot@bumper/test",
        config={"check_hostname": False, "auto_reconnect": False},
    )

    await client.connect(f"mqtts://{HOST}:{MQTT_PORT}/", cafile=bumper.ca_cert)

    yield client

    await client.disconnect()


@pytest.fixture
async def conf_server_client(aiohttp_client):
    confserver = bumper.ConfServer("127.0.0.1:11111", False)
    confserver.confserver_app()

    client = await aiohttp_client(confserver.app)

    yield client

    await client.close()