def test_publish_success(self, _mock_utils, mock_paho, _mock_json):
"""
Should publish a message successfully
"""
mock_paho.Client().publish.return_value = (0, MagicMock())
client = MQTTClient("123", "987", False, False)
client.publish()
mock_paho.Client().publish.assert_called_once()
keys_len = len(client.pubmmap.keys())
self.assertGreater(keys_len, 0)
def test_publish_error(self, mock_utils, mock_paho, _mock_json):
"""
Should not publish a message successfully
"""
mock_paho.Client().publish.return_value = (10, MagicMock())
client = MQTTClient("123", "987", False, False)
client.publish()
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)
