def test_get_config_throw_exception_if_profile_invalid(): invalid_config = DatabricksConfig.from_token(None, None) update_and_persist_config(DEFAULT_SECTION, invalid_config) with pytest.raises(InvalidConfigurationError): get_config()
def get_config(self): return DatabricksConfig.from_token("Override", "Token!")
import shutil import tempfile import unittest from uuid import uuid4 from click.testing import CliRunner from cookiecutter.main import cookiecutter from databricks_cli.configure.provider import DatabricksConfig from path import Path from retry import retry CICD_TEMPLATES_URI = "https://github.com/databrickslabs/cicd-templates.git" TEST_HOST = "https:/dbx.cloud.databricks.com" TEST_TOKEN = "dapiDBXTEST" test_dbx_config = DatabricksConfig.from_token(TEST_HOST, TEST_TOKEN) @retry(tries=10, delay=5, backoff=5) def initialize_cookiecutter(project_name): cookiecutter( CICD_TEMPLATES_URI, no_input=True, extra_context={"project_name": project_name} ) def invoke_cli_runner(*args, **kwargs): """ Helper method to invoke the CliRunner while asserting that the exit code is actually 0. """ expected_error = ( kwargs.pop("expected_error") if kwargs.get("expected_error") else None
def wrapper(test, *args, **kwargs): config = DatabricksConfig.from_token('test-host', 'test-token') update_and_persist_config(DEFAULT_SECTION, config) return test(*args, **kwargs)
def test_databricks_params_token(get_config): get_config.return_value = DatabricksConfig.from_token("host", "mytoken", insecure=False) params = databricks_utils.get_databricks_host_creds() assert params.host == "host" assert params.token == "mytoken" assert not params.ignore_tls_verification
def test_is_valid(self): config = DatabricksConfig.from_password(TEST_HOST, TEST_USER, TEST_PASSWORD) assert config.is_valid config = DatabricksConfig.from_token(TEST_HOST, TEST_TOKEN) assert config.is_valid
def test_is_valid_with_token(self): config = DatabricksConfig.from_token(TEST_HOST, TEST_TOKEN) assert not config.is_valid_with_password assert config.is_valid_with_token
def test_from_token(self): config = DatabricksConfig.from_token(TEST_HOST, TEST_TOKEN) assert config.host == TEST_HOST assert config.token == TEST_TOKEN
def _configure_cli_token(profile, insecure): config = get_config_for_profile(profile) host = click.prompt(PROMPT_HOST, default=config.host, type=_DbfsHost()) token = click.prompt(PROMPT_TOKEN, default=config.token) new_config = DatabricksConfig.from_token(host, token, insecure) update_and_persist_config(profile, new_config)