@@ -94,7 +94,7 @@ def sql_backend(ws, env_or_skip) -> SqlBackend:
9494
9595
9696@retried (on = [NotFound , InvalidParameterValue ], timeout = timedelta (minutes = 5 ))
97- @pytest .mark .parametrize ('prepare_tables_for_migration' , [( 'regular' ) ], indirect = True )
97+ @pytest .mark .parametrize ('prepare_tables_for_migration' , ['regular' ], indirect = True )
9898def test_migration_job_ext_hms (ws , installation_ctx , prepare_tables_for_migration , env_or_skip ):
9999 # this test spins up clusters using ext hms cluster policy, which will have a startup time of ~ 7-10m
100100 # skip this test if not in nightly test job or debug mode
@@ -143,9 +143,16 @@ def test_running_real_assessment_job_ext_hms(
143143 if os .path .basename (sys .argv [0 ]) not in {"_jb_pytest_runner.py" , "testlauncher.py" }:
144144 env_or_skip ("TEST_NIGHTLY" )
145145
146+ ext_hms_cluster_id = env_or_skip ("TEST_EXT_HMS_CLUSTER_ID" )
146147 ext_hms_ctx = installation_ctx .replace (
147148 skip_dashboards = True ,
148- config_transform = lambda wc : dataclasses .replace (wc , override_clusters = None ),
149+ config_transform = lambda wc : dataclasses .replace (
150+ wc ,
151+ override_clusters = {
152+ "main" : ext_hms_cluster_id ,
153+ "table_migration" : ext_hms_cluster_id ,
154+ },
155+ ),
149156 extend_prompts = {
150157 r"Instance pool id to be set.*" : env_or_skip ("TEST_INSTANCE_POOL_ID" ),
151158 r".*Do you want to update the existing installation?.*" : 'yes' ,
0 commit comments