示例#1
0
    def test_reconnect(self, _mock_json, _mock_paho, _mock_utils,
                       _mock_cert_utils):
        """
        Should reconnect successfully.
        """
        client = MQTTClient("123", "987", False, False)

        client.reconnect()

        client.mqttc.reconnect.assert_called_once()
示例#2
0
    def test_reconnect_after_renew(self, _mock_json, _mock_paho, _mock_utils,
                                   _mock_cert_utils):
        """
        Should reconnect successfully after being renewed.
        """
        client = MQTTClient("123", "987", False, True)
        client.is_renewed = True

        mock_configure_mqtt = patch.object(client, "configure_mqtt").start()
        mock_connect = patch.object(client, "connect").start()
        mock_mqttc = patch.object(client, "mqttc").start()

        client.reconnect()

        mock_mqttc.loop_stop.assert_called_once()
        mock_connect.assert_called_once()
        mock_configure_mqtt.assert_called_once()

        mock_connect.stop()
        mock_configure_mqtt.stop()
        mock_mqttc.stop()
示例#3
0
    def test_not_reconnect(self, _mock_json, _mock_paho, _mock_utils,
                           _mock_cert_utils):
        """
        Should not reconnect because disconnect_forever is True
        """
        client = MQTTClient("123", "987", False, False)
        client.disconnect_forever = True

        mock_configure_mqtt = patch.object(client, "configure_mqtt").start()
        mock_connect = patch.object(client, "connect").start()
        mock_mqttc = patch.object(client, "mqttc").start()

        client.reconnect()

        mock_mqttc.loop_stop.assert_not_called()
        mock_connect.assert_not_called()
        mock_configure_mqtt.assert_not_called()

        mock_mqttc.reconnect.assert_not_called()

        mock_connect.stop()
        mock_configure_mqtt.stop()
        mock_mqttc.stop()