File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed
Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change 11"""Tests for the DeviceManager class."""
22
33from collections .abc import Generator
4- from unittest .mock import patch
4+ from unittest .mock import Mock , patch
55
66import pytest
77
1616
1717
1818@pytest .fixture (autouse = True )
19- def setup_mqtt_session () -> Generator [None , None , None ]:
19+ def setup_mqtt_session () -> Generator [Mock , None , None ]:
2020 """Fixture to set up the MQTT session for the tests."""
21- with patch ("roborock.devices.device_manager.create_mqtt_session" ):
22- yield
21+ with patch ("roborock.devices.device_manager.create_mqtt_session" ) as mock_create_session :
22+ mock_unsub = Mock ()
23+ mock_create_session .return_value .subscribe .return_value = mock_unsub
24+ yield mock_create_session
2325
2426
2527async def home_home_data_no_devices () -> HomeData :
You can’t perform that action at this time.
0 commit comments