Commit 530f299 1 parent 46aa5ed commit 530f299 Copy full SHA for 530f299
File tree 1 file changed +7
-5
lines changed
metadata-ingestion-modules/airflow-plugin/tests/integration
1 file changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -217,11 +217,6 @@ def _run_airflow(
217
217
conn_type = "datahub-file" ,
218
218
host = str (meta_file ),
219
219
).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 (),
225
220
# Configure fake credentials for the Snowflake connection.
226
221
"AIRFLOW_CONN_MY_SNOWFLAKE" : Connection (
227
222
conn_id = "my_snowflake" ,
@@ -262,6 +257,13 @@ def _run_airflow(
262
257
"SQLALCHEMY_SILENCE_UBER_WARNING" : "1" ,
263
258
}
264
259
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
+
265
267
if not HAS_AIRFLOW_STANDALONE_CMD :
266
268
raise pytest .skip ("Airflow standalone command is not available" )
267
269
You can’t perform that action at this time.
0 commit comments