def test_mqtt_disconnect_tries_reconnect(self, mock_sleep): mqttc = mock.MagicMock() mqttc.reconnect.side_effect = [1, 1, 1, 0] mqtt._mqtt_on_disconnect(mqttc, {}, 1) self.assertTrue(mqttc.reconnect.called) self.assertEqual(4, len(mqttc.reconnect.mock_calls)) self.assertEqual([1, 2, 4], [call[1][0] for call in mock_sleep.mock_calls])
def test_mqtt_disconnect_tries_no_reconnect_on_stop(self): mqttc = mock.MagicMock() mqtt._mqtt_on_disconnect(mqttc, {}, 0) self.assertFalse(mqttc.reconnect.called)