def test_locust_on_publish_failure(self, mock_utils, _mock_paho):
"""
Should fire locust failure on publish
"""
client = MQTTClient("123", "987", False, False)
client.locust_on_publish(client.mqttc, {}, 0)
mock_utils.fire_locust_failure.assert_called_once()
def test_locust_on_publish(self, mock_utils, _mock_paho):
"""
Should fire locust success on publish callback
"""
mid = MagicMock()
client = MQTTClient("123", "987", False, False)
client.pubmmap[mid] = {
'name': 'name',
'start_time': 'time',
'payload': 'payload'
}
client.locust_on_publish(client.mqttc, {}, mid)
mock_utils.fire_locust_success.assert_called_once()
def test_locust_on_publish_failure(self, mock_utils,
mock_logging,
mock_cert_client,
paho_mqtt_mock,
json_mock):
"""
Should fire locust failure on publish
"""
client = MQTTClient("123", "987", False, False)
client.locust_on_publish(client.mqttc, {}, 0)
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_publish(self, mock_utils,
mock_logging,
mock_cert_client,
paho_mqtt_mock,
json_mock):
"""
Should fire locust success on publish callback
"""
mid = MagicMock()
client = MQTTClient("123", "987", False, False)
client.pubmmap[mid] = {'name': 'name',
'start_time': 'time', 'payload': 'payload'}
client.locust_on_publish(client.mqttc, {}, mid)
mock_utils.time_delta.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()
