def test_jupyterhub_auth(monkeypatch):
with pytest.raises(ValueError) as exc:
get_auth("jupyterhub")
assert "JUPYTERHUB_API_TOKEN" in str(exc.value)
monkeypatch.setenv("JUPYTERHUB_API_TOKEN", "abcde")
auth = get_auth("jupyterhub")
assert auth.api_token == "abcde"
assert isinstance(auth, JupyterHubAuth)
# Parameters override environment variable
assert JupyterHubAuth(api_token="other").api_token == "other"
def test_config_auth_kwargs_template_environment_vars(monkeypatch):
monkeypatch.setenv("TEST_USER", "bruce")
config = {
"gateway": {"auth": {"type": "basic", "kwargs": {"username": "******"}}}
}
with dask.config.set(config):
auth = get_auth()
assert isinstance(auth, BasicAuth)
assert auth.username == "bruce"
def test_get_auth():
# Pass through existing auth objects
auth = BasicAuth()
assert get_auth(auth) is auth
# Auth by keyword name
auth = get_auth("basic")
assert isinstance(auth, BasicAuth)
auth = get_auth("kerberos")
assert isinstance(auth, KerberosAuth)
# Auth from config
config = {"gateway": {"auth": {"type": "basic", "kwargs": {}}}}
with dask.config.set(config):
auth = get_auth()
assert isinstance(auth, BasicAuth)
# Auth from config with import path
config = {
"gateway": {"auth": {"type": "dask_gateway.auth.BasicAuth", "kwargs": {}}}
}
with dask.config.set(config):
auth = get_auth()
assert isinstance(auth, BasicAuth)
# Auth from config with kwargs
config = {"gateway": {"auth": {"type": "basic", "kwargs": {"username": "******"}}}}
with dask.config.set(config):
auth = get_auth()
assert isinstance(auth, BasicAuth)
assert auth.username == "bruce"
# Errors
with pytest.raises(TypeError):
get_auth(1)
with pytest.raises(TypeError):
get_auth(lambda: 1)
with pytest.raises(ImportError):
get_auth("dask_gateway.auth.Foo")
with pytest.raises(ImportError):
get_auth("not_a_real_module_name_foo_barrr")
