Skip to content

Commit 53a555f

Browse files
update unit tests to pass ttypes.TSparkParameter
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 0682761 commit 53a555f

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 (
@@ -354,7 +355,7 @@ def test_command_execution_advanced(
354355
"status": {"state": "SUCCEEDED"},
355356
}
356357
mock_http_client._make_request.return_value = execute_response
357-
param = {"name": "param1", "value": "value1", "type": "STRING"}
358+
param = ttypes.TSparkParameter(name="param1", value="value1", type="STRING")
358359

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

0 commit comments

Comments
 (0)