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()