Skip to content

Commit 530f299

Browse files
committed
Fix test
1 parent 46aa5ed commit 530f299

File tree

1 file changed

+7
-5
lines changed
  • metadata-ingestion-modules/airflow-plugin/tests/integration

1 file changed

+7
-5
lines changed

metadata-ingestion-modules/airflow-plugin/tests/integration/test_plugin.py

+7-5
Original file line numberDiff line numberDiff line change
@@ -217,11 +217,6 @@ def _run_airflow(
217217
conn_type="datahub-file",
218218
host=str(meta_file),
219219
).get_uri(),
220-
f"AIRFLOW_CONN_{datahub_connection_name_2.upper()}": Connection(
221-
conn_id="datahub_file_default2",
222-
conn_type="datahub-file",
223-
host=str(meta_file2),
224-
).get_uri(),
225220
# Configure fake credentials for the Snowflake connection.
226221
"AIRFLOW_CONN_MY_SNOWFLAKE": Connection(
227222
conn_id="my_snowflake",
@@ -262,6 +257,13 @@ def _run_airflow(
262257
"SQLALCHEMY_SILENCE_UBER_WARNING": "1",
263258
}
264259

260+
if multiple_connections and not is_v1:
261+
environment[f"AIRFLOW_CONN_{datahub_connection_name_2.upper()}"] = Connection(
262+
conn_id="datahub_file_default2",
263+
conn_type="datahub-file",
264+
host=str(meta_file2)
265+
).get_uri()
266+
265267
if not HAS_AIRFLOW_STANDALONE_CMD:
266268
raise pytest.skip("Airflow standalone command is not available")
267269

0 commit comments

Comments
 (0)