def test_publishing_succes(self, mock_utils,
mock_logging,
mock_cert_client,
paho_mqtt_mock,
json_mock):
"""
Should publish a message successfully
"""
paho_mqtt_mock.Client().publish.return_value = (0, MagicMock())
client = MQTTClient("123", "987", False, False)
client.setup()
client.publishing()
paho_mqtt_mock.Client().publish.assert_called_once()
keys_len = len(client.pubmmap.keys())
self.assertGreater(keys_len, 0)
mock_utils.reset_mock()
mock_logging.reset_mock()
mock_cert_client.reset_mock()
paho_mqtt_mock.reset_mock()
json_mock.reset_mock()
def test_publishing_error(self, mock_utils,
mock_logging,
mock_cert_client,
paho_mqtt_mock,
json_mock):
"""
Should not publish a message successfully
"""
paho_mqtt_mock.Client().publish.return_value = (10, MagicMock())
client = MQTTClient("123", "987", False, False)
client.setup()
client.publishing()
mock_utils.error_message.assert_called_once_with(10)
mock_utils.fire_locust_failure.assert_called_once()
keys_len = len(client.pubmmap.keys())
self.assertEqual(keys_len, 0)
mock_utils.reset_mock()
mock_logging.reset_mock()
mock_cert_client.reset_mock()
paho_mqtt_mock.reset_mock()
json_mock.reset_mock()
