@@ -134,7 +134,7 @@ def test_mqtt_on_connect(self, base_client: MockBaseDeyeMqttClient) -> None:
134134 # Call _mqtt_on_connect
135135 with patch .object (base_client ._mqtt , "subscribe" ) as mock_subscribe :
136136 with patch .object (base_client ._mqtt , "publish" ) as mock_publish :
137- base_client ._mqtt_on_connect (base_client ._mqtt , None , {}, 0 )
137+ base_client ._mqtt_on_connect (base_client ._mqtt , None , {}, 0 , {} )
138138 mock_subscribe .assert_called_once_with (topic1 )
139139 mock_publish .assert_called_once_with (pending_topic , pending_command )
140140 assert len (base_client ._pending_commands ) == 0
@@ -278,7 +278,7 @@ async def test_set_mqtt_info(self, classic_client: DeyeClassicMqttClient) -> Non
278278 assert classic_client ._mqtt_host == "test.mqtt.host"
279279 assert classic_client ._mqtt_ssl_port == 8883
280280 assert classic_client ._endpoint == "test_endpoint"
281- classic_client ._mqtt .username_pw_set .assert_called_once_with (
281+ cast ( MagicMock , classic_client ._mqtt ) .username_pw_set .assert_called_once_with (
282282 "test_user" , "test_password"
283283 )
284284
@@ -458,7 +458,7 @@ async def test_set_mqtt_info(self, fog_client: DeyeFogMqttClient) -> None:
458458 assert fog_client ._mqtt_host == "test.mqtt.host"
459459 assert fog_client ._mqtt_ssl_port == 8883
460460 assert fog_client ._topic == "fogcloud/app/test_user/sub"
461- fog_client ._mqtt .username_pw_set .assert_called_once_with (
461+ cast ( MagicMock , fog_client ._mqtt ) .username_pw_set .assert_called_once_with (
462462 "test_user" , "test_password"
463463 )
464464
0 commit comments