kopia lustrzana https://github.com/Yakifo/amqtt
95 wiersze
3.0 KiB
Python
95 wiersze
3.0 KiB
Python
import asyncio
|
|
import logging
|
|
from importlib.metadata import EntryPoint
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from amqtt.broker import Broker
|
|
from amqtt.client import MQTTClient
|
|
from amqtt.mqtt.constants import QOS_0
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
all_sys_topics = [
|
|
'$SYS/broker/version',
|
|
'$SYS/broker/load/bytes/received',
|
|
'$SYS/broker/load/bytes/sent',
|
|
'$SYS/broker/messages/received',
|
|
'$SYS/broker/messages/sent',
|
|
'$SYS/broker/time',
|
|
'$SYS/broker/uptime',
|
|
'$SYS/broker/uptime/formatted',
|
|
'$SYS/broker/clients/connected',
|
|
'$SYS/broker/clients/disconnected',
|
|
'$SYS/broker/clients/maximum',
|
|
'$SYS/broker/clients/total',
|
|
'$SYS/broker/messages/inflight',
|
|
'$SYS/broker/messages/inflight/in',
|
|
'$SYS/broker/messages/inflight/out',
|
|
'$SYS/broker/messages/inflight/stored',
|
|
'$SYS/broker/messages/publish/received',
|
|
'$SYS/broker/messages/publish/sent',
|
|
'$SYS/broker/messages/retained/count',
|
|
'$SYS/broker/messages/subscriptions/count',
|
|
'$SYS/broker/heap/size',
|
|
'$SYS/broker/heap/maximum',
|
|
'$SYS/broker/cpu/percent',
|
|
'$SYS/broker/cpu/maximum',
|
|
]
|
|
|
|
|
|
|
|
# test broker sys
|
|
@pytest.mark.asyncio
|
|
async def test_broker_sys_plugin() -> None:
|
|
|
|
sys_topic_flags = {sys_topic:False for sys_topic in all_sys_topics}
|
|
|
|
class MockEntryPoints:
|
|
|
|
def select(self, group) -> list[EntryPoint]:
|
|
match group:
|
|
case 'tests.mock_plugins':
|
|
return [
|
|
EntryPoint(name='BrokerSysPlugin', group='tests.mock_plugins', value='amqtt.plugins.sys.broker:BrokerSysPlugin'),
|
|
]
|
|
case _:
|
|
return list()
|
|
|
|
|
|
with patch("amqtt.plugins.manager.entry_points", side_effect=MockEntryPoints) as mocked_mqtt_publish:
|
|
|
|
config = {
|
|
"listeners": {
|
|
"default": {"type": "tcp", "bind": "127.0.0.1:1883", "max_connections": 10},
|
|
},
|
|
'sys_interval': 1
|
|
}
|
|
|
|
broker = Broker(plugin_namespace='tests.mock_plugins', config=config)
|
|
await broker.start()
|
|
client = MQTTClient()
|
|
await client.connect("mqtt://127.0.0.1:1883/")
|
|
await client.subscribe([("$SYS/#", QOS_0),])
|
|
await client.publish('test/topic', b'my test message')
|
|
await asyncio.sleep(2)
|
|
sys_msg_count = 0
|
|
try:
|
|
while sys_msg_count < 30:
|
|
message = await client.deliver_message(timeout_duration=1)
|
|
if '$SYS' in message.topic:
|
|
sys_msg_count += 1
|
|
assert message.topic in sys_topic_flags
|
|
sys_topic_flags[message.topic] = True
|
|
|
|
except asyncio.TimeoutError:
|
|
logger.debug(f"TimeoutError after {sys_msg_count} messages")
|
|
|
|
await client.disconnect()
|
|
await broker.shutdown()
|
|
|
|
assert sys_msg_count > 1
|
|
|
|
assert all(sys_topic_flags.values()), f'topic not received: {[ topic for topic, flag in sys_topic_flags.items() if not flag ]}'
|