add env variable for kafka url

smoke-test/test_e2e.py

@@ -11,12 +11,11 @@
 from tests.utils import (
     get_frontend_url,
     get_gms_url,
+    get_kafka_broker_url,
     get_sleep_info,
     ingest_file_via_rest,
 )
 
-KAFKA_BROKER = "localhost:9092"
-
 bootstrap_sample_data = "../metadata-ingestion/examples/mce_files/bootstrap_mce.json"
 usage_sample_data = (
     "../metadata-ingestion/tests/integration/bigquery-usage/bigquery_usages_golden.json"
@@ -139,7 +138,7 @@ def test_ingestion_via_kafka(wait_for_healthchecks):
                 "type": "datahub-kafka",
                 "config": {
                     "connection": {
-                        "bootstrap": KAFKA_BROKER,
+                        "bootstrap": get_kafka_broker_url(),
                     }
                 },
             },

smoke-test/tests/utils.py

@@ -15,6 +15,10 @@ def get_frontend_url():
     return os.getenv("DATAHUB_FRONTEND_URL") or "http://localhost:9002"
 
 
+def get_kafka_broker_url():
+    return os.getenv("DATAHUB_KAFKA_URL") or "localhost:9092"
+
+
 def get_sleep_info():
     return (
         os.environ.get("DATAHUB_TEST_SLEEP_BETWEEN") or 60,