Coverage for tests / brokers / mqtt / test_consume.py: 97%

1import asyncio (empty)

2from unittest.mock import MagicMock (empty)

3 

4import pytest (empty)

5 

6from tests.brokers.base.consume import BrokerRealConsumeTestcase (empty)

7 

8from .basic import MQTTTestcaseConfig (empty)

9 

10 

11@pytest.mark.connected() (empty)

12@pytest.mark.mqtt() (empty)

13@pytest.mark.asyncio() (empty)

14class TestConsume(MQTTTestcaseConfig, BrokerRealConsumeTestcase): (empty)

15 async def test_consume_with_filter(self, queue, mock): (empty)

16 if self.version == "3.1.1": (empty)

17 pytest.skip("content_type filtering not supported in MQTT 3.1.1") (empty)

18 await super().test_consume_with_filter(queue, mock) (empty)

19 

20 @pytest.mark.asyncio() (empty)

21 async def test_iteration( (empty)

22 self, 

23 queue: str, 

24 mock: MagicMock, 

25 ) -> None: 

26 # Overridden since order is not guaranteed in MQTT. 

27 expected_messages = {"test_message_1", "test_message_2"} (empty)

28 

29 broker = self.get_broker(apply_types=True) (empty)

30 

31 args, kwargs = self.get_subscriber_params(queue) (empty)

32 subscriber = broker.subscriber(*args, **kwargs) (empty)

33 

34 async with self.patch_broker(broker) as br: (empty)

35 await br.start() (empty)

36 

37 async def publish_test_message(): (empty)

38 for msg in expected_messages: (empty)

39 await br.publish(msg, queue) (empty)

40 

41 async def consume(): (empty)

42 index_message = 0 (empty)

43 async for msg in subscriber: 43 ↛ exitline 43 didn't return from function 'consume' because the loop on line 43 didn't complete(empty)

44 result_message = await msg.decode() (empty)

45 

46 mock(result_message) (empty)

47 

48 index_message += 1 (empty)

49 if index_message >= len(expected_messages): (empty)

50 break (empty)

51 

52 await asyncio.wait( (empty)

53 ( 

54 asyncio.create_task(consume()), 

55 asyncio.create_task(publish_test_message()), 

56 ), 

57 timeout=self.timeout, 

58 ) 

59 

60 msgs = {call.args[0] for call in mock.mock_calls} (empty)

61 assert msgs == expected_messages (empty)