mirror of https://github.com/databricks/cli.git
add acc test for oauth'
This commit is contained in:
parent
5056070d11
commit
ae8ebaa757
|
@ -0,0 +1,45 @@
|
||||||
|
{
|
||||||
|
"method": "GET",
|
||||||
|
"path": "/oidc/.well-known/oauth-authorization-server"
|
||||||
|
}
|
||||||
|
{
|
||||||
|
"headers": {
|
||||||
|
"Authorization": [
|
||||||
|
"Basic [ENCODED_AUTH]"
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"method": "POST",
|
||||||
|
"path": "/oidc/v1/token",
|
||||||
|
"raw_body": "grant_type=client_credentials\u0026scope=all-apis"
|
||||||
|
}
|
||||||
|
{
|
||||||
|
"method": "GET",
|
||||||
|
"path": "/oidc/.well-known/oauth-authorization-server"
|
||||||
|
}
|
||||||
|
{
|
||||||
|
"headers": {
|
||||||
|
"Authorization": [
|
||||||
|
"Basic [ENCODED_AUTH]"
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"method": "POST",
|
||||||
|
"path": "/oidc/v1/token",
|
||||||
|
"raw_body": "grant_type=client_credentials\u0026scope=all-apis"
|
||||||
|
}
|
||||||
|
{
|
||||||
|
"headers": {
|
||||||
|
"Authorization": [
|
||||||
|
"Bearer oauth-token"
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"method": "POST",
|
||||||
|
"path": "/telemetry-ext",
|
||||||
|
"body": {
|
||||||
|
"uploadTime": "UNIX_TIME_MILLIS",
|
||||||
|
"items": [],
|
||||||
|
"protoLogs": [
|
||||||
|
"{\"frontend_log_event_id\":\"[UUID]\",\"entry\":{\"databricks_cli_log\":{\"execution_context\":{\"cmd_exec_id\":\"[UUID]\",\"version\":\"[DEV_VERSION]\",\"command\":\"selftest_send-telemetry\",\"operating_system\":\"OS\",\"execution_time_ms\":\"SMALL_INT\",\"exit_code\":0},\"cli_test_event\":{\"name\":\"VALUE1\"}}}}",
|
||||||
|
"{\"frontend_log_event_id\":\"[UUID]\",\"entry\":{\"databricks_cli_log\":{\"execution_context\":{\"cmd_exec_id\":\"[UUID]\",\"version\":\"[DEV_VERSION]\",\"command\":\"selftest_send-telemetry\",\"operating_system\":\"OS\",\"execution_time_ms\":\"SMALL_INT\",\"exit_code\":0},\"cli_test_event\":{\"name\":\"VALUE2\"}}}}"
|
||||||
|
]
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,5 @@
|
||||||
|
Telemetry logs uploaded successfully
|
||||||
|
Response:
|
||||||
|
{"errors":null,"numProtoSuccess":2}
|
||||||
|
|
||||||
|
2025/02/21 16:56:09 INFO Ignoring basic auth, because oauth-m2m is preferred sdk=true
|
|
@ -0,0 +1,4 @@
|
||||||
|
|
||||||
|
>>> [CLI] selftest send-telemetry
|
||||||
|
waiting for telemetry process to finish
|
||||||
|
[wait_pid] process has ended
|
|
@ -1,7 +1,12 @@
|
||||||
export DATABRICKS_CLI_TELEMETRY_PID_FILE=./telemetry.pid
|
export DATABRICKS_CLI_TELEMETRY_PID_FILE=./telemetry.pid
|
||||||
export DATABRICKS_CLI_TELEMETRY_UPLOAD_LOGS_FILE=./out.upload_process.txt
|
export DATABRICKS_CLI_TELEMETRY_UPLOAD_LOGS_FILE=./out.upload_process.txt
|
||||||
|
|
||||||
# TODO: Move the environment variable calculation to the daemon library.
|
# Unset the token which is configured by default
|
||||||
|
# in acceptance tests
|
||||||
|
export DATABRICKS_TOKEN=""
|
||||||
|
|
||||||
|
export DATABRICKS_CLIENT_ID=client_id
|
||||||
|
export DATABRICKS_CLIENT_SECRET=client_secret
|
||||||
|
|
||||||
trace $CLI selftest send-telemetry
|
trace $CLI selftest send-telemetry
|
||||||
|
|
|
@ -0,0 +1,19 @@
|
||||||
|
[[Server]]
|
||||||
|
Pattern = "POST /telemetry-ext"
|
||||||
|
Response.Body = '''
|
||||||
|
{
|
||||||
|
"numProtoSuccess": 2
|
||||||
|
}
|
||||||
|
'''
|
||||||
|
|
||||||
|
# TODO: Remove the INFO log line.
|
||||||
|
|
||||||
|
[[Repls]]
|
||||||
|
Old = 'execution_time_ms\\\":\d{1,5},'
|
||||||
|
New = 'execution_time_ms\":\"SMALL_INT\",'
|
||||||
|
|
||||||
|
# "client_id:client_secret" in base64 is Y2xpZW50X2lkOmNsaWVudF9zZWNyZXQ=, expect to
|
||||||
|
# see this in Authorization header
|
||||||
|
[[Repls]]
|
||||||
|
Old = "Y2xpZW50X2lkOmNsaWVudF9zZWNyZXQ="
|
||||||
|
New = "[ENCODED_AUTH]"
|
|
@ -1,11 +0,0 @@
|
||||||
[[Server]]
|
|
||||||
Pattern = "POST /telemetry-ext"
|
|
||||||
Response.Body = '''
|
|
||||||
{
|
|
||||||
"numProtoSuccess": 2
|
|
||||||
}
|
|
||||||
'''
|
|
||||||
|
|
||||||
[[Repls]]
|
|
||||||
Old = 'execution_time_ms\\\":\d{1,5},'
|
|
||||||
New = 'execution_time_ms\":\"SMALL_INT\",'
|
|
Loading…
Reference in New Issue