When the token is used in the databricks connection, the host name is being pulled from extra json data

https://github.com/apache/airflow/blob/fdd9b6f65b608c516b8a062b058972d9a45ec9e3/airflow/providers/databricks/hooks/databricks.py#L160


I think the host name should be pulled from databricks_conn directly as the line 164, where basic auth is being used https://github.com/apache/airflow/blob/fdd9b6f65b608c516b8a062b058972d9a45ec9e3/airflow/providers/databricks/hooks/databricks.py#L164

Thanks!

Liusong


tests/providers/databricks/hooks/test_databricks.py
STATE_MESSAGE = 'Waiting for cluster'
GET_RUN_RESPONSE = {
    'job_id': JOB_ID,
    'run_page_url': RUN_PAGE_URL,
    'state': {'life_cycle_state': LIFE_CYCLE_STATE, 'state_message': STATE_MESSAGE},
}
NOTEBOOK_PARAMS = {"dry-run": "true", "oldest-time-to-consider": "1457570074236"}
JAR_PARAMS = ["param1", "param2"]
RESULT_STATE = None  
def run_now_endpoint(host):
    """
    Utility function to generate the run now endpoint given the host.
    """
    return 'https://{}/api/2.0/jobs/run-now'.format(host)
def submit_run_endpoint(host):
    """
    Utility function to generate the submit run endpoint given the host.
    """
    return 'https://{}/api/2.0/jobs/runs/submit'.format(host)
def get_run_endpoint(host):
    """
    Utility function to generate the get run endpoint given the host.
    """
    return 'https://{}/api/2.0/jobs/runs/get'.format(host)
def cancel_run_endpoint(host):
    """
    Utility function to generate the get run endpoint given the host.
    """
    return 'https://{}/api/2.0/jobs/runs/cancel'.format(host)
def start_cluster_endpoint(host):