Skip to content

Commit a051188

Browse files
fix: add auth_type:oauth for databricks, catch binascii.Error
Co-Authored-By: Itamar Hartstein <haritamar@gmail.com>
1 parent b0cc1f4 commit a051188

2 files changed

Lines changed: 3 additions & 1 deletion

File tree

tests/profiles/generate_profiles.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
from __future__ import annotations
55

66
import base64
7+
import binascii
78
import json
89
import os
910
from pathlib import Path
@@ -83,7 +84,7 @@ def main(
8384
if secrets_b64:
8485
try:
8586
decoded: dict = json.loads(base64.b64decode(secrets_b64))
86-
except (ValueError, json.JSONDecodeError) as e:
87+
except (binascii.Error, json.JSONDecodeError) as e:
8788
raise click.ClickException(
8889
f"Failed to decode ${secrets_json_env}: {e}"
8990
) from e

tests/profiles/profiles.yml.j2

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ elementary_tests:
6060
http_path: {{ databricks_http_path }}
6161
catalog: {{ databricks_catalog }}
6262
schema: {{ schema_name }}
63+
auth_type: oauth
6364
client_id: {{ databricks_client_id }}
6465
client_secret: {{ databricks_client_secret }}
6566
threads: 4

0 commit comments

Comments
 (0)