From 87edcfdbb0dddd00329e074ff82f7b26f327460d Mon Sep 17 00:00:00 2001 From: v-chen_data Date: Wed, 25 Sep 2024 15:26:44 -0700 Subject: [PATCH] update test --- .../data_prep/test_convert_delta_to_json.py | 20 +++++++++++++++---- 1 file changed, 16 insertions(+), 4 deletions(-) diff --git a/tests/a_scripts/data_prep/test_convert_delta_to_json.py b/tests/a_scripts/data_prep/test_convert_delta_to_json.py index e623467bf7..f7d492f1a9 100644 --- a/tests/a_scripts/data_prep/test_convert_delta_to_json.py +++ b/tests/a_scripts/data_prep/test_convert_delta_to_json.py @@ -229,7 +229,10 @@ def test_dbconnect_called( DATABRICKS_TOKEN = 'token' use_serverless = False - mock_cluster_response = Namespace(spark_version='14.1.0-scala2.12') + mock_cluster_response = Namespace( + spark_version='14.1.0-scala2.12', + data_security_mode='SINGLE_USER', + ) mock_workspace_client.return_value.clusters.get.return_value = mock_cluster_response mock_remote = MagicMock() @@ -286,7 +289,10 @@ def test_sqlconnect_called_dbr13( DATABRICKS_TOKEN = 'token' use_serverless = False - mock_cluster_response = Namespace(spark_version='13.0.0-scala2.12') + mock_cluster_response = Namespace( + spark_version='13.0.0-scala2.12', + data_security_mode='SINGLE_USER', + ) mock_workspace_client.return_value.clusters.get.return_value = mock_cluster_response fetch_DT( @@ -338,7 +344,10 @@ def test_sqlconnect_called_dbr14( DATABRICKS_TOKEN = 'token' use_serverless = False - mock_cluster_response = Namespace(spark_version='14.2.0-scala2.12') + mock_cluster_response = Namespace( + spark_version='14.2.0-scala2.12', + data_security_mode='SINGLE_USER', + ) mock_workspace_client.return_value.clusters.get.return_value = mock_cluster_response fetch_DT( @@ -390,7 +399,10 @@ def test_sqlconnect_called_https( DATABRICKS_TOKEN = 'token' use_serverless = False - mock_cluster_response = Namespace(spark_version='14.2.0-scala2.12') + mock_cluster_response = Namespace( + spark_version='14.2.0-scala2.12', + data_security_mode='SINGLE_USER', + ) mock_workspace_client.return_value.clusters.get.return_value = mock_cluster_response fetch_DT(