def test_locust_on_message_exception(self, _mock_utils, _mock_paho, _mock_json): """ Should raise an exception on message callback """ client = MQTTClient("123", "987", False, False) with self.assertRaises(Exception): client.locust_on_message(client.mqttc, {}, {}) client.locust_on_message(client.mqttc, {}, None)
def test_locust_on_message(self, mock_utils, _mock_paho, mock_json): """ Should fire locust sucess on message callback """ message: mqtt.MQTTMessage = mqtt.MQTTMessage() message.payload = str.encode(str({"timestamp": 0})) mock_json.loads.return_value = {"timestamp": 0} client = MQTTClient("123", "987", False, False) client.locust_on_message(client.mqttc, {}, message) mock_utils.fire_locust_success.assert_called_once()
def test_locust_on_message_exception(self, mock_utils, mock_logging, mock_cert_client, paho_mqtt_mock, json_mock): """ Should raize an exception on message callback """ client = MQTTClient("123", "987", False, False) with self.assertRaises(Exception): MQTTClient.locust_on_message(client.mqttc, {}, {}) MQTTClient.locust_on_message(client.mqttc, {}, None) 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_message(self, mock_utils, mock_logging, mock_cert_client, paho_mqtt_mock, json_mock): """ Should fire locust sucess on message callback """ message: mqtt.MQTTMessage = mqtt.MQTTMessage() message.payload = str.encode(str({"timestamp": 0})) json_mock.loads.return_value = {"timestamp": 0} client = MQTTClient("123", "987", False, False) MQTTClient.locust_on_message(client.mqttc, {}, message) mock_utils.fire_locust_success.assert_called_once() mock_utils.time_delta.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()