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