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