"""
测试用模拟对象 — 不连接真实 MQTT Broker

提供 ``FakeMQTT``（模拟 amqtt ``MQTTClient`` 的最小接口）与辅助函数，
用于对 ``chongming_mqtt`` 各客户端做离线单元测试。
"""
import asyncio


class FakeMQTT:
    """
    模拟 amqtt MQTTClient 的最小接口

    记录 publish / subscribe / unsubscribe / disconnect 调用；
    ``on_publish`` 回调可在发布时模拟"对方应用"的行为（如回发响应）。
    """

    def __init__(self, on_publish=None):
        self.published: list[tuple[str, bytes, object]] = []  # (topic, payload, qos)
        self.subscribed: list[tuple[str, object]] = []        # (topic, qos)
        self.unsubscribed: list[str] = []
        self.disconnected = False
        self._on_publish = on_publish

    async def publish(self, topic, payload, qos):
        self.published.append((topic, payload, qos))
        if self._on_publish is not None:
            await self._on_publish(topic, payload)

    async def subscribe(self, topics):
        self.subscribed.extend(list(topics))

    async def unsubscribe(self, topics):
        self.unsubscribed.extend(list(topics))

    async def disconnect(self):
        self.disconnected = True


def attach(client, on_publish=None) -> FakeMQTT:
    """把 FakeMQTT 挂到被测客户端（跳过真实连接），并标记为运行中"""
    fake = FakeMQTT(on_publish=on_publish)
    client._client = fake
    client._running = True
    return fake


def run(coro):
    """运行一个异步协程（同步测试入口）"""
    return asyncio.run(coro)