Skip to content

Commit ed446a0

Browse files
test getting the list of allowed configurations
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent ad0e527 commit ed446a0

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

tests/unit/test_sea_backend.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -545,6 +545,20 @@ def test_utility_methods(self, sea_client):
545545
assert len(configs) > 0
546546
assert "ANSI_MODE" in configs
547547

548+
# Test getting the list of allowed configurations with specific keys
549+
allowed_configs = SeaDatabricksClient.get_allowed_session_configurations()
550+
expected_keys = {
551+
"ANSI_MODE",
552+
"ENABLE_PHOTON",
553+
"LEGACY_TIME_PARSER_POLICY",
554+
"MAX_FILE_PARTITION_BYTES",
555+
"READ_ONLY_EXTERNAL_METASTORE",
556+
"STATEMENT_TIMEOUT",
557+
"TIMEZONE",
558+
"USE_CACHED_RESULT",
559+
}
560+
assert set(allowed_configs) == expected_keys
561+
548562
# Test _extract_description_from_manifest
549563
manifest_obj = MagicMock()
550564
manifest_obj.schema = {

0 commit comments

Comments
 (0)