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