Skip to content

Commit 7985639

Browse files
fix param type in unit tests
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 077a71c commit 7985639

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/unit/test_sea_backend.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
_filter_session_configuration,
1414
)
1515
from databricks.sql.backend.types import SessionId, CommandId, CommandState, BackendType
16+
from databricks.sql.thrift_api.TCLIService import ttypes
1617
from databricks.sql.types import SSLOptions
1718
from databricks.sql.auth.authenticators import AuthProvider
1819
from databricks.sql.exc import (
@@ -355,7 +356,7 @@ def test_command_execution_advanced(
355356
"status": {"state": "SUCCEEDED"},
356357
}
357358
mock_http_client._make_request.return_value = execute_response
358-
param = {"name": "param1", "value": "value1", "type": "STRING"}
359+
param = ttypes.TSparkParameter(name="param1", value="value1", type="STRING")
359360

360361
with patch.object(sea_client, "get_execution_result"):
361362
sea_client.execute_command(

0 commit comments

Comments
 (0)