"""Test kafka connector.""" from product_review.connectors.kafka import get_producer def test_get_producer(mocker, app_context): """Test get_producer.""" mocker.patch( "product_review.config.KAFKA_BOOTSTRAP_SERVERS", "kafka-bootstrap-servers" ) mock_event_producer = mocker.patch("product_review.connectors.kafka.EventProducer") mock_string_serializer = mocker.patch( "product_review.connectors.kafka.StringSerializer" ) mock_json_serializer = mocker.patch( "product_review.connectors.kafka.SimpleJSONSerializer" ) get_producer.cache_clear() producer = get_producer() mock_event_producer.assert_called_once_with( bootstrap_servers="kafka-bootstrap-servers", key_serializer=mock_string_serializer.return_value, value_serializer=mock_json_serializer.return_value, security_protocol="SSL", ) assert producer is mock_event_producer.return_value