예제 #1
0
def test_get_url():
    client = ApiClient(host='https://databricks.com', jobs_api_version='2.1')
    assert client.get_url('') == 'https://databricks.com/api/2.0'
    assert client.get_url('/') == 'https://databricks.com/api/2.0/'
    assert client.get_url(
        '/endpoint') == 'https://databricks.com/api/2.0/endpoint'
    assert client.get_url(
        '/jobs/list') == 'https://databricks.com/api/2.1/jobs/list'
    assert client.get_url('/jobs/list',
                          '3.0') == 'https://databricks.com/api/3.0/jobs/list'
예제 #2
0
def test_api_client_url_parsing():
    client = ApiClient(host='https://databricks.com')
    assert client.get_url('') == 'https://databricks.com/api/2.0'

    client = ApiClient(host='https://databricks.com/?o=123')
    assert client.get_url('') == 'https://databricks.com/api/2.0'

    client = ApiClient(host='https://databricks.com?o=123')
    assert client.get_url('') == 'https://databricks.com/api/2.0'

    # NOTE: this technically is not possible since we validate that the "host" has a prefix of https:// in
    # databricks_cli.configure.cli
    client = ApiClient(host='http://databricks.com')
    assert client.get_url('') == 'http://databricks.com/api/2.0'