def test_locust_on_connect_failure(self, mock_utils, mock_paho):
"""
Should fire locust failure on connection callback
"""
mock_paho.MQTT_ERR_SUCCESS = 1
client = MQTTClient("123", "987", False, False)
client.locust_on_connect(client.mqttc, {}, {}, 101010)
mock_utils.error_message.assert_called_once()
mock_utils.fire_locust_failure.assert_called_once()
def test_locust_on_connect(self, mock_utils, mock_paho):
"""
Should fire locust success on connection callback
"""
mock_paho.MQTT_ERR_SUCCESS = 1
client = MQTTClient("123", "987", False, False)
client.subscribe = MagicMock()
client.locust_on_connect(client.mqttc, {}, {}, 1)
client.subscribe.assert_called_once()
mock_utils.fire_locust_success.assert_called_once()
def test_locust_on_connect_failure(self, mock_utils,
mock_logging,
mock_cert_client,
paho_mqtt_mock,
json_mock):
"""
Should fire locust failure on connection callback
"""
paho_mqtt_mock.MQTT_ERR_SUCCESS = 1
client = MQTTClient("123", "987", False, False)
client.locust_on_connect(client.mqttc, {}, {}, 101010)
mock_utils.error_message.assert_called_once()
mock_utils.fire_locust_failure.assert_called_once()
mock_utils.reset_mock()
mock_logging.reset_mock()
mock_cert_client.reset_mock()
paho_mqtt_mock.reset_mock()
json_mock.reset_mock()
def test_locust_on_connect(self, mock_utils,
mock_logging,
mock_cert_client,
paho_mqtt_mock,
json_mock):
"""
Should fire locust success on connection callback
"""
paho_mqtt_mock.MQTT_ERR_SUCCESS = 1
client = MQTTClient("123", "987", False, False)
client.subscribing = MagicMock()
client.locust_on_connect(client.mqttc, {}, {}, 1)
client.subscribing.assert_called_once()
mock_utils.fire_locust_success.assert_called_once()
mock_utils.reset_mock()
mock_logging.reset_mock()
mock_cert_client.reset_mock()
paho_mqtt_mock.reset_mock()
json_mock.reset_mock()
